feat: 🎸 Associate function to a site on creation

locales/en.json

@@ -357,5 +357,6 @@
   "uploadingAssets": "Uploading assets...",
   "assetsUploadSuccess": "Assets uploaded successfully",
   "uploadAssetsFailed": "Failed to upload function assets to Fleek. Check your network connection and try again.",
-  "assetsNotSupportedInSgx": "Function assets are not supported in SGX"
+  "assetsNotSupportedInSgx": "Function assets are not supported in SGX",
+  "functionsSite": "The ID of the site you want to deploy your function to"
 }

src/commands/functions/create.ts

@@ -1,20 +1,30 @@
+import { SiteNotFoundError } from '@fleek-platform/errors';
 import { output } from '../../cli';
 import type { SdkGuardedFunction } from '../../guards/types';
 import { withGuards } from '../../guards/withGuards';
 import { t } from '../../utils/translation';
 import { getFunctionNameOrPrompt } from './prompts/getFunctionNameOrPrompt';
+import { isSiteIdValid } from './utils/isSiteIdValid';
 
 type CreateFunctionArgs = {
   name?: string;
+  siteId?: string;
 };
 
 const createAction: SdkGuardedFunction<CreateFunctionArgs> = async ({
   args,
   sdk,
 }) => {
-  const functionName = await getFunctionNameOrPrompt({ name: args.name });
+  const { name, siteId } = args;
+  const functionName = await getFunctionNameOrPrompt({ name });
 
-  const newFunction = await sdk.functions().create({ name: functionName });
+  if (siteId && !(await isSiteIdValid({ siteId, sdk }))) {
+    throw new SiteNotFoundError({ site: { id: siteId } });
+  }
+
+  const newFunction = await sdk
+    .functions()
+    .create({ name: functionName, siteId });
 
   output.printNewLine();
   output.success(t('commonNameCreateSuccess', { name: 'function' }));

src/commands/functions/index.ts

@@ -27,9 +27,10 @@ export default (program: Command): Command => {
   cmd
     .command('create')
     .option('-n, --name <functionName>', t('functionName'))
+    .option('--site <siteId>', t('functionsSite'))
     .description(t('functionsCreateDescription'))
-    .action((options: { name?: string }) =>
-      createActionHandler({ name: options.name }),
+    .action((options: { name?: string; site?: string }) =>
+      createActionHandler({ name: options.name, siteId: options.site }),
     );
 
   cmd

src/commands/functions/utils/isSiteIdValid.ts

@@ -0,0 +1,14 @@
+import type { FleekSdk } from '@fleek-platform/sdk/node';
+
+export const isSiteIdValid = ({
+  siteId,
+  sdk,
+}: { siteId: string; sdk: FleekSdk }) => {
+  if (!siteId) return true;
+
+  return sdk
+    .sites()
+    .get({ id: siteId })
+    .then(() => true)
+    .catch(() => false);
+};