Fix knobs for widgetbook 3.0.0, fix generating widgetbook entries for…

… the whole directory at once

src/commands/generate_entries_for_directory.ts

@@ -30,7 +30,7 @@ async function generateWidgetbookEntriesForDirectoryImpl(
 
       const clazz = parseTextToClass(fileContentString);
 
-      await writeWidgetbookEntry(clazz);
+      await writeWidgetbookEntry(clazz, filePath);
     } else if (fileType === vscode.FileType.Directory) {
       const subdirectoryPath = path.join(directoryPath, fileName);
 

src/commands/generate_entry_for_widget.ts

@@ -15,7 +15,7 @@ async function generateWidgetbookEntryForWidgetInScope(): Promise<void> {
 
   const clazz = parseTextToClass(fileContentFromCurrentLine);
 
-  await writeWidgetbookEntry(clazz);
+  await writeWidgetbookEntry(clazz, activeEditor.document.fileName);
 }
 
 export { generateWidgetbookEntryForWidgetInScope };

src/generators/file_content/base_generator.ts

@@ -52,7 +52,7 @@ abstract class BaseFileContentGenerator implements FileContentGenerator {
   >([
     [
       "bool",
-      (fieldName) => `context.knobs.nullableBoolean(label: '${fieldName}')`,
+      (fieldName) => `context.knobs.booleanOrNull(label: '${fieldName}')`,
     ],
     [
       "String",

src/generators/file_content/impl/3_2_0_generator.ts

@@ -6,32 +6,12 @@ import { BaseFileContentGenerator } from "../base_generator";
 
 class FileContentGenerator3_2_0 extends BaseFileContentGenerator {
   applyMigrations(): void {
-    this.knobForType.set(
-      "String",
-      (fieldName) =>
-        `context.knobs.string(label: '${fieldName}', initialValue: '${fieldName}')`
-    );
     this.knobForType.set("double", (fieldName) =>
       this.numberKnob(fieldName, "")
     );
-    this.knobForType.set("int", (fieldName) =>
-      this.numberKnob(fieldName, ".toInt()")
-    );
-
-    this.knobForNullableType.set(
-      "bool",
-      (fieldName) => `context.knobs.booleanOrNull(label: '${fieldName}')`
-    );
-    this.knobForNullableType.set(
-      "String",
-      (fieldName) => `context.knobs.stringOrNull(label: '${fieldName}')`
-    );
     this.knobForNullableType.set("double", (fieldName) =>
       this.nullableNumberKnob(fieldName, "")
     );
-    this.knobForNullableType.set("int", (fieldName) =>
-      this.nullableNumberKnob(fieldName, "?.toInt()")
-    );
   }
 
   manualComponentDeclaration(): string {

src/generators/path/generator.ts

@@ -1,5 +1,8 @@
 interface PathGenerator {
-  prepareWidgetbookEntryFilePath(className: string): string | null;
+  prepareWidgetbookEntryFilePath(
+    className: string,
+    widgetFilePath: string
+  ): string | null;
 }
 
 export { PathGenerator };

src/generators/path/generator_impl.ts

@@ -1,21 +1,20 @@
 import { snakeCase } from "change-case";
-import * as vscode from "vscode";
 import { Configuration } from "../../configuration/configuration";
 import { PathGenerator } from "./generator";
 
 class PathGeneratorImpl implements PathGenerator {
-  prepareWidgetbookEntryFilePath(className: string): string | null {
-    const activeTextEditor = vscode.window.activeTextEditor;
-    if (!activeTextEditor) return null;
-
-    const currentPath = activeTextEditor.document.fileName;
-    const rootDir = Configuration.rootProjectDirectoryName();
-    const widgetbookWidgetsDir = Configuration.widgetsDirectoryPath();
+  prepareWidgetbookEntryFilePath(
+    className: string,
+    widgetFilePath: string
+  ): string | null {
+    const rootDirectory = Configuration.rootProjectDirectoryName();
+    const widgetbookWidgetsDirectory = Configuration.widgetsDirectoryPath();
     const classNameSnakeCase = snakeCase(className);
-    const projectRootPath = currentPath.substringUpToAndIncluding(rootDir);
-    const widgetFilePath = `${projectRootPath}/${widgetbookWidgetsDir}/${classNameSnakeCase}.dart`;
+    const projectRootPath =
+      widgetFilePath.substringUpToAndIncluding(rootDirectory);
+    const outputFilePath = `${projectRootPath}/${widgetbookWidgetsDirectory}/${classNameSnakeCase}.dart`;
 
-    return widgetFilePath;
+    return outputFilePath;
   }
 }
 export { PathGeneratorImpl };

src/util/file_util.ts

@@ -8,10 +8,16 @@ import { PathGeneratorFactory } from "../generators/path/factory";
 
 const ENCODING = "utf8";
 
-async function writeWidgetbookEntry(clazz: DartClass): Promise<void> {
+async function writeWidgetbookEntry(
+  clazz: DartClass,
+  widgetFilePath: string
+): Promise<void> {
   const pathGenerator = PathGeneratorFactory.create();
 
-  const filePath = pathGenerator.prepareWidgetbookEntryFilePath(clazz.name);
+  const filePath = pathGenerator.prepareWidgetbookEntryFilePath(
+    clazz.name,
+    widgetFilePath
+  );
 
   if (filePath === null) {
     // TODO Show error dialog or something