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

167 lines
6.8 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
2019-09-25 22:08:56 +02:00
import { ApiService } from 'jslib/abstractions/api.service';
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';
2019-09-25 22:08:56 +02:00
import { Collection } from 'jslib/models/export/collection';
2018-12-17 16:30:06 +01:00
import { Folder } from 'jslib/models/export/folder';
2019-09-25 22:08:56 +02:00
import { CollectionRequest } from 'jslib/models/request/collectionRequest';
import { SelectionReadOnlyRequest } from 'jslib/models/request/selectionReadOnlyRequest';
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';
2019-09-25 22:08:56 +02:00
import { OrganizationCollectionResponse } from '../models/response/organizationCollectionResponse';
import { OrganizationCollectionRequest } from '../models/request/organizationCollectionRequest';
2018-05-15 03:19:49 +02:00
import { CliUtils } from '../utils';
2019-10-01 17:16:24 +02:00
import { Utils } from 'jslib/misc/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,
2019-09-25 22:08:56 +02:00
private userService: UserService, private cryptoService: CryptoService,
private apiService: ApiService) { }
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);
2019-09-25 22:08:56 +02:00
case 'org-collection':
2019-10-01 17:16:24 +02:00
return await this.createOrganizationCollection(req, cmd);
2018-05-15 03:19:49 +02:00
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
}
}
2019-09-25 22:08:56 +02:00
2019-10-01 17:16:24 +02:00
private async createOrganizationCollection(req: OrganizationCollectionRequest, cmd: program.Command) {
if (cmd.organizationid == null || cmd.organizationid === '') {
return Response.badRequest('--organizationid <organizationid> required.');
}
if (!Utils.isGuid(cmd.organizationid)) {
return Response.error('`' + cmd.organizationid + '` is not a GUID.');
}
if (cmd.organizationid !== req.organizationId) {
return Response.error('--organizationid <organizationid> does not match request object.');
}
2019-09-25 22:08:56 +02:00
try {
const orgKey = await this.cryptoService.getOrgKey(req.organizationId);
if (orgKey == null) {
throw new Error('No encryption key for this organization.');
}
const groups = req.groups == null ? null :
req.groups.map((g) => new SelectionReadOnlyRequest(g.id, g.readOnly, g.hidePasswords));
2019-09-25 22:08:56 +02:00
const request = new CollectionRequest();
request.name = (await this.cryptoService.encrypt(req.name, orgKey)).encryptedString;
request.externalId = req.externalId;
request.groups = groups;
const response = await this.apiService.postCollection(req.organizationId, request);
const view = Collection.toView(req);
view.id = response.id;
const res = new OrganizationCollectionResponse(view, groups);
2019-09-25 22:08:56 +02:00
return Response.success(res);
} catch (e) {
return Response.error(e);
}
}
2018-05-15 03:19:49 +02:00
}