added auth requirement, with echo.config.json for referral code

Author: Trynax

packages/app/control/src/app/api/v1/referrals/register-template/route.ts

@@ -5,6 +5,7 @@ import {
   outro,
   select,
   text,
+  confirm,
   spinner,
   log,
   isCancel,
@@ -14,6 +15,7 @@ import chalk from 'chalk';
 import { Command } from 'commander';
 import degit from 'degit';
 import { existsSync, readdirSync, readFileSync, writeFileSync } from 'fs';
+import { readFile } from 'fs/promises';
 import path from 'path';
 import { spawn } from 'child_process';
 
@@ -185,65 +187,67 @@ function isExternalTemplate(template: string): boolean {
   );
 }
 
-function extractGithubUsername(templateUrl: string): string | null {
-  const match = templateUrl.match(
-    /github\.com\/([^\/]+)/
-  );
-  return match?.[1] ?? null;
+async function extractReferralCodeFromTemplate(
+  templatePath: string
+): Promise<string | null> {
+  const configFile = path.join(templatePath, 'echo.config.json');
+
+  if (!existsSync(configFile)) {
+    return null;
+  }
+
+  try {
+    const content = await readFile(configFile, 'utf-8');
+    const config = JSON.parse(content);
+    return config.referralCode || config.echo?.referralCode || null;
+  } catch {
+    return null;
+  }
 }
 
 async function registerTemplateReferral(
   appId: string,
-  templateUrl: string
+  templatePath: string,
+  apiKey: string
 ): Promise<void> {
-  const githubUsername = extractGithubUsername(templateUrl);
-  
-  if (!githubUsername) {
-    log.warn('Could not extract GitHub username from template URL');
-    return;
-  }
-
   try {
-    log.step(`Registering template referral for ${githubUsername}...`);
-    
-    const response = await fetch(
-      `${ECHO_BASE_URL}/api/v1/referrals/register-template`,
-      {
-        method: 'POST',
-        headers: {
-          'Content-Type': 'application/json',
-        },
-        body: JSON.stringify({
-          appId,
-          githubUsername,
-          templateUrl,
-        }),
-      }
-    );
+    const referralCode = await extractReferralCodeFromTemplate(templatePath);
 
-    if (!response.ok) {
-      const errorText = await response.text();
-      log.warn(`Referral registration failed: ${response.status} - ${errorText}`);
+    if (!referralCode) {
+      log.info('No referral code found in template echo.config.json');
       return;
     }
 
-    const result = (await response.json()) as {
-      status?: string;
-      referrerUsername?: string;
-      message?: string;
-    };
-    
-    if (result.status === 'registered' && result.referrerUsername) {
-      log.success(
-        `Template creator ${result.referrerUsername} registered as referrer`
+    log.step(`Found template referral code, applying...`);
+
+    const response = await fetch(`${ECHO_BASE_URL}/api/v1/user/referral`, {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json',
+        Authorization: `Bearer ${apiKey}`,
+      },
+      body: JSON.stringify({
+        echoAppId: appId,
+        code: referralCode,
+      }),
+    });
+
+    if (!response.ok) {
+      const errorData = (await response.json()) as { message?: string };
+      log.warn(
+        `Referral code could not be applied: ${errorData.message || 'Unknown error'}`
       );
-    } else if (result.status === 'skipped') {
-      log.info(`Referral skipped: ${result.message || 'Unknown reason'}`);
-    } else if (result.status === 'not_found') {
-      log.info(`Template creator ${githubUsername} not found on Echo`);
+      return;
+    }
+
+    const result = (await response.json()) as { success?: boolean };
+    if (result.success) {
+      log.success('Template referral code applied successfully');
     }
   } catch (error) {
-    log.warn(`Referral registration error: ${error instanceof Error ? error.message : 'Unknown error'}`);
+    log.warn(
+      `Referral registration error: ${error instanceof Error ? error.message : 'Unknown error'}`
+    );
   }
 }
 
@@ -372,10 +376,6 @@ async function createApp(projectDir: string, options: CreateAppOptions) {
 
   log.step(`Using App ID: ${appId}`);
 
-  if (isExternal) {
-    await registerTemplateReferral(appId, template);
-  }
-
   const absoluteProjectPath = path.resolve(projectDir);
 
   // Check if directory already exists
@@ -436,7 +436,36 @@ async function createApp(projectDir: string, options: CreateAppOptions) {
 
     log.step('Configuring project files');
 
-    // Update package.json with the name of the project
+    if (isExternal) {
+      const referralCode = await extractReferralCodeFromTemplate(
+        absoluteProjectPath
+      );
+
+      if (referralCode) {
+        const shouldApplyReferral = await confirm({
+          message: 'This template includes a referral code. Apply it?',
+          initialValue: true,
+        });
+
+        if (!isCancel(shouldApplyReferral) && shouldApplyReferral) {
+          const apiKey = await text({
+            message: 'Enter your Echo API key:',
+            placeholder: 'Your API key from https://echo.merit.systems/keys',
+            validate: (value: string) => {
+              if (!value.trim()) {
+                return 'API key is required to apply referral code';
+              }
+              return;
+            },
+          });
+
+          if (!isCancel(apiKey)) {
+            await registerTemplateReferral(appId, absoluteProjectPath, apiKey);
+          }
+        }
+      }
+    }
+
     const packageJsonPath = path.join(absoluteProjectPath, 'package.json');
     // Technically this is checked above, but good practice to check again
     if (existsSync(packageJsonPath)) {