SukkaSurge/Build/build-stream-service.ts

// @ts-check
import type { Span } from './trace';
import { task } from './trace';

import { ALL, NORTH_AMERICA, EU, HK, TW, JP, KR } from '../Source/stream';
import { SHARED_DESCRIPTION } from './lib/constants';
import { RulesetOutput } from './lib/create-file';

export const createRulesetForStreamService = (span: Span, fileId: string, title: string, streamServices: Array<import('../Source/stream').StreamService>) => {
  return span.traceChildAsync(fileId, async (childSpan) => Promise.all([
    // Domains
    new RulesetOutput(childSpan, fileId, 'non_ip')
      .withTitle(`Sukka's Ruleset - Stream Services: ${title}`)
      .withDescription([
        ...SHARED_DESCRIPTION,
        '',
        ...streamServices.map((i) => `- ${i.name}`)
      ])
      .addFromRuleset(streamServices.flatMap((i) => i.rules))
      .write(),
    // IP
    new RulesetOutput(childSpan, fileId, 'ip')
      .withTitle(`Sukka's Ruleset - Stream Services IPs: ${title}`)
      .withDescription([
        ...SHARED_DESCRIPTION,
        '',
        ...streamServices.map((i) => `- ${i.name}`)
      ])
      .bulkAddCIDR4NoResolve(streamServices.flatMap(i => i.ip?.v4 ?? []))
      .bulkAddCIDR6NoResolve(streamServices.flatMap(i => i.ip?.v6 ?? []))
      .write()
  ]));
};

export const buildStreamService = task(require.main === module, __filename)(async (span) => {
  return Promise.all([
    createRulesetForStreamService(span, 'stream', 'All', ALL),
    createRulesetForStreamService(span, 'stream_us', 'North America', NORTH_AMERICA),
    createRulesetForStreamService(span, 'stream_eu', 'Europe', EU),
    createRulesetForStreamService(span, 'stream_hk', 'Hong Kong', HK),
    createRulesetForStreamService(span, 'stream_tw', 'Taiwan', TW),
    createRulesetForStreamService(span, 'stream_jp', 'Japan', JP),
    // createRulesetForStreamService('stream_au', 'Oceania', AU),
    createRulesetForStreamService(span, 'stream_kr', 'Korean', KR)
    // createRulesetForStreamService('stream_south_east_asia', 'South East Asia', SOUTH_EAST_ASIA)
  ]);
});