Refactor getExportScheduleDetails to include workbookId

Author: pballai

embedding_qs_series_2_api_use_cases/helpers/get-export-schedule-details.js

@@ -1,21 +1,21 @@
 require("dotenv").config();
 const axios = require("axios");
 const getBearerToken = require("./get-access-token");
-
 const baseURL = process.env.BASE_URL;
 
 /**
  * Get detailed export schedule information including recipients
- * @param {string} scheduleId - The schedule notification ID
+ * @param {string} workbookId - The workbook ID
+ * @param {string} scheduleId - The schedule notification ID (optional - used to filter results)
  * @returns {Promise<Object>} The detailed schedule data including recipients
  */
-async function getExportScheduleDetails(scheduleId) {
-  if (!scheduleId) {
-    throw new Error('Schedule ID is required');
+async function getExportScheduleDetails(workbookId, scheduleId = null) {
+  if (!workbookId) {
+    throw new Error('Workbook ID is required');
   }
 
   if (process.env.DEBUG === "true") {
-    console.log(`Fetching detailed schedule information for ID: ${scheduleId}`);
+    console.log(`Fetching schedule information for workbook: ${workbookId}`);
   }
 
   const bearerToken = await getBearerToken();
@@ -25,7 +25,7 @@ async function getExportScheduleDetails(scheduleId) {
 
   try {
     const response = await axios.get(
-      `${baseURL}/schedules/${scheduleId}`,
+      `${baseURL}/v2/workbooks/${workbookId}/schedules`,
       {
         headers: {
           "Authorization": `Bearer ${bearerToken}`,
@@ -38,16 +38,30 @@ async function getExportScheduleDetails(scheduleId) {
       throw new Error('No schedule data returned from API');
     }
 
-    const schedule = response.data;
+    let schedule = response.data;
+    
+    // If scheduleId is provided, filter to find the specific schedule
+    if (scheduleId) {
+      if (Array.isArray(schedule)) {
+        schedule = schedule.find(s => s.scheduleId === scheduleId || s.id === scheduleId);
+        if (!schedule) {
+          throw new Error(`Schedule with ID ${scheduleId} not found in workbook ${workbookId}`);
+        }
+      }
+    }
 
     if (process.env.DEBUG === "true") {
-      console.log(`Successfully fetched schedule details for: ${schedule.configV2?.title || scheduleId}`);
-      
-      // Log recipient information if available
-      if (schedule.target && schedule.target.length > 0) {
-        console.log(`Found ${schedule.target.length} recipients:`, schedule.target.map(t => t.email));
+      if (Array.isArray(schedule)) {
+        console.log(`Successfully fetched ${schedule.length} schedules for workbook`);
       } else {
-        console.log('No recipients found in schedule details');
+        console.log(`Successfully fetched schedule details for: ${schedule.configV2?.title || scheduleId}`);
+        
+        // Log recipient information if available
+        if (schedule.target && schedule.target.length > 0) {
+          console.log(`Found ${schedule.target.length} recipients:`, schedule.target.map(t => t.email));
+        } else {
+          console.log('No recipients found in schedule details');
+        }
       }
     }
 
@@ -56,13 +70,11 @@ async function getExportScheduleDetails(scheduleId) {
     const errorMsg = error.response?.data?.message || 
                     error.response?.data?.error || 
                     error.message;
-
     if (process.env.DEBUG === "true") {
-      console.error(`Error fetching schedule details for ${scheduleId}:`, errorMsg);
+      console.error(`Error fetching schedule details for workbook ${workbookId}:`, errorMsg);
     }
-
     throw new Error(`Failed to get schedule details: ${errorMsg}`);
   }
 }
 
-module.exports = getExportScheduleDetails;
+module.exports = getExportScheduleDetails;