Refactor: adjust output

2026-01-29 01:51:52 +08:00 · 2024-09-21 18:20:24 +08:00
parent 1783cccf7f
commit 7c372b3b8c
8 changed files with 627 additions and 662 deletions
--- a/Build/lib/rules/ruleset.ts
+++ b/Build/lib/rules/ruleset.ts
@@ -0,0 +1,112 @@
+import type { Span } from '../../trace';
+import createKeywordFilter from '../aho-corasick';
+import { appendArrayInPlace } from '../append-array-in-place';
+import { appendArrayFromSet } from '../misc';
+import type { SingboxSourceFormat } from '../singbox';
+import { sortDomains } from '../stable-sort-domain';
+import { RuleOutput } from './base';
+
+export class RulesetOutput extends RuleOutput {
+  constructor(span: Span, id: string, protected type: 'non_ip' | 'ip') {
+    super(span, id);
+  }
+
+  private $computed: [domain: string[], domainSuffix: string[], sortedDomainRules: string[]] | null = null;
+  private get computed() {
+    if (!this.$computed) {
+      const kwfilter = createKeywordFilter(this.domainKeywords);
+
+      const domains: string[] = [];
+      const domainSuffixes: string[] = [];
+      const sortedDomainRules: string[] = [];
+
+      for (const domain of sortDomains(this.domainTrie.dump(), this.apexDomainMap, this.subDomainMap)) {
+        if (kwfilter(domain)) {
+          continue;
+        }
+        if (domain[0] === '.') {
+          domainSuffixes.push(domain.slice(1));
+          sortedDomainRules.push(`DOMAIN-SUFFIX,${domain.slice(1)}`);
+        } else {
+          domains.push(domain);
+          sortedDomainRules.push(`DOMAIN,${domain}`);
+        }
+      }
+
+      this.$computed = [domains, domainSuffixes, sortedDomainRules];
+    }
+    return this.$computed;
+  }
+
+  surge(): string[] {
+    const results: string[] = ['DOMAIN,this_ruleset_is_made_by_sukkaw.ruleset.skk.moe'];
+    appendArrayInPlace(results, this.computed[2]);
+
+    appendArrayFromSet(results, this.domainKeywords, i => `DOMAIN-KEYWORD,${i}`);
+    appendArrayFromSet(results, this.domainWildcard, i => `DOMAIN-WILDCARD,${i}`);
+
+    appendArrayFromSet(results, this.userAgent, i => `USER-AGENT,${i}`);
+
+    appendArrayFromSet(results, this.processName, i => `PROCESS-NAME,${i}`);
+    appendArrayFromSet(results, this.processPath, i => `PROCESS-NAME,${i}`);
+
+    appendArrayInPlace(results, this.otherRules);
+
+    appendArrayFromSet(results, this.urlRegex, i => `URL-REGEX,${i}`);
+
+    appendArrayFromSet(results, this.ipcidrNoResolve, i => `IP-CIDR,${i},no-resolve`);
+    appendArrayFromSet(results, this.ipcidr6NoResolve, i => `IP-CIDR6,${i},no-resolve`);
+    appendArrayFromSet(results, this.ipasnNoResolve, i => `IP-ASN,${i},no-resolve`);
+    appendArrayFromSet(results, this.groipNoResolve, i => `GEOIP,${i},no-resolve`);
+
+    appendArrayFromSet(results, this.ipcidr, i => `IP-CIDR,${i}`);
+    appendArrayFromSet(results, this.ipcidr6, i => `IP-CIDR6,${i}`);
+    appendArrayFromSet(results, this.ipasn, i => `IP-ASN,${i}`);
+    appendArrayFromSet(results, this.geoip, i => `GEOIP,${i}`);
+
+    return results;
+  }
+
+  clash(): string[] {
+    const results: string[] = ['DOMAIN,this_ruleset_is_made_by_sukkaw.ruleset.skk.moe'];
+
+    appendArrayInPlace(results, this.computed[2]);
+
+    appendArrayFromSet(results, this.domainKeywords, i => `DOMAIN-KEYWORD,${i}`);
+    appendArrayFromSet(results, this.domainWildcard, i => `DOMAIN-REGEX,${RuleOutput.domainWildCardToRegex(i)}`);
+
+    appendArrayFromSet(results, this.processName, i => `PROCESS-NAME,${i}`);
+    appendArrayFromSet(results, this.processPath, i => `PROCESS-PATH,${i}`);
+
+    // appendArrayInPlace(results, this.otherRules);
+
+    appendArrayFromSet(results, this.ipcidrNoResolve, i => `IP-CIDR,${i},no-resolve`);
+    appendArrayFromSet(results, this.ipcidr6NoResolve, i => `IP-CIDR6,${i},no-resolve`);
+    appendArrayFromSet(results, this.ipasnNoResolve, i => `IP-ASN,${i},no-resolve`);
+    appendArrayFromSet(results, this.groipNoResolve, i => `GEOIP,${i},no-resolve`);
+
+    appendArrayFromSet(results, this.ipcidr, i => `IP-CIDR,${i}`);
+    appendArrayFromSet(results, this.ipcidr6, i => `IP-CIDR6,${i}`);
+    appendArrayFromSet(results, this.ipasn, i => `IP-ASN,${i}`);
+    appendArrayFromSet(results, this.geoip, i => `GEOIP,${i}`);
+
+    return results;
+  }
+
+  singbox(): string[] {
+    const singbox: SingboxSourceFormat = {
+      version: 2,
+      rules: [{
+        domain: ['this_ruleset_is_made_by_sukkaw.ruleset.skk.moe'].concat(this.computed[0]),
+        domain_suffix: this.computed[1],
+        domain_keyword: Array.from(this.domainKeywords),
+        domain_regex: Array.from(this.domainWildcard).map(RuleOutput.domainWildCardToRegex),
+        ip_cidr: appendArrayFromSet([], [this.ipcidr, this.ipcidrNoResolve, this.ipcidr6, this.ipcidr6NoResolve]),
+        process_name: Array.from(this.processName),
+        process_path: Array.from(this.processPath)
+      }]
+    };
+
+    return RuleOutput.jsonToLines(singbox);
+  }
+}