2018-05-14 19:37:52 +02:00
|
|
|
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';
|
|
|
|
|
2018-05-14 20:54:19 +02:00
|
|
|
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';
|
2018-05-14 22:25:14 +02:00
|
|
|
import { TemplateResponse } from '../models/response/templateResponse';
|
|
|
|
|
2018-05-15 18:18:47 +02:00
|
|
|
import { Card } from '../models/card';
|
|
|
|
import { Cipher } from '../models/cipher';
|
2018-05-15 18:53:08 +02:00
|
|
|
import { Collection } from '../models/collection';
|
2018-05-15 18:18:47 +02:00
|
|
|
import { Field } from '../models/field';
|
2018-05-15 18:53:08 +02:00
|
|
|
import { Folder } from '../models/folder';
|
2018-05-15 18:18:47 +02:00
|
|
|
import { Identity } from '../models/identity';
|
|
|
|
import { Login } from '../models/login';
|
|
|
|
import { LoginUri } from '../models/loginUri';
|
|
|
|
import { SecureNote } from '../models/secureNote';
|
2018-05-14 20:54:19 +02:00
|
|
|
|
2018-05-14 19:37:52 +02:00
|
|
|
export class GetCommand {
|
|
|
|
constructor(private cipherService: CipherService, private folderService: FolderService,
|
|
|
|
private collectionService: CollectionService, private totpService: TotpService) { }
|
|
|
|
|
2018-05-14 20:54:19 +02:00
|
|
|
async run(object: string, id: string, cmd: program.Command): Promise<Response> {
|
2018-05-14 22:25:14 +02:00
|
|
|
switch (object.toLowerCase()) {
|
2018-05-14 19:37:52 +02:00
|
|
|
case 'item':
|
2018-05-14 20:54:19 +02:00
|
|
|
return await this.getCipher(id);
|
2018-05-14 19:37:52 +02:00
|
|
|
case 'totp':
|
2018-05-14 20:54:19 +02:00
|
|
|
return await this.getTotp(id);
|
2018-05-14 19:37:52 +02:00
|
|
|
case 'folder':
|
2018-05-14 20:54:19 +02:00
|
|
|
return await this.getFolder(id);
|
2018-05-14 19:37:52 +02:00
|
|
|
case 'collection':
|
2018-05-14 20:54:19 +02:00
|
|
|
return await this.getCollection(id);
|
2018-05-14 22:25:14 +02:00
|
|
|
case 'template':
|
|
|
|
return await this.getTemplate(id);
|
2018-05-14 19:37:52 +02:00
|
|
|
default:
|
2018-05-14 20:54:19 +02:00
|
|
|
return Response.badRequest('Unknown object.');
|
2018-05-14 19:37:52 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
private async getCipher(id: string) {
|
|
|
|
const cipher = await this.cipherService.get(id);
|
|
|
|
if (cipher == null) {
|
2018-05-14 20:54:19 +02:00
|
|
|
return Response.notFound();
|
2018-05-14 19:37:52 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
const decCipher = await cipher.decrypt();
|
2018-05-14 20:54:19 +02:00
|
|
|
const res = new CipherResponse(decCipher);
|
|
|
|
return Response.success(res);
|
2018-05-14 19:37:52 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
private async getTotp(id: string) {
|
|
|
|
const cipher = await this.cipherService.get(id);
|
|
|
|
if (cipher == null) {
|
2018-05-14 20:54:19 +02:00
|
|
|
return Response.notFound();
|
2018-05-14 19:37:52 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
if (cipher.type !== CipherType.Login) {
|
2018-05-14 20:54:19 +02:00
|
|
|
return Response.badRequest('Not a login.');
|
2018-05-14 19:37:52 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
const decCipher = await cipher.decrypt();
|
|
|
|
if (decCipher.login.totp == null || decCipher.login.totp === '') {
|
2018-05-14 20:54:19 +02:00
|
|
|
return Response.error('No TOTP available for this login.');
|
2018-05-14 19:37:52 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
const totp = await this.totpService.getCode(decCipher.login.totp);
|
|
|
|
if (totp == null) {
|
2018-05-14 20:54:19 +02:00
|
|
|
return Response.error('Couldn\'t generate TOTP code.');
|
2018-05-14 19:37:52 +02:00
|
|
|
}
|
|
|
|
|
2018-05-14 20:54:19 +02:00
|
|
|
const res = new StringResponse(totp);
|
|
|
|
return Response.success(res);
|
2018-05-14 19:37:52 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
private async getFolder(id: string) {
|
|
|
|
const folder = await this.folderService.get(id);
|
|
|
|
if (folder == null) {
|
2018-05-14 20:54:19 +02:00
|
|
|
return Response.notFound();
|
2018-05-14 19:37:52 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
const decFolder = await folder.decrypt();
|
2018-05-14 20:54:19 +02:00
|
|
|
const res = new FolderResponse(decFolder);
|
|
|
|
return Response.success(res);
|
2018-05-14 19:37:52 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
private async getCollection(id: string) {
|
|
|
|
const collection = await this.collectionService.get(id);
|
|
|
|
if (collection == null) {
|
2018-05-14 20:54:19 +02:00
|
|
|
return Response.notFound();
|
2018-05-14 19:37:52 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
const decCollection = await collection.decrypt();
|
2018-05-14 20:54:19 +02:00
|
|
|
const res = new CollectionResponse(decCollection);
|
|
|
|
return Response.success(res);
|
2018-05-14 19:37:52 +02:00
|
|
|
}
|
2018-05-14 22:25:14 +02:00
|
|
|
|
|
|
|
private async getTemplate(id: string) {
|
|
|
|
let template: any = null;
|
|
|
|
switch (id.toLowerCase()) {
|
|
|
|
case 'item':
|
2018-05-15 18:18:47 +02:00
|
|
|
template = Cipher.template();
|
2018-05-14 22:25:14 +02:00
|
|
|
break;
|
|
|
|
case 'field':
|
2018-05-15 18:18:47 +02:00
|
|
|
template = Field.template();
|
2018-05-14 22:25:14 +02:00
|
|
|
break;
|
|
|
|
case 'login':
|
2018-05-15 18:18:47 +02:00
|
|
|
template = Login.template();
|
2018-05-14 22:25:14 +02:00
|
|
|
break;
|
|
|
|
case 'loginuri':
|
2018-05-15 18:18:47 +02:00
|
|
|
template = LoginUri.template();
|
2018-05-14 22:25:14 +02:00
|
|
|
break;
|
|
|
|
case 'card':
|
2018-05-15 18:18:47 +02:00
|
|
|
template = Card.template();
|
2018-05-14 22:25:14 +02:00
|
|
|
break;
|
|
|
|
case 'identity':
|
2018-05-15 18:18:47 +02:00
|
|
|
template = Identity.template();
|
2018-05-14 22:25:14 +02:00
|
|
|
break;
|
|
|
|
case 'securenote':
|
2018-05-15 18:18:47 +02:00
|
|
|
template = SecureNote.template();
|
2018-05-14 22:25:14 +02:00
|
|
|
break;
|
2018-05-15 18:53:08 +02:00
|
|
|
case 'folder':
|
|
|
|
template = Folder.template();
|
|
|
|
break;
|
|
|
|
case 'collection':
|
|
|
|
template = Collection.template();
|
|
|
|
break;
|
2018-05-14 22:25:14 +02:00
|
|
|
default:
|
2018-05-14 23:13:57 +02:00
|
|
|
return Response.badRequest('Unknown template object.');
|
2018-05-14 22:25:14 +02:00
|
|
|
}
|
2018-05-15 05:40:11 +02:00
|
|
|
|
2018-05-14 22:25:14 +02:00
|
|
|
const res = new TemplateResponse(template);
|
|
|
|
return Response.success(res);
|
|
|
|
}
|
2018-05-14 19:37:52 +02:00
|
|
|
}
|