Introduce save flag for saving to context on creation (#114)

Author: wolfo951

docs/imodel/create.md

@@ -30,6 +30,10 @@ Create an empty iModel within a specified iTwin.
     - **`latitude`**: `number`  
     - **`longitude`**: `number`
 
+- **`--save`**
+  Save the iTwin id to the [context](./../environment.md).  
+  **Type:** `flag` **Required:** No
+
 ## Examples
 
 ```bash

docs/itwin/create.md

@@ -51,6 +51,10 @@ Create a new iTwin with specified properties.
   Optional parent iTwin Id. Defaults to user's Account iTwin.  
   **Type:** `string` **Required:** No
 
+- **`--save`**
+  Save the iTwin id to the [context](./../environment.md).  
+  **Type:** `flag` **Required:** No
+
 ## Examples
 
 ```bash

src/commands/imodel/create.ts

@@ -52,6 +52,10 @@ export default class CreateIModel extends BaseCommand {
         helpValue: '<string>',
         required: true,
       }),
+      save: Flags.boolean({
+        description: 'Save the iModel id to the context.',
+        required: false,
+      }),
     };
 
     async run() {
@@ -71,7 +75,9 @@ export default class CreateIModel extends BaseCommand {
           name: flags.name,
         },
       });
-  
+
+      this.setContext(iModel.iTwinId, iModel.id);
+        
       return this.logAndReturnResult(iModel);
     }
   }

src/commands/itwin/create.ts

@@ -73,6 +73,10 @@ export default class CreateITwin extends BaseCommand {
         helpValue: '<string>',
         required: false,
       }),
+      save: Flags.boolean({
+        description: 'Save the iTwin id to the context.',
+        required: false,
+      }),
       status: Flags.string({
         description: 'Status of the iTwin. Defaults to Active.',
         helpValue: '<string>',
@@ -116,6 +120,15 @@ export default class CreateITwin extends BaseCommand {
       {
         this.error(JSON.stringify(creatediTwin.error, null, 2));
       }
+
+      if (flags.save) {
+        if(creatediTwin.data?.id === undefined){
+          this.log("iTwin Id not found in response. Cannot save to context.");
+        }
+        else {
+          this.setContext(creatediTwin.data.id);
+        }
+      }
 
       return this.logAndReturnResult(creatediTwin.data);
     }

src/extensions/base-command.ts

@@ -263,7 +263,7 @@ export default abstract class BaseCommand extends Command {
     return this.config.runCommand<T>(command, mergedArgs);
   }
 
-  protected async setContext(iTwinId: string, iModelId: string | undefined) {
+  protected async setContext(iTwinId: string, iModelId?: string) {
     const contextPath = this.config.cacheDir + '/context.json';
 
     const context: UserContext = {
@@ -277,6 +277,8 @@ export default abstract class BaseCommand extends Command {
 
     fs.writeFileSync(contextPath, JSON.stringify(context, null, 2), 'utf8');
 
+    this.log(`Context set to iTwinId: ${iTwinId}, iModelId: ${iModelId}`);
+
     return context;
   }
 }