1
0
mirror of https://github.com/bitwarden/browser.git synced 2024-10-30 08:10:34 +01:00
bitwarden-browser/src/commands/create.command.ts

122 lines
4.6 KiB
TypeScript
Raw Normal View History

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';
2018-05-18 16:55:50 +02:00
import { CryptoService } from 'jslib/abstractions/crypto.service';
import { FolderService } from 'jslib/abstractions/folder.service';
import { UserService } from 'jslib/abstractions/user.service';
2018-05-15 03:19:49 +02:00
2018-12-17 16:30:06 +01:00
import { Cipher } from 'jslib/models/export/cipher';
import { Folder } from 'jslib/models/export/folder';
2019-03-16 03:34:59 +01:00
import { Response } from 'jslib/cli/models/response';
import { CipherResponse } from '../models/response/cipherResponse';
import { FolderResponse } from '../models/response/folderResponse';
2018-05-15 03:19:49 +02:00
import { CliUtils } from '../utils';
2018-05-15 03:19:49 +02:00
export class CreateCommand {
2018-05-18 16:55:50 +02:00
constructor(private cipherService: CipherService, private folderService: FolderService,
private userService: UserService, private cryptoService: CryptoService) { }
2018-05-15 03:19:49 +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 21:33:36 +02:00
if (requestJson == null || requestJson === '') {
return Response.badRequest('`requestJson` was not provided.');
}
2018-05-17 21:33:36 +02:00
try {
const reqJson = Buffer.from(requestJson, 'base64').toString();
2018-05-17 21:33:36 +02:00
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);
const newCipher = await this.cipherService.get(cipher.id);
const decCipher = await newCipher.decrypt();
const res = new CipherResponse(decCipher);
return Response.success(res);
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);
}
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();
}
if (cipher.organizationId == null && !(await this.userService.canAccessPremium())) {
return Response.error('Premium status is required to use this feature.');
2018-05-18 16:55:50 +02:00
}
const encKey = await this.cryptoService.getEncKey();
if (encKey == null) {
return Response.error('You must update your encryption key before you can use this feature. ' +
'See https://help.bitwarden.com/article/update-encryption-key/');
}
2018-05-17 21:33:36 +02:00
try {
const fileBuf = fs.readFileSync(filePath);
await this.cipherService.saveAttachmentRawWithServer(cipher, path.basename(filePath),
new Uint8Array(fileBuf).buffer);
const updatedCipher = await this.cipherService.get(cipher.id);
const decCipher = await updatedCipher.decrypt();
const res = new CipherResponse(decCipher);
return Response.success(res);
2018-05-17 21:33:36 +02:00
} 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);
const newFolder = await this.folderService.get(folder.id);
const decFolder = await newFolder.decrypt();
const res = new FolderResponse(decFolder);
return Response.success(res);
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
}
}
}