Added new codify file resolution logic. WIP default document endpoint fetch

Author: kevinwang5658

src/api/dashboard/index.ts

@@ -21,5 +21,24 @@ export const DashboardApiClient = {
     }
 
     return json;
+  },
+
+  async getDefaultDocumentId(): Promise<string> {
+    const login = LoginHelper.get()?.credentials;
+    if (!login) {
+      throw new Error('Not logged in');
+    }
+
+    // const res = await fetch(
+    //   `${API_BASE_URL}/api/v1/documents/default/id`,
+    //   { method: 'GET', headers: { 'Content-Type': 'application/json', 'authorization': login.accessToken } },
+    // );
+
+    // const json = await res.json();
+    // if (!res.ok) {
+    //   throw new Error(JSON.stringify(json, null, 2));
+    // }
+
+    return '1b80818e-5304-4158-80a3-82e17ff2c79e';
   }
 }

src/common/initialize-plugins.ts

@@ -8,6 +8,9 @@ import { CODIFY_FILE_REGEX, CodifyParser } from '../parser/index.js';
 import { DependencyMap, PluginManager } from '../plugins/plugin-manager.js';
 import { Reporter } from '../ui/reporters/reporter.js';
 import { LoginHelper } from '../connect/login-helper.js';
+import { FileUtils } from '../utils/file.js';
+import { validate } from 'uuid';
+import { DashboardApiClient } from '../api/dashboard/index.js';
 
 export interface InitializeArgs {
   path?: string;
@@ -28,11 +31,11 @@ export class PluginInitOrchestrator {
     args: InitializeArgs,
     reporter: Reporter,
   ): Promise<InitializationResult> {
+    const codifyPath = await PluginInitOrchestrator.resolveCodifyRootPath(args, reporter);
+
     let project = await PluginInitOrchestrator.parse(
-      args.path,
-      args.allowEmptyProject ?? false,
-      reporter
-    )
+      codifyPath,
+    );
     if (args.transformProject) {
       project = await args.transformProject(project);
     }
@@ -47,32 +50,9 @@ export class PluginInitOrchestrator {
 
   private static async parse(
     fileOrDir: string | undefined,
-    allowEmptyProject: boolean,
-    reporter: Reporter
   ): Promise<Project> {
     ctx.subprocessStarted(SubProcessName.PARSE);
 
-    const isLoggedIn = LoginHelper.get()?.isLoggedIn ?? false;
-
-    if (!fileOrDir && !allowEmptyProject && !isLoggedIn) {
-      ctx.subprocessFinished(SubProcessName.PARSE);
-      ctx.subprocessStarted(SubProcessName.CREATE_ROOT_FILE)
-      const createRootCodifyFile = await reporter.promptConfirmation('\nNo codify file found. Do you want to create a root file at ~/codify.jsonc?');
-
-      if (createRootCodifyFile) {
-        await fs.writeFile(
-          path.resolve(os.homedir(), 'codify.jsonc'),
-          '[]',
-          { encoding: 'utf8', flag: 'wx' }
-        ); // flag: 'wx' prevents overwrites if the file exists
-      }
-
-      ctx.subprocessFinished(SubProcessName.CREATE_ROOT_FILE)
-
-      console.log('Created ~/codify.jsonc file')
-      process.exit(0);
-    }
-
     const project = fileOrDir
       ? await CodifyParser.parse(fileOrDir)
       : Project.empty()
@@ -82,19 +62,59 @@ export class PluginInitOrchestrator {
     return project
   }
 
-  private static async findCodifyJson(dir?: string): Promise<null | string> {
-    dir = dir ?? process.cwd();
+  /** Resolve the root codify file to run.
+   * Order:
+   * 1. If path is specified, return that.
+   * 2. If path is a dir with only one *codify.json|*codify.jsonc|*codify.json5|*codify.yaml, return that.
+   * 3. If path is a UUID, return file from Codify cloud.
+   * 4. If multiple exists in the path (dir), then prompt the user to select one.
+   * 5. If no path is provided, run steps 2 - 4 for the current dir.
+   * 6. If none exists, return default file from codify cloud.
+   * 7. If user is not logged in, return an error.
+   *
+   * @param args
+   * @private
+   */
+  private static async resolveCodifyRootPath(args: InitializeArgs, reporter: Reporter): Promise<string | undefined> {
+    const inputPath = args.path ?? process.cwd();
+
+    // Cloud files will be fetched and processed later in the parser.
+    const isCloud = validate(inputPath);
+    if (isCloud) {
+      return inputPath;
+    }
+
+    // Direct files can have its path returned.
+    const isPathDir = await FileUtils.isDir(inputPath);
+    if (!isPathDir) {
+      return inputPath;
+    }
+
+    const filesInDir = await fs.readdir(inputPath);
+    const codifyFiles = filesInDir.filter((f) => CODIFY_FILE_REGEX.test(f))
+
+    if (codifyFiles.length === 1) {
+      return codifyFiles[0];
+    }
+
+    if (codifyFiles.length > 0) {
+      const answer = await reporter.promptOptions(
+        'Multiple codify files found in dir. Please select one:',
+        codifyFiles,
+      );
+
+      return path.join(inputPath, codifyFiles[answer]);
+    }
 
-    const filesInDir = await fs.readdir(dir);
-    if (filesInDir.some((f) => CODIFY_FILE_REGEX.test(f))) {
-      return dir;
+    if (LoginHelper.get()?.isLoggedIn) {
+      return DashboardApiClient.getDefaultDocumentId();
     }
 
-    if (dir.includes(os.homedir()) && dir !== os.homedir()) {
-      return this.findCodifyJson(path.dirname(dir))
+    if (args.allowEmptyProject) {
+      return undefined;
     }
 
-    return null;
+    throw new Error('No codify files found.');
   }
 
 }

src/parser/index.ts

@@ -71,6 +71,7 @@ class Parser {
 
     return Promise.all(filePaths.map(
       async (p) => {
+        // If path is a uuid and doesn't exist as a file, it's a cloud file
         if (validate(p) && !(await FileUtils.fileExists(p))) {
           return cloudReader.read(p)
         }

src/ui/reporters/default-reporter.tsx

@@ -219,6 +219,8 @@ export class DefaultReporter implements Reporter {
   }
 
   async promptOptions(message:string, options:string[]): Promise<number> {
+    const prevRenderState = this.getRenderState();
+
     const result = await this.updateStateAndAwaitEvent<string>(
       () => this.updateRenderState(RenderStatus.PROMPT_OPTIONS, { message, options }),
       RenderEvent.PROMPT_RESULT
@@ -229,7 +231,7 @@ export class DefaultReporter implements Reporter {
     // This was added because there was a very hard to debug memory bug with Yoga (ink.js layout engine). Could not
     // identify the root cause of the problem but this alleviates it.
     await sleep(50)
-    this.updateRenderState(RenderStatus.NOTHING, null);
+    this.updateRenderState(prevRenderState.status, prevRenderState.data);
     await sleep(50);
 
     return options.indexOf(result);