create attachment command

src/commands/create.command.ts

@@ -1,4 +1,6 @@
 import * as program from 'commander';
+import * as fs from 'fs';
+import * as path from 'path';
 
 import { CipherService } from 'jslib/abstractions/cipher.service';
 import { FolderService } from 'jslib/services/folder.service';
@@ -6,6 +8,7 @@ import { FolderService } from 'jslib/services/folder.service';
 import { Response } from '../models/response';
 import { StringResponse } from '../models/response/stringResponse';
 
+import { Attachment } from '../models/attachment';
 import { Cipher } from '../models/cipher';
 import { Folder } from '../models/folder';
 
@@ -15,6 +18,8 @@ export class CreateCommand {
     constructor(private cipherService: CipherService, private folderService: FolderService) { }
 
     async run(object: string, requestJson: string, cmd: program.Command): Promise<Response> {
+        let req: any = null;
+        if (object !== 'attachment') {
             if (requestJson == null || requestJson === '') {
                 requestJson = await CliUtils.readStdin();
             }
@@ -23,17 +28,19 @@ export class CreateCommand {
                 return Response.badRequest('`requestJson` was not provided.');
             }
 
-        let req: any = null;
             try {
                 const reqJson = new Buffer(requestJson, 'base64').toString();
                 req = JSON.parse(reqJson);
             } catch (e) {
                 return Response.badRequest('Error parsing the encoded request data.');
             }
+        }
 
         switch (object.toLowerCase()) {
             case 'item':
                 return await this.createCipher(req);
+            case 'attachment':
+                return await this.createAttachment(cmd);
             case 'folder':
                 return await this.createFolder(req);
             default:
@@ -51,6 +58,35 @@ export class CreateCommand {
         }
     }
 
+    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
+
+        const cipher = await this.cipherService.get(cmd.itemid);
+        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);
+        }
+    }
+
     private async createFolder(req: Folder) {
         const folder = await this.folderService.encrypt(Folder.toView(req));
         try {

src/program.ts

@@ -267,11 +267,14 @@ export class Program {
 
         program
             .command('create <object> [encodedJson]')
+            .option('--file <file>', 'Path to file for attachment.')
+            .option('--itemid <itemid>', 'Attachment\'s item id.')
             .description('Create an object.')
             .on('--help', () => {
                 writeLn('\n  Objects:');
                 writeLn('');
                 writeLn('    item');
+                writeLn('    attachment');
                 writeLn('    folder');
                 writeLn('');
                 writeLn('  Notes:');
@@ -282,6 +285,7 @@ export class Program {
                 writeLn('');
                 writeLn('    bw create folder eyJuYW1lIjoiTXkgRm9sZGVyIn0K');
                 writeLn('    echo \'eyJuYW1lIjoiTXkgRm9sZGVyIn0K\' | bw create folder');
+                writeLn('    bw create attachment --file ./myfile.csv --itemid 16b15b89-65b3-4639-ad2a-95052a6d8f66');
                 writeLn('');
             })
             .action(async (object, encodedJson, cmd) => {