import * as program from 'commander'; import { CipherType } from 'jslib/enums/cipherType'; import { CipherService } from 'jslib/abstractions/cipher.service'; import { CollectionService } from 'jslib/abstractions/collection.service'; import { FolderService } from 'jslib/abstractions/folder.service'; import { TotpService } from 'jslib/abstractions/totp.service'; import { CipherView } from 'jslib/models/view/cipherView'; import { CollectionView } from 'jslib/models/view/collectionView'; import { FolderView } from 'jslib/models/view/folderView'; import { Response } from '../models/response'; import { CipherResponse } from '../models/response/cipherResponse'; import { CollectionResponse } from '../models/response/collectionResponse'; import { FolderResponse } from '../models/response/folderResponse'; import { StringResponse } from '../models/response/stringResponse'; import { TemplateResponse } from '../models/response/templateResponse'; import { Card } from '../models/card'; import { Cipher } from '../models/cipher'; import { Collection } from '../models/collection'; import { Field } from '../models/field'; import { Folder } from '../models/folder'; import { Identity } from '../models/identity'; import { Login } from '../models/login'; import { LoginUri } from '../models/loginUri'; import { SecureNote } from '../models/secureNote'; import { CliUtils } from '../utils'; export class GetCommand { constructor(private cipherService: CipherService, private folderService: FolderService, private collectionService: CollectionService, private totpService: TotpService) { } async run(object: string, id: string, cmd: program.Command): Promise { if (id != null) { id = id.toLowerCase(); } switch (object.toLowerCase()) { case 'item': return await this.getCipher(id); case 'totp': return await this.getTotp(id); case 'folder': return await this.getFolder(id); case 'collection': return await this.getCollection(id); case 'template': return await this.getTemplate(id); default: return Response.badRequest('Unknown object.'); } } private async getCipher(id: string) { let decCipher: CipherView = null; if (this.isGuid(id)) { const cipher = await this.cipherService.get(id); if (cipher != null) { decCipher = await cipher.decrypt(); } } else if (id.trim() !== '') { let ciphers = await this.cipherService.getAllDecrypted(); ciphers = CliUtils.searchCiphers(ciphers, id); if (ciphers.length > 1) { return Response.multipleResults(ciphers.map((c) => c.id)); } if (ciphers.length > 0) { decCipher = ciphers[0]; } } if (decCipher == null) { return Response.notFound(); } const res = new CipherResponse(decCipher); return Response.success(res); } private async getTotp(id: string) { const cipherResponse = await this.getCipher(id); if (!cipherResponse.success) { return cipherResponse; } const cipher = cipherResponse.data as CipherResponse; if (cipher.type !== CipherType.Login) { return Response.badRequest('Not a login.'); } if (cipher.login.totp == null || cipher.login.totp === '') { return Response.error('No TOTP available for this login.'); } const totp = await this.totpService.getCode(cipher.login.totp); if (totp == null) { return Response.error('Couldn\'t generate TOTP code.'); } const res = new StringResponse(totp); return Response.success(res); } private async getFolder(id: string) { let decFolder: FolderView = null; if (this.isGuid(id)) { const folder = await this.folderService.get(id); if (folder != null) { decFolder = await folder.decrypt(); } } else if (id.trim() !== '') { let folders = await this.folderService.getAllDecrypted(); folders = CliUtils.searchFolders(folders, id); if (folders.length > 1) { return Response.multipleResults(folders.map((f) => f.id)); } if (folders.length > 0) { decFolder = folders[0]; } } if (decFolder == null) { return Response.notFound(); } const res = new FolderResponse(decFolder); return Response.success(res); } private async getCollection(id: string) { let decCollection: CollectionView = null; if (this.isGuid(id)) { const collection = await this.collectionService.get(id); if (collection != null) { decCollection = await collection.decrypt(); } } else if (id.trim() !== '') { let collections = await this.collectionService.getAllDecrypted(); collections = CliUtils.searchCollections(collections, id); if (collections.length > 1) { return Response.multipleResults(collections.map((c) => c.id)); } if (collections.length > 0) { decCollection = collections[0]; } } if (decCollection == null) { return Response.notFound(); } const res = new CollectionResponse(decCollection); return Response.success(res); } private isGuid(id: string) { return RegExp(/^[0-9a-f]{8}-[0-9a-f]{4}-[1-5][0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/, 'i').test(id); } private async getTemplate(id: string) { let template: any = null; switch (id.toLowerCase()) { case 'item': template = Cipher.template(); break; case 'field': template = Field.template(); break; case 'login': template = Login.template(); break; case 'loginuri': template = LoginUri.template(); break; case 'card': template = Card.template(); break; case 'identity': template = Identity.template(); break; case 'securenote': template = SecureNote.template(); break; case 'folder': template = Folder.template(); break; case 'collection': template = Collection.template(); break; default: return Response.badRequest('Unknown template object.'); } const res = new TemplateResponse(template); return Response.success(res); } }