[ULX-1596] feat: add custom prompts partials

src/context/directory/handlers/prompts.ts

@@ -9,6 +9,7 @@ import {
   Prompts,
   PromptSettings,
   AllPromptsByLanguage,
+  CustomPromptsConfig,
 } from '../../../tools/auth0/handlers/prompts';
 
 type ParsedPrompts = ParsedAsset<'prompts', Prompts>;
@@ -25,6 +26,10 @@ const getCustomTextFile = (promptsDirectory: string) => {
   return path.join(promptsDirectory, 'custom-text.json');
 };
 
+const getPartialsFile = (promptsDirectory: string) => {
+  return path.join(promptsDirectory, 'partials.json');
+};
+
 function parse(context: DirectoryContext): ParsedPrompts {
   const promptsDirectory = getPromptsDirectory(context.filePath);
   if (!existsMustBeDir(promptsDirectory)) return { prompts: null }; // Skip
@@ -47,10 +52,33 @@ function parse(context: DirectoryContext): ParsedPrompts {
     }) as AllPromptsByLanguage;
   })();
 
+  const partials = (() => {
+    const partialsFile = getPartialsFile(promptsDirectory);
+    if (!isFile(partialsFile)) return {};
+    const partialsFileContent = loadJSON(partialsFile, {
+      mappings: context.mappings,
+      disableKeywordReplacement: context.disableKeywordReplacement,
+    }) as CustomPromptsConfig;
+
+    Object.entries(partialsFileContent).forEach(([promptName, partialsArray]) => {
+      partialsArray.forEach((partialConfig, i) => {
+        if (partialConfig.template) {
+          partialsFileContent[promptName][i].template = context.loadFile(
+            path.join(promptsDirectory, partialConfig.template),
+            promptsDirectory
+          );
+        }
+      });
+    });
+
+    return partialsFileContent;
+  })();
+
   return {
     prompts: {
       ...promptsSettings,
       customText,
+      partials,
     },
   };
 }
@@ -60,7 +88,7 @@ async function dump(context: DirectoryContext): Promise<void> {
 
   if (!prompts) return;
 
-  const { customText, ...promptsSettings } = prompts;
+  const { customText, partials, ...promptsSettings } = prompts;
 
   const promptsDirectory = getPromptsDirectory(context.filePath);
   ensureDirSync(promptsDirectory);
@@ -72,6 +100,10 @@ async function dump(context: DirectoryContext): Promise<void> {
   if (!customText) return;
   const customTextFile = getCustomTextFile(promptsDirectory);
   dumpJSON(customTextFile, customText);
+
+  if (!partials) return;
+  const partialsFile = getPartialsFile(promptsDirectory);
+  dumpJSON(partialsFile, partials);
 }
 
 const promptsHandler: DirectoryHandler<ParsedPrompts> = {

src/context/directory/index.ts

@@ -21,7 +21,7 @@ export default class DirectoryContext {
   assets: Assets;
   disableKeywordReplacement: boolean;
 
-  constructor(config: Config, mgmtClient: Auth0APIClient) {
+  constructor(config: Config, mgmtClient) {
     this.filePath = config.AUTH0_INPUT_FILE;
     this.config = config;
     this.mappings = config.AUTH0_KEYWORD_REPLACE_MAPPINGS || {};

src/context/index.ts

@@ -161,19 +161,19 @@ export const setupContext = async (
       return new AuthenticationClient({
         domain: AUTH0_DOMAIN,
         clientId: AUTH0_CLIENT_ID,
-        clientAssertionSigningKey: readFileSync(AUTH0_CLIENT_SIGNING_KEY_PATH),
+        clientAssertionSigningKey: readFileSync(AUTH0_CLIENT_SIGNING_KEY_PATH, 'utf-8'),
         clientAssertionSigningAlg: !!AUTH0_CLIENT_SIGNING_ALGORITHM
           ? AUTH0_CLIENT_SIGNING_ALGORITHM
           : undefined,
       });
     })();
-
-    const clientCredentials = await authClient.clientCredentialsGrant({
+    
+    const clientCredentials = await authClient.oauth.clientCredentialsGrant({
       audience: config.AUTH0_AUDIENCE
         ? config.AUTH0_AUDIENCE
         : `https://${config.AUTH0_DOMAIN}/api/v2/`,
     });
-    return clientCredentials.access_token;
+    return clientCredentials.data.access_token;
   })();
 
   const mgmtClient = new ManagementClient({

src/tools/auth0/handlers/prompts.ts

@@ -153,13 +153,64 @@ export type PromptsCustomText = {
 export type Prompts = Partial<
   PromptSettings & {
     customText: AllPromptsByLanguage;
+    partials: CustomPromptPartials;
   }
 >;
 
 export type AllPromptsByLanguage = Partial<{
   [key in Language]: Partial<PromptsCustomText>;
 }>;
 
+const customPromptsPromptTypes = [
+  'login',
+  'login-id',
+  'login-password',
+  'signup',
+  'signup-id',
+  'signup-password',
+] as const;
+
+export type CustomPromptsPromptTypes = typeof customPromptsPromptTypes[number];
+
+const customPromptsScreenTypes = [
+  'login',
+  'login-id',
+  'login-password',
+  'signup',
+  'signup-id',
+  'signup-password',
+] as const;
+
+export type CustomPromptsScreenTypes = typeof customPromptsPromptTypes[number];
+
+export type CustomPromptsInsertionPoints =
+  | 'form-content-start'
+  | 'form-content-end'
+  | 'form-footer-start'
+  | 'form-footer-end'
+  | 'secondary-actions-start'
+  | 'secondary-actions-end';
+
+
+export type CustomPromptPartialsScreens = Partial<{
+  [screen in CustomPromptsScreenTypes]: Partial<{
+    [insertionPoint in CustomPromptsInsertionPoints]: string;
+  }>;
+}>;
+
+export type CustomPromptPartials = Partial<{
+  [prompt in CustomPromptsPromptTypes]: CustomPromptPartialsScreens;
+}>;
+
+export type CustomPromptsConfig = {
+  [prompt in CustomPromptsPromptTypes]: [
+    {
+      name: string;
+      template: string;
+    }
+  ];
+};
+
 export default class PromptsHandler extends DefaultHandler {
   existing: Prompts;
 
@@ -179,9 +230,12 @@ export default class PromptsHandler extends DefaultHandler {
 
     const customText = await this.getCustomTextSettings();
 
+    const partials = await this.getCustomPromptsPartials();
+
     return {
       ...promptsSettings,
       customText,
+      partials,
     };
   }
 
@@ -236,19 +290,57 @@ export default class PromptsHandler extends DefaultHandler {
       });
   }
 
+  async getCustomPromptsPartials(): Promise<CustomPromptPartials> {
+    return this.client.pool
+      .addEachTask({
+        data: customPromptsPromptTypes.map((promptType) => ({ promptType })),
+        generator: ({ promptType }) =>
+          this.client.prompts
+            .getPartials({
+              prompt: promptType,
+            })
+            .then((partialsData: CustomPromptPartials) => {
+              if (isEmpty(partialsData)) return null;
+              return {
+                [promptType]: {
+                  ...partialsData,
+                },
+              };
+            }),
+      })
+      .promise()
+      .then((partialsDataWithNulls) =>
+        partialsDataWithNulls
+          .filter(Boolean)
+          .reduce(
+            (
+              acc: CustomPromptPartials,
+              partialsData: { [prompt: string]: CustomPromptPartials }
+            ) => {
+              const [promptName] = Object.keys(partialsData);
+              acc[promptName] = partialsData[promptName];
+              return acc;
+            },
+            {}
+          )
+      );
+  }
+
   async processChanges(assets: Assets): Promise<void> {
     const { prompts } = assets;
 
     if (!prompts) return;
 
-    const { customText, ...promptSettings } = prompts;
+    const { partials, customText, ...promptSettings } = prompts;
 
     if (!isEmpty(promptSettings)) {
       await this.client.prompts.updateSettings({}, promptSettings);
     }
 
     await this.updateCustomTextSettings(customText);
 
+    await this.updateCustomPromptsPartials(partials);
+
     this.updated += 1;
     this.didUpdate(prompts);
   }
@@ -284,4 +376,24 @@ export default class PromptsHandler extends DefaultHandler {
       })
       .promise();
   }
+
+  async updateCustomPromptsPartials(partials: Prompts['partials']): Promise<void> {
+    /*
+      Note: deletes are not currently supported
+    */
+    if (!partials) return;
+
+    await this.client.pool
+      .addEachTask({
+        data: Object.keys(partials).map((prompt: CustomPromptsPromptTypes) => {
+          const body = partials[prompt] || {};
+          return {
+            body,
+            prompt,
+          };
+        }),
+        generator: ({ prompt, body }) => this.client.prompts.updatePartials({ prompt }, body),
+      })
+      .promise();
+  }
 }

src/types.ts

@@ -5,6 +5,9 @@ import {
   Prompts,
   PromptsCustomText,
   PromptSettings,
+  CustomPromptPartials,
+  CustomPromptsPromptTypes,
+  CustomPromptPartialsScreens,
 } from './tools/auth0/handlers/prompts';
 import { Tenant } from './tools/auth0/handlers/tenant';
 import { Theme } from './tools/auth0/handlers/themes';
@@ -159,6 +162,14 @@ export type BaseAuth0APIClient = {
       prompt: PromptTypes;
       language: Language;
     }) => Promise<Partial<PromptsCustomText>>;
+    updatePartials: (arg0: {
+      prompt: CustomPromptsPromptTypes;
+    },
+      body: CustomPromptPartialsScreens
+    ) => Promise<void>;
+    getPartials: (arg0: {
+      prompt: CustomPromptsPromptTypes;
+    }) => Promise<CustomPromptPartials>;
     getSettings: () => Promise<PromptSettings>;
     updateSettings: (arg0: {}, arg1: Partial<PromptSettings>) => Promise<void>;
   };