Merge pull request #481 from TogetherCrew/mediawiki-changes

Mediawiki changes

Author: Behzad-rabiei

src/services/module.service.ts

@@ -140,23 +140,16 @@ const handleHivemindWebsiteCase = async (platform: any) => {
  * @param {Object} platform - Platform object
  */
 const handleHivemindMediaWikiCase = async (platform: any) => {
-  console.log('Handling Hivemind MediaWiki case for platform:', platform);
   const platformDoc = await platformService.getPlatformById(platform.platform);
 
   if (!platformDoc) return;
 
   const isActivated = platform.metadata?.activated;
-  const existingWorkflowId = platformDoc.get('metadata.workflowId');
 
   if (isActivated === true) {
-    console.log('Platform is activated, checking for existing workflow ID:', existingWorkflowId);
-    if (!existingWorkflowId) {
-      console.log('No existing workflow ID found, executing new workflow for platform:', platform.platform);
-      const workflowId = await temporalMediaWiki.executeWorkflow(platform.platform);
-      console.log('New workflow ID created:', workflowId);
-      platformDoc.set('metadata.workflowId', workflowId);
-      await platformDoc.save();
-    }
+    temporalMediaWiki.executeWorkflow(platformDoc.id);
+  } else if (isActivated === false) {
+    temporalMediaWiki.terminateWorkflow(platformDoc.id);
   }
 };
 

src/services/temporal/mediaWiki.service.ts

@@ -12,29 +12,26 @@ const logger = parentLogger.child({ module: 'MediaWikiTemporalService' });
 class TemporalMediaWikiService extends TemporalCoreService {
   public async executeWorkflow(platformId: Types.ObjectId) {
     const client: Client = await this.getClient();
-    const payload = {
-      platform_id: platformId,
-    };
+    const payload = platformId;
     try {
-      const workflowHandle = await client.workflow.execute('MediaWikiETLWorkflow', {
+      client.workflow.execute('MediaWikiETLWorkflow', {
         taskQueue: queues.TEMPORAL_QUEUE_PYTHON_HEAVY,
         args: [payload],
-        workflowId: `mediawiki/${platformId}/${uuidv4()}`,
+        workflowId: `api:mediawikietl:${platformId}`,
       });
-      logger.info(`Started MediaWiki workflow with ID: ${workflowHandle}`);
-      return workflowHandle;
     } catch (error) {
       logger.error(`Failed to trigger MediaWiki workflow: ${(error as Error).message}`);
       throw new Error(`Failed to trigger MediaWiki workflow: ${(error as Error).message}`);
     }
   }
 
-  public async terminateWorkflow(workflowId: string): Promise<void> {
+  public async terminateWorkflow(platformId: Types.ObjectId): Promise<void> {
     const client: Client = await this.getClient();
-    const handle = client.workflow.getHandle(workflowId);
-    const description = await handle.describe();
-    if (description.status.name !== 'TERMINATED' && description.status.name !== 'COMPLETED') {
-      await handle.terminate('Terminated due to schedule deletion');
+    try {
+      client.workflow.getHandle(`api:mediawikietl:${platformId}`).terminate();
+    } catch (error) {
+      logger.error(`Failed to terminate MediaWiki workflow: ${(error as Error).message}`);
+      throw new Error(`Failed to terminate MediaWiki workflow: ${(error as Error).message}`);
     }
   }
 }