build-stream-service.ts 2.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. // @ts-check
  2. import type { Span } from './trace';
  3. import { task } from './trace';
  4. import { ALL, NORTH_AMERICA, EU, HK, TW, JP, KR } from '../Source/stream';
  5. import { SHARED_DESCRIPTION } from './constants/description';
  6. import { RulesetOutput } from './lib/create-file';
  7. export function createRulesetForStreamService(span: Span,
  8. fileId: string, title: string,
  9. streamServices: Array<import('../Source/stream').StreamService>) {
  10. return span.traceChildAsync(fileId, async (childSpan) => Promise.all([
  11. // Domains
  12. new RulesetOutput(childSpan, fileId, 'non_ip')
  13. .withTitle(`Sukka's Ruleset - Stream Services: ${title}`)
  14. .withDescription([
  15. ...SHARED_DESCRIPTION,
  16. '',
  17. ...streamServices.map((i) => `- ${i.name}`)
  18. ])
  19. .addFromRuleset(streamServices.flatMap((i) => i.rules))
  20. .write(),
  21. // IP
  22. new RulesetOutput(childSpan, fileId, 'ip')
  23. .withTitle(`Sukka's Ruleset - Stream Services IPs: ${title}`)
  24. .withDescription([
  25. ...SHARED_DESCRIPTION,
  26. '',
  27. ...streamServices.map((i) => `- ${i.name}`)
  28. ])
  29. .bulkAddCIDR4NoResolve(streamServices.flatMap(i => i.ip?.v4 ?? []))
  30. .bulkAddCIDR6NoResolve(streamServices.flatMap(i => i.ip?.v6 ?? []))
  31. .write()
  32. ]));
  33. }
  34. export const buildStreamService = task(require.main === module, __filename)(async (span) => Promise.all([
  35. createRulesetForStreamService(span, 'stream', 'All', ALL),
  36. createRulesetForStreamService(span, 'stream_us', 'North America', NORTH_AMERICA),
  37. createRulesetForStreamService(span, 'stream_eu', 'Europe', EU),
  38. createRulesetForStreamService(span, 'stream_hk', 'Hong Kong', HK),
  39. createRulesetForStreamService(span, 'stream_tw', 'Taiwan', TW),
  40. createRulesetForStreamService(span, 'stream_jp', 'Japan', JP),
  41. // createRulesetForStreamService('stream_au', 'Oceania', AU),
  42. createRulesetForStreamService(span, 'stream_kr', 'Korean', KR)
  43. // createRulesetForStreamService('stream_south_east_asia', 'South East Asia', SOUTH_EAST_ASIA)
  44. ]));