Updates

package-shared/functions/api/users/api-send-email-code.d.ts

@@ -1,4 +1,4 @@
-declare function _exports({ email, database, email_login_field, mail_domain, mail_port, sender, mail_username, mail_password, html, useLocal, response, }: {
+declare function _exports({ email, database, email_login_field, mail_domain, mail_port, sender, mail_username, mail_password, html, useLocal, response, extraCookies, }: {
     email: string;
     database: string;
     email_login_field?: string;
@@ -12,6 +12,7 @@ declare function _exports({ email, database, email_login_field, mail_domain, mai
     response?: http.ServerResponse & {
         [x: string]: any;
     };
+    extraCookies?: import("../../../../package-shared/types").CookieObject[];
 }): Promise<import("../../../types").SendOneTimeCodeEmailResponse>;
 export = _exports;
 import http = require("http");

package-shared/functions/api/users/api-send-email-code.js

@@ -1,11 +1,11 @@
 // @ts-check
 
-const LOCAL_DB_HANDLER = require("../../../utils/backend/global-db/LOCAL_DB_HANDLER");
 const varDatabaseDbHandler = require("../../backend/varDatabaseDbHandler");
 const nodemailer = require("nodemailer");
 const http = require("http");
 const getAuthCookieNames = require("../../backend/cookies/get-auth-cookie-names");
 const encrypt = require("../../dsql/encrypt");
+const serializeCookies = require("../../../utils/serialize-cookies");
 
 /**
  * # Send Email Login Code
@@ -22,6 +22,7 @@ const encrypt = require("../../dsql/encrypt");
  * @param {string} param.html
  * @param {boolean} [param.useLocal]
  * @param {http.ServerResponse & Object<string,any>} [param.response]
+ * @param {import("../../../../package-shared/types").CookieObject[]} [param.extraCookies]
  *
  * @returns {Promise<import("../../../types").SendOneTimeCodeEmailResponse>}
  */
@@ -37,6 +38,7 @@ module.exports = async function apiSendEmailCode({
     html,
     useLocal,
     response,
+    extraCookies,
 }) {
     if (email?.match(/ /)) {
         return {
@@ -123,16 +125,39 @@ module.exports = async function apiSendEmailCode({
         };
 
         if (response) {
-            const keyNames = getAuthCookieNames();
-            const oneTimeCodeCookieName = keyNames.oneTimeCodeName;
+            const cookieKeyNames = getAuthCookieNames();
+            const oneTimeCodeCookieName = cookieKeyNames.oneTimeCodeName;
 
             const encryptedPayload = encrypt({
                 data: JSON.stringify(resObject),
             });
 
-            response?.setHeader("Set-Cookie", [
-                `${oneTimeCodeCookieName}=${encryptedPayload};samesite=strict;path=/;HttpOnly=true;Secure=true`,
-            ]);
+            if (!encryptedPayload) {
+                throw new Error(
+                    "apiSendEmailCode Error: Failed to encrypt payload"
+                );
+            }
+
+            /** @type {import("../../../../package-shared/types").CookieObject} */
+            const oneTimeCookieObject = {
+                name: oneTimeCodeCookieName,
+                value: encryptedPayload,
+                sameSite: "Strict",
+                path: "/",
+                httpOnly: true,
+                secure: true,
+            };
+
+            /** @type {import("../../../../package-shared/types").CookieObject[]} */
+            const cookiesObjectArray = extraCookies
+                ? [...extraCookies, oneTimeCookieObject]
+                : [oneTimeCookieObject];
+
+            const serializedCookies = serializeCookies({
+                cookies: cookiesObjectArray,
+            });
+
+            response.setHeader("Set-Cookie", serializedCookies);
         }
 
         return resObject;

package-shared/types/index.d.ts

@@ -1199,4 +1199,16 @@ export type SendOneTimeCodeEmailResponse = {
     email?: string;
     msg?: string;
 };
+export type CookieObject = {
+    name: string;
+    value: string;
+    domain?: string;
+    path?: string;
+    expires?: Date;
+    maxAge?: number;
+    secure?: boolean;
+    httpOnly?: boolean;
+    sameSite?: "Strict" | "Lax" | "None";
+    priority?: "Low" | "Medium" | "High";
+};
 export {};

package-shared/types/index.ts

@@ -1421,3 +1421,16 @@ export type SendOneTimeCodeEmailResponse = {
     email?: string;
     msg?: string;
 };
+
+export type CookieObject = {
+    name: string;
+    value: string;
+    domain?: string;
+    path?: string;
+    expires?: Date;
+    maxAge?: number;
+    secure?: boolean;
+    httpOnly?: boolean;
+    sameSite?: "Strict" | "Lax" | "None";
+    priority?: "Low" | "Medium" | "High";
+};

package-shared/utils/serialize-cookies.d.ts

@@ -0,0 +1,10 @@
+export = serializeCookies;
+/**
+ *
+ * @param {object} params
+ * @param {import("../types").CookieObject[]} params.cookies
+ * @returns {string[]}
+ */
+declare function serializeCookies({ cookies }: {
+    cookies: import("../types").CookieObject[];
+}): string[];

package-shared/utils/serialize-cookies.js

@@ -0,0 +1,48 @@
+// @ts-check
+
+/**
+ *
+ * @param {object} params
+ * @param {import("../types").CookieObject[]} params.cookies
+ * @returns {string[]}
+ */
+function serializeCookies({ cookies }) {
+    /** @type {string[]} */
+    let cookiesStringsArray = [];
+
+    for (let i = 0; i < cookies.length; i++) {
+        const cookieObject = cookies[i];
+        let cookieString = `${cookieObject.name}=${cookieObject.value}`;
+
+        if (cookieObject.maxAge) {
+            cookieString += `;Max-Age=${cookieObject.maxAge}`;
+        }
+        if (cookieObject.path) {
+            cookieString += `;Path=${cookieObject.path}`;
+        }
+        if (cookieObject.domain) {
+            cookieString += `;Domain=${cookieObject.domain}`;
+        }
+        if (cookieObject.secure) {
+            cookieString += ";Secure";
+        }
+        if (cookieObject.httpOnly) {
+            cookieString += ";HttpOnly";
+        }
+        if (cookieObject.sameSite) {
+            cookieString += `;SameSite=${cookieObject.sameSite}`;
+        }
+        if (cookieObject.expires) {
+            cookieString += `;expires=${cookieObject.expires}`;
+        }
+        if (cookieObject.priority) {
+            cookieString += `;priority=${cookieObject.priority}`;
+        }
+
+        cookiesStringsArray.push(cookieString);
+    }
+
+    return cookiesStringsArray;
+}
+
+module.exports = serializeCookies;