feat: add modal for alignment and latency config (#34)

OwnZones · Oct 16, 2024 · f611fd7 · f611fd7
1 parent de2d760
commit f611fd7
Show file tree

Hide file tree

Showing 20 changed files with 1,151 additions and 59 deletions.
diff --git a/src/api/ateliereLive/ingest.ts b/src/api/ateliereLive/ingest.ts
@@ -1,6 +1,7 @@
 import {
   ResourcesCompactIngestResponse,
   ResourcesIngestResponse,
+  ResourcesIngestStreamResponse,
   ResourcesSourceResponse,
   ResourcesThumbnailResponse
 } from '../../../types/ateliere-live';
@@ -16,46 +17,50 @@ export async function getUuidFromIngestName(
   ingestName: string,
   useCache = true
 ) {
-  const cache = INGEST_UUID_CACHE.get(ingestName);
-  if (cache && useCache) {
-    return cache;
-  }
-  const ingests = await getIngests();
-  const ingest = ingests.find((ingest) => ingest.name === ingestName);
+  if (ingestName !== undefined) {
+    const cache = INGEST_UUID_CACHE.get(ingestName);
+    if (cache && useCache) {
+      return cache;
+    }
+    const ingests = await getIngests();
+    const ingest = ingests.find((ingest) => ingest.name === ingestName);
 
-  if (ingest && ingest.uuid) {
+    if (!ingest) {
+      console.warn(`Could not find ingest ${ingestName}`);
+      throw 'get_uuid';
+    }
     INGEST_UUID_CACHE.set(ingestName, ingest.uuid);
     return ingest.uuid;
   }
-  console.warn(`Could not find UUID for ${ingestName}`);
-  throw 'get_uuid';
 }
 
 export async function getSourceIdFromSourceName(
   ingestUuid: string,
   sourceName: string,
   useCache = true
 ) {
-  let ingestCache = SOURCE_ID_CACHE.get(ingestUuid);
-  if (!ingestCache) {
-    ingestCache = new Map();
-    SOURCE_ID_CACHE.set(ingestUuid, ingestCache);
-  }
-  const cache = ingestCache?.get(sourceName);
-  if (cache && useCache) {
-    return cache;
-  }
-  const ingest = await getIngest(ingestUuid);
-  const source = ingest.sources?.find((source) => source.name === sourceName);
+  if (ingestUuid !== undefined && sourceName !== undefined) {
+    let ingestCache = SOURCE_ID_CACHE.get(ingestUuid);
+    if (!ingestCache) {
+      ingestCache = new Map();
+      SOURCE_ID_CACHE.set(ingestUuid, ingestCache);
+    }
+    const cache = ingestCache?.get(sourceName);
+    if (cache && useCache) {
+      return cache;
+    }
+    const ingest = await getIngest(ingestUuid);
+    const source = ingest.sources?.find((source) => source.name === sourceName);
 
-  if (source && source.source_id !== undefined) {
-    ingestCache.set(sourceName, source.source_id);
-    return source.source_id;
+    if (source && source.source_id !== undefined) {
+      ingestCache.set(sourceName, source.source_id);
+      return source.source_id;
+    }
+    console.warn(
+      `Could not find id for source ${sourceName} in ingest ${ingestUuid}`
+    );
+    throw `Could not find id for source ${sourceName} in ingest ${ingestUuid}`;
   }
-  console.warn(
-    `Could not find id for source ${sourceName} in ingest ${ingestUuid}`
-  );
-  throw `Could not find id for source ${sourceName} in ingest ${ingestUuid}`;
 }
 
 export async function getIngests(): Promise<ResourcesCompactIngestResponse[]> {
@@ -205,3 +210,24 @@ export async function getIngestSources(
   const errorText = await response.text();
   throw new Error(errorText);
 }
+
+export async function getIngestStreams(
+  ingestUuid: string
+): Promise<ResourcesIngestStreamResponse[]> {
+  const response = await fetch(
+    new URL(
+      LIVE_BASE_API_PATH + `/ingests/${ingestUuid}/streams?expand=true`,
+      process.env.LIVE_URL
+    ),
+    {
+      method: 'GET',
+      headers: {
+        authorization: getAuthorizationHeader()
+      }
+    }
+  );
+  if (response.ok) {
+    return response.json();
+  }
+  throw await response.text();
+}
diff --git a/src/api/ateliereLive/pipelines/streams/streams.ts b/src/api/ateliereLive/pipelines/streams/streams.ts
@@ -76,7 +76,7 @@ export async function createStream(
     });
 
     const sourceId = await getSourceIdFromSourceName(
-      ingestUuid,
+      ingestUuid || '',
       source.ingest_source_name,
       false
     );
@@ -108,8 +108,8 @@ export async function createStream(
       );
 
       const stream: PipelineStreamSettings = {
-        ingest_id: ingestUuid,
-        source_id: sourceId,
+        ingest_id: ingestUuid || '',
+        source_id: sourceId || 0,
         pipeline_id: pipeline.pipeline_id!,
         input_slot: input_slot,
         alignment_ms: pipeline.alignment_ms,
@@ -357,3 +357,26 @@ export async function deleteStream(streamUuid: string) {
   }
   throw await response.json();
 }
+
+export async function updateStream(streamUuid: string, alignment_ms: number) {
+  const response = await fetch(
+    new URL(
+      LIVE_BASE_API_PATH + `/streams/${streamUuid}`,
+      process.env.LIVE_URL
+    ),
+    {
+      method: 'PATCH',
+      headers: {
+        authorization: getAuthorizationHeader()
+      },
+      body: JSON.stringify({ alignment_ms: alignment_ms }),
+      next: {
+        revalidate: 0
+      }
+    }
+  );
+  if (response.ok) {
+    return true;
+  }
+  throw await response.json();
+}
diff --git a/src/api/manager/productions.ts b/src/api/manager/productions.ts
@@ -91,3 +91,161 @@ export async function deleteProduction(id: string): Promise<void> {
 function deleteMonitoring(db: Db, productionId: string) {
   db.collection('monitoring').deleteMany({ productionId: productionId });
 }
+
+export async function getProductionPipelineSourceAlignment(
+  productionId: string,
+  pipelineId: string,
+  sourceId: number
+) {
+  const production = await getProduction(productionId);
+
+  if (!production) {
+    console.error('Production not found');
+    return null;
+  }
+
+  const pipeline = production.production_settings.pipelines.find(
+    (p) => p.pipeline_id === pipelineId
+  );
+
+  if (!pipeline) {
+    console.error('Pipeline not found');
+    return null;
+  }
+
+  const source = pipeline?.sources?.find(
+    (source) => String(source.source_id) === String(sourceId)
+  );
+
+  if (!source) {
+    console.error('Source not found');
+    return null;
+  }
+
+  const alignment =
+    source?.settings?.alignment_ms !== undefined
+      ? source.settings.alignment_ms
+      : pipeline?.alignment_ms;
+
+  return alignment;
+}
+
+export async function setProductionPipelineSourceAlignment(
+  productionId: string,
+  pipelineId: string,
+  sourceId: number,
+  alignment_ms: number
+) {
+  const db = await getDatabase();
+
+  try {
+    const result = await db.collection('productions').updateOne(
+      {
+        _id: new ObjectId(productionId),
+        'production_settings.pipelines.pipeline_id': pipelineId,
+        'production_settings.pipelines.sources.source_id': sourceId
+      },
+      {
+        $set: {
+          'production_settings.pipelines.$[p].sources.$[s].settings.alignment_ms':
+            alignment_ms
+        }
+      },
+      {
+        arrayFilters: [
+          { 'p.pipeline_id': pipelineId },
+          { 's.source_id': sourceId }
+        ]
+      }
+    );
+
+    if (result.matchedCount === 0) {
+      console.error('No matching pipeline or source found to update');
+      return null;
+    }
+
+    return true;
+  } catch (error) {
+    console.error('Database error:', error);
+    throw new Error('Error updating pipeline source alignment');
+  }
+}
+
+export async function getProductionSourceLatency(
+  productionId: string,
+  pipelineId: string,
+  sourceId: number
+) {
+  const production = await getProduction(productionId);
+
+  if (!production) {
+    console.error('Production not found');
+    return null;
+  }
+
+  const pipeline = production.production_settings.pipelines.find(
+    (p) => p.pipeline_id === pipelineId
+  );
+
+  if (!pipeline) {
+    console.error('Pipeline not found');
+    return null;
+  }
+
+  const source = pipeline?.sources?.find(
+    (source) => String(source.source_id) === String(sourceId)
+  );
+
+  if (!source) {
+    console.error('Source not found');
+    return null;
+  }
+
+  const latency =
+    source?.settings?.max_network_latency_ms !== undefined
+      ? source.settings.max_network_latency_ms
+      : pipeline?.max_network_latency_ms;
+
+  return latency;
+}
+
+export async function setProductionPipelineSourceLatency(
+  productionId: string,
+  pipelineId: string,
+  sourceId: number,
+  max_network_latency_ms: number
+) {
+  const db = await getDatabase();
+
+  try {
+    const result = await db.collection('productions').updateOne(
+      {
+        _id: new ObjectId(productionId),
+        'production_settings.pipelines.pipeline_id': pipelineId,
+        'production_settings.pipelines.sources.source_id': sourceId
+      },
+      {
+        $set: {
+          'production_settings.pipelines.$[p].sources.$[s].settings.max_network_latency_ms':
+            max_network_latency_ms
+        }
+      },
+      {
+        arrayFilters: [
+          { 'p.pipeline_id': pipelineId },
+          { 's.source_id': sourceId }
+        ]
+      }
+    );
+
+    if (result.matchedCount === 0) {
+      console.error('No matching pipeline or source found to update');
+      return null;
+    }
+
+    return true;
+  } catch (error) {
+    console.error('Database error:', error);
+    throw new Error('Error updating pipeline source latency');
+  }
+}