@xuda.io/ai_module 1.1.4772 → 1.1.4774

package/index.mjs

@@ -16,6 +16,7 @@ import { mdToPdf } from 'md-to-pdf';
 import _ from 'lodash';
 
 import { NodeVM, VMScript } from 'vm2';
+import admin from 'firebase-admin';
 
 puppeteer.use(StealthPlugin());
 import process from 'process';
@@ -28,6 +29,11 @@ global._conf = (
   })
 ).default;
 
+const _defaultAppConfig = {
+  credential: admin.credential.cert(_conf.firebase_serviceAccount),
+};
+admin.initializeApp(_defaultAppConfig);
+
 const module_path = path.join(process.env.XUDA_HOME, 'cpi') + (!_conf.is_debug ? '/node_modules/@xuda.io' : '');
 const runtime_modules_path = path.join(process.env.XUDA_HOME, 'dist', 'runtime', 'js', 'modules');
 
@@ -6943,6 +6949,8 @@ export const get_profile_picture = async function (uid, account_type = 'business
   const uniqueId = Date.now() + '_' + Math.random().toString(36).substring(7);
   const tempOutputPath = path.join(tempDir, `output_${uniqueId}.png`);
 
+  const { email } = metadata;
+
   let prompt,
     url,
     factual_prompt,
@@ -6986,6 +6994,32 @@ export const get_profile_picture = async function (uid, account_type = 'business
       break;
   }
 
+  const getUserPhotoByEmail = async function (email) {
+    try {
+      const auth = admin.auth();
+      const userRecord = await auth.getUserByEmail(email);
+
+      // photoURL is the profile picture URL (or null if not set)
+      const photoURL = userRecord.photoURL;
+      // console.log(`User found: ${JSON.stringify(userRecord)}`);
+
+      // console.log(`User found: ${userRecord.uid}`);
+      // console.log(`Display name: ${userRecord.displayName || 'None'}`);
+      // console.log(`Email verified: ${userRecord.emailVerified}`);
+      // console.log(`Profile picture URL: ${photoURL || 'No photo set'}`);
+
+      return photoURL;
+    } catch (error) {
+      if (error.code === 'auth/user-not-found') {
+        console.log(`No user found with email: ${email}`);
+        return null;
+      }
+
+      console.error('Error fetching user:', error);
+      // throw error;
+    }
+  };
+
   const get_url = async function () {
     const ret = await submit_chat_gpt_prompt({
       uid,
@@ -7001,6 +7035,8 @@ export const get_profile_picture = async function (uid, account_type = 'business
 
   // debugger;
   try {
+    url = await getUserPhotoByEmail(email);
+
     // first attempt
     if (account_type === 'business') {
       url = await get_url();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@xuda.io/ai_module",
-  "version": "1.1.4772",
+  "version": "1.1.4774",
   "description": "Xuda AI Module",
   "main": "index.mjs",
   "type": "module",