tango-app-api-audio-analytics 1.0.1 → 1.0.2

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "tango-app-api-audio-analytics",
-    "version": "1.0.1",
+    "version": "1.0.2",
     "description": "audioAnalytics",
     "main": "index.js",
     "type": "module",

package/src/controllers/audioAnalytics.controller.js

@@ -1,4 +1,4 @@
-import { insertWithId, logger } from 'tango-app-api-middleware';
+import { insertWithId, updateOpenSearchData, searchOpenSearchData, logger } from 'tango-app-api-middleware';
 import { randomUUID } from 'crypto';
 
 const EXTERNAL_STREAM_API = 'http://172.236.179.51:8000/stream';
@@ -44,6 +44,110 @@ export async function createCohort( req, res ) {
   }
 }
 
+export async function updateCohort( req, res ) {
+  try {
+    const { cohortId, ...updateFields } = req.body;
+
+    const metrics = updateFields.metrics ?
+      updateFields.metrics.map( ( metric ) => ( {
+        ...metric,
+        metricId: metric.metricId || randomUUID(),
+        hasContext: Array.isArray( metric.contexts ) && metric.contexts.length > 0,
+        isNumeric: metric.isNumeric ?? false,
+      } ) ) :
+      undefined;
+
+    const updateData = {
+      ...( updateFields.cohortName && { cohortName: updateFields.cohortName } ),
+      ...( updateFields.cohortDescription && { cohortDescription: updateFields.cohortDescription } ),
+      ...( updateFields.source && { source: updateFields.source } ),
+      ...( metrics && { metrics } ),
+      updatedAt: new Date().toISOString(),
+    };
+
+    const result = await updateOpenSearchData( 'tango-audio-config', cohortId, updateData );
+    logger.info( { result } );
+
+    if ( result && result.body && result.body.result === 'updated' ) {
+      return res.sendSuccess( { result: 'Cohort updated successfully', cohortId } );
+    } else {
+      return res.sendError( 'Failed to update cohort', 500 );
+    }
+  } catch ( error ) {
+    const err = error.message || 'Internal Server Error';
+    logger.error( { error, message: req.body, function: 'updateCohort' } );
+    return res.sendError( err, 500 );
+  }
+}
+
+export async function getCohort( req, res ) {
+  try {
+    const { cohortId } = req.params;
+
+    const query = {
+      query: {
+        match: { cohortId },
+      },
+    };
+
+    const result = await searchOpenSearchData( 'tango-audio-config', query );
+    logger.info( { result } );
+
+    if ( !result || result?.hits?.hits?.length === 0 ) {
+      return res.sendError( 'Cohort not found', 404 );
+    }
+
+    return res.sendSuccess( result?.hits?.hits?.[0]?._source );
+  } catch ( error ) {
+    const err = error.message || 'Internal Server Error';
+    logger.error( { error, message: req.params, function: 'getCohort' } );
+    return res.sendError( err, 500 );
+  }
+}
+
+export async function listCohortsByClient( req, res ) {
+  try {
+    const { clientId, limit = 10, offset = 0 } = req.query;
+    logger.info( { message: req.query, function: 'listCohortsByClient' } );
+    const query = {
+      query: {
+        match: { clientId },
+      },
+      size: Number( limit ),
+      from: Number( offset ),
+      sort: [ { createdAt: { order: 'desc' } } ],
+    };
+
+    const result = await searchOpenSearchData( 'tango-audio-config', query );
+    logger.info( { result } );
+
+    const total = result?.body?.hits?.total?.value || 0;
+    const cohorts = ( result?.body?.hits?.hits || [] ).map( ( hit ) => {
+      const source = hit._source;
+      return {
+        cohortId: source.cohortId,
+        cohortName: source.cohortName,
+        cohortDescription: source.cohortDescription,
+        metricsAdded: source.metrics?.length || 0,
+      };
+    } );
+
+    return res.sendSuccess( {
+      cohorts,
+      pagination: {
+        total,
+        limit: Number( limit ),
+        offset: Number( offset ),
+        hasMore: Number( offset ) + Number( limit ) < total,
+      },
+    } );
+  } catch ( error ) {
+    const err = error.message || 'Internal Server Error';
+    logger.error( { error, message: req.query, function: 'listCohortsByClient' } );
+    return res.sendError( err, 500 );
+  }
+}
+
 export const callExternalStreamAPI = async ( req, res ) => {
   try {
     const { prompt, storeId, productModule, country, region, fromDate, toDate, clusters } = req.body;

package/src/dtos/audioAnalytics.dtos.js

@@ -536,3 +536,33 @@ export const createCohortSchema = joi.object( {
 export const createCohortValid = {
   body: createCohortSchema,
 };
+
+const updateCohortSchema = joi.object( {
+  cohortId: joi.string().required(),
+  cohortName: joi.string().optional(),
+  cohortDescription: joi.string().optional(),
+  source: joi.array().optional(),
+  metrics: joi.array().items( cohortMetricSchema ).optional(),
+} );
+
+export const updateCohortValid = {
+  body: updateCohortSchema,
+};
+
+const getCohortSchema = joi.object( {
+  cohortId: joi.string().required(),
+} );
+
+export const getCohortValid = {
+  params: getCohortSchema,
+};
+
+const listCohortsByClientSchema = joi.object( {
+  clientId: joi.string().required(),
+  limit: joi.number().integer().min( 1 ).max( 100 ).optional().default( 10 ),
+  offset: joi.number().integer().min( 0 ).optional().default( 0 ),
+} );
+
+export const listCohortsByClientValid = {
+  query: listCohortsByClientSchema,
+};

package/src/routes/audioAnalytics.routes.js

@@ -1,8 +1,8 @@
 
 import express from 'express';
 import { validate } from 'tango-app-api-middleware';
-import { createCohortValid } from '../dtos/audioAnalytics.dtos.js';
-import { createCohort } from '../controllers/audioAnalytics.controller.js';
+import { createCohortValid, updateCohortValid, getCohortValid, listCohortsByClientValid } from '../dtos/audioAnalytics.dtos.js';
+import { createCohort, updateCohort, getCohort, listCohortsByClient } from '../controllers/audioAnalytics.controller.js';
 
 export const audioAnalyticsrouter = express.Router(); ;
 
@@ -13,6 +13,9 @@ audioAnalyticsrouter.get( '/test', ( req, res ) => {
 } );
 
 audioAnalyticsrouter.post( '/create-cohort', validate( createCohortValid ), createCohort );
+audioAnalyticsrouter.post( '/update-cohort', validate( updateCohortValid ), updateCohort );
+audioAnalyticsrouter.get( '/get-cohort/:cohortId', validate( getCohortValid ), getCohort );
+audioAnalyticsrouter.get( '/list-cohorts', validate( listCohortsByClientValid ), listCohortsByClient );
 
 
 export default audioAnalyticsrouter;