Computed Fields Scripts API in Siren Investigate

  context.registerComputations({
    mainComputation: {
      compute: async (records, computeOptions) => {
        const firstFieldName = computeOptions.computedFields[0].name; // 'computed-id'
        for (const record of records) {
          const uuid = await record.getFirstRawFieldValue('generatedUUID');
          await context.setComputedField(record, firstFieldName, `${firstFieldName}:${uuid}`);
        }
      },
      outputFields: ['computed-id'],
      dependentFields: ['generatedUUID'] // computed field from a previously executed script
    }
  });

  const OUTPUT_FIELDS = ['my_field'];

  const getRelatedArticlesByCompanyId = async (companyIds: string[]) => {
    const alias = 'company_id';
    const scriptSource = `if (doc.containsKey('${alias}') && !doc['${alias}'].empty) { return doc['${alias}'].value; }`

    const scriptFields = {
      projected_company: {
        script: {
          source: scriptSource
        }
      }
    };

    const joinQuery = {
      join: {
        indices: ['company'],
        on: ['companies', 'id'],
        request: {
          project: [
            {
              field: {
                name: 'id',
                alias
              }
            }
          ],
          query: {
            terms: {
              id: companyIds
            }
          }
        }
      }
    };

    const response = await sirenapi.elasticsearch.post('/article/_search', {
      size: 10000,
      _source: false,
      script_fields: scriptFields,
      query: joinQuery
    });

    const groupedByCompanyId = response.hits.hits.reduce((acc, hit) => {
      const companyId = hit.fields?.projected_company?.[0];
      if (acc[companyId]) {
        acc[companyId].push(hit);
      } else {
        acc[companyId] = [hit];
      }
      return acc;
    }, {});

    return groupedByCompanyId;
  };

  context.registerComputations({
    mainComputation: {
      compute: async (records, options) => {
        const companyIds = await Promise.all(records.map(record => record.getFirstRawFieldValue('id')));
        const articlesGroupedByCompanyId = await getRelatedArticlesByCompanyId(companyIds);

        for (const record of records) {
          for (const { name } of options.computedFields) {
            const companyId = await record.getFirstRawFieldValue('id');
            const relatedArticles = articlesGroupedByCompanyId[companyId] || [];
            const relatedArticleIds = relatedArticles.map(hit => `${hit._index}/_doc/${hit._id}`);
            await context.setComputedField(record, name, relatedArticleIds);
          }
        }
      },
      outputFields: OUTPUT_FIELDS
    }
  });