2018-05-15 03:19:49 +02:00
|
|
|
import * as program from 'commander';
|
2018-05-17 21:33:36 +02:00
|
|
|
import * as fs from 'fs';
|
|
|
|
import * as path from 'path';
|
2018-05-15 03:19:49 +02:00
|
|
|
|
|
|
|
import { CipherService } from 'jslib/abstractions/cipher.service';
|
|
|
|
import { FolderService } from 'jslib/services/folder.service';
|
|
|
|
|
|
|
|
import { Response } from '../models/response';
|
2018-05-15 18:49:09 +02:00
|
|
|
import { StringResponse } from '../models/response/stringResponse';
|
2018-05-15 03:19:49 +02:00
|
|
|
|
2018-05-15 18:18:47 +02:00
|
|
|
import { Cipher } from '../models/cipher';
|
|
|
|
import { Folder } from '../models/folder';
|
2018-05-15 03:19:49 +02:00
|
|
|
|
2018-05-17 04:40:48 +02:00
|
|
|
import { CliUtils } from '../utils';
|
|
|
|
|
2018-05-15 03:19:49 +02:00
|
|
|
export class CreateCommand {
|
|
|
|
constructor(private cipherService: CipherService, private folderService: FolderService) { }
|
|
|
|
|
2018-05-17 04:40:48 +02:00
|
|
|
async run(object: string, requestJson: string, cmd: program.Command): Promise<Response> {
|
2018-05-17 21:33:36 +02:00
|
|
|
let req: any = null;
|
|
|
|
if (object !== 'attachment') {
|
|
|
|
if (requestJson == null || requestJson === '') {
|
|
|
|
requestJson = await CliUtils.readStdin();
|
|
|
|
}
|
2018-05-17 04:40:48 +02:00
|
|
|
|
2018-05-17 21:33:36 +02:00
|
|
|
if (requestJson == null || requestJson === '') {
|
|
|
|
return Response.badRequest('`requestJson` was not provided.');
|
|
|
|
}
|
2018-05-17 04:40:48 +02:00
|
|
|
|
2018-05-17 21:33:36 +02:00
|
|
|
try {
|
|
|
|
const reqJson = new Buffer(requestJson, 'base64').toString();
|
|
|
|
req = JSON.parse(reqJson);
|
|
|
|
} catch (e) {
|
|
|
|
return Response.badRequest('Error parsing the encoded request data.');
|
|
|
|
}
|
2018-05-15 16:50:06 +02:00
|
|
|
}
|
|
|
|
|
2018-05-15 03:19:49 +02:00
|
|
|
switch (object.toLowerCase()) {
|
|
|
|
case 'item':
|
|
|
|
return await this.createCipher(req);
|
2018-05-17 21:33:36 +02:00
|
|
|
case 'attachment':
|
|
|
|
return await this.createAttachment(cmd);
|
2018-05-15 03:19:49 +02:00
|
|
|
case 'folder':
|
|
|
|
return await this.createFolder(req);
|
|
|
|
default:
|
|
|
|
return Response.badRequest('Unknown object.');
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-05-15 18:18:47 +02:00
|
|
|
private async createCipher(req: Cipher) {
|
|
|
|
const cipher = await this.cipherService.encrypt(Cipher.toView(req));
|
2018-05-15 03:19:49 +02:00
|
|
|
try {
|
|
|
|
await this.cipherService.saveWithServer(cipher);
|
2018-05-15 18:49:09 +02:00
|
|
|
return Response.success(new StringResponse(cipher.id));
|
2018-05-15 03:19:49 +02:00
|
|
|
} catch (e) {
|
2018-05-15 17:08:55 +02:00
|
|
|
return Response.error(e);
|
2018-05-15 03:19:49 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-05-17 21:33:36 +02:00
|
|
|
private async createAttachment(cmd: program.Command) {
|
|
|
|
if (cmd.itemid == null || cmd.itemid === '') {
|
|
|
|
return Response.badRequest('--itemid <itemid> required.');
|
|
|
|
}
|
|
|
|
if (cmd.file == null || cmd.file === '') {
|
|
|
|
return Response.badRequest('--file <file> required.');
|
|
|
|
}
|
|
|
|
const filePath = path.resolve(cmd.file);
|
|
|
|
if (!fs.existsSync(cmd.file)) {
|
|
|
|
return Response.badRequest('Cannot find file at ' + filePath);
|
|
|
|
}
|
|
|
|
|
|
|
|
// TODO: premium and key check
|
|
|
|
|
2018-05-17 21:55:44 +02:00
|
|
|
const itemId = cmd.itemid.toLowerCase();
|
|
|
|
const cipher = await this.cipherService.get(itemId);
|
2018-05-17 21:33:36 +02:00
|
|
|
if (cipher == null) {
|
|
|
|
return Response.notFound();
|
|
|
|
}
|
|
|
|
|
|
|
|
try {
|
|
|
|
const fileBuf = fs.readFileSync(filePath);
|
|
|
|
await this.cipherService.saveAttachmentRawWithServer(cipher, path.basename(filePath),
|
|
|
|
new Uint8Array(fileBuf).buffer);
|
|
|
|
return Response.success();
|
|
|
|
} catch (e) {
|
|
|
|
return Response.error(e);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-05-15 18:18:47 +02:00
|
|
|
private async createFolder(req: Folder) {
|
|
|
|
const folder = await this.folderService.encrypt(Folder.toView(req));
|
2018-05-15 03:19:49 +02:00
|
|
|
try {
|
|
|
|
await this.folderService.saveWithServer(folder);
|
2018-05-15 18:49:09 +02:00
|
|
|
return Response.success(new StringResponse(folder.id));
|
2018-05-15 03:19:49 +02:00
|
|
|
} catch (e) {
|
2018-05-15 17:08:55 +02:00
|
|
|
return Response.error(e);
|
2018-05-15 03:19:49 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|