Support benchmark create updates for v2 (#15)

Author: MicroFish91

src/commands/benchmarks/createBenchmark/BenchmarkCreateContext.ts

@@ -9,6 +9,7 @@ export type BenchmarkCreateContext = AuthContext & {
     benchmarkRealReturn?: number;
     benchmarkStdDev?: number;
     benchmarkDrawdownYears?: number;
+    benchmarkRebalanceThresholdPct?: number;
 
     benchmark?: Benchmark;
 };

src/commands/benchmarks/createBenchmark/BenchmarkCreateStep.ts

@@ -32,6 +32,7 @@ export class BenchmarkCreateStep<T extends BenchmarkCreateContext> extends Execu
             real_return_pct: context.benchmarkRealReturn,
             std_dev_pct: context.benchmarkStdDev,
             drawdown_yrs: context.benchmarkDrawdownYears,
+            rec_rebalance_threshold_pct: context.benchmarkRebalanceThresholdPct,
             is_deprecated: false,
         });
         if (response.error) {

src/commands/benchmarks/createBenchmark/BenchmarkRebalanceThresholdStep.ts

@@ -0,0 +1,35 @@
+import { l10n } from "vscode";
+import { PromptStep } from "../../../wizard/PromptStep";
+import { validationUtils } from "../../../utils/validationUtils";
+import { BenchmarkCreateContext } from "./BenchmarkCreateContext";
+
+export class BenchmarkRebalanceThresholdStep<T extends BenchmarkCreateContext> extends PromptStep<T> {
+    async prompt(context: T): Promise<void> {
+        context.benchmarkRebalanceThresholdPct = Number((await context.ui.showInputBox({
+            title: this.title,
+            prompt: l10n.t('Enter benchmark\'s recommended rebalance threshold (%)'),
+            validateInput: this.validateInput,
+        })));
+    }
+
+    shouldPrompt(context: T): boolean {
+        return !context.benchmarkRebalanceThresholdPct;
+    }
+
+    private validateInput(value: string): string | undefined {
+        if (!validationUtils.hasValidCharLength(value)) {
+            return validationUtils.getInvalidCharLengthMessage();
+        }
+
+        const num: number = Number(value);
+        if (isNaN(num) || !Number.isInteger(num)) {
+            return l10n.t('Value must be a whole number.');
+        }
+
+        if (num < 1 || num > 99) {
+            return l10n.t('Value must be between 1 and 99.');
+        }
+
+        return undefined;
+    }
+}

src/commands/benchmarks/createBenchmark/createBenchmark.ts

@@ -14,6 +14,7 @@ import { BenchmarkDrawdownYearsStep } from "./BenchmarkDrawdownYearsStep";
 import { BenchmarkAssetCategoriesStep } from "./BenchmarkAssetCategoriesStep";
 import { BenchmarkAssetAmountsStep } from "./BenchmarkAssetAmountsStep";
 import { BenchmarkCreateStep } from "./BenchmarkCreateStep";
+import { BenchmarkRebalanceThresholdStep } from "./BenchmarkRebalanceThresholdStep";
 
 export async function createBenchmark(context: CommandContext, item: BenchmarksItem): Promise<void> {
     const wizardContext: BenchmarkCreateContext = {
@@ -29,6 +30,7 @@ export async function createBenchmark(context: CommandContext, item: BenchmarksI
             new BenchmarkDescriptionStep(),
             new BenchmarkAssetCategoriesStep(),
             new BenchmarkAssetAmountsStep(),
+            new BenchmarkRebalanceThresholdStep(),
             new BenchmarkRealReturnStep(),
             new BenchmarkStdDevStep(),
             new BenchmarkDrawdownYearsStep(),