@jsbot-tellus/core 2.9.1 → 2.10.0

package/src/core/plugin/params/type-level.ts

@@ -11,7 +11,46 @@ type Trim<Source extends string> = TrimLeft<TrimRight<Source>>
 type StripLength<Type extends string> =
     Trim<Type> extends `${infer Base}(${string})` ? Trim<Base> : Trim<Type>
 
-type ParamValue<Type extends string> =
+type Unescape<Source extends string> =
+    Source extends `${infer Head}\\${infer Next}${infer Rest}`
+        ? `${Head}${Next}${Unescape<Rest>}`
+        : Source
+
+type SplitUnionScan<
+    Source extends string,
+    Quote extends string | null = null,
+    Current extends string = '',
+    Acc extends string[] = [],
+> = Source extends `${infer Char}${infer Rest}`
+    ? Char extends '\\'
+        ? Rest extends `${infer Next}${infer Tail}`
+            ? SplitUnionScan<Tail, Quote, `${Current}${Char}${Next}`, Acc>
+            : [...Acc, `${Current}${Char}`]
+        : Char extends '"' | "'"
+          ? Quote extends null
+              ? SplitUnionScan<Rest, Char, `${Current}${Char}`, Acc>
+              : Char extends Quote
+                ? SplitUnionScan<Rest, null, `${Current}${Char}`, Acc>
+                : SplitUnionScan<Rest, Quote, `${Current}${Char}`, Acc>
+          : Char extends '|'
+            ? Quote extends null
+                ? SplitUnionScan<Rest, null, '', [...Acc, Current]>
+                : SplitUnionScan<Rest, Quote, `${Current}${Char}`, Acc>
+            : SplitUnionScan<Rest, Quote, `${Current}${Char}`, Acc>
+    : [...Acc, Current]
+
+type SplitUnion<Source extends string> = Source extends `${string}|${string}`
+    ? SplitUnionScan<Source>
+    : [Source]
+
+type LiteralValue<Type extends string> =
+    Trim<Type> extends `'${infer Inner}'`
+        ? Unescape<Inner>
+        : Trim<Type> extends `"${infer Inner}"`
+          ? Unescape<Inner>
+          : never
+
+type BuiltinValue<Type extends string> =
     StripLength<Type> extends 'string' | 'text'
         ? string
         : StripLength<Type> extends
@@ -21,19 +60,88 @@ type ParamValue<Type extends string> =
             ? boolean
             : unknown
 
+type ParamValuePart<Type extends string> =
+    Trim<Type> extends `'${string}'` | `"${string}"`
+        ? LiteralValue<Type>
+        : BuiltinValue<Type>
+
+type ParamValue<Type extends string> =
+    SplitUnion<Type>[number] extends infer Part
+        ? Part extends string
+            ? ParamValuePart<Part>
+            : never
+        : never
+
+type SplitTypeScan<
+    Source extends string,
+    Quote extends string | null = null,
+    Name extends string = '',
+> = Source extends `${infer Char}${infer Rest}`
+    ? Char extends '\\'
+        ? Rest extends `${infer Next}${infer Tail}`
+            ? SplitTypeScan<Tail, Quote, `${Name}${Char}${Next}`>
+            : [Name, null]
+        : Char extends '"' | "'"
+          ? Quote extends null
+              ? SplitTypeScan<Rest, Char, `${Name}${Char}`>
+              : Char extends Quote
+                ? SplitTypeScan<Rest, null, `${Name}${Char}`>
+                : SplitTypeScan<Rest, Quote, `${Name}${Char}`>
+          : Char extends ':'
+            ? Quote extends null
+                ? [Name, Rest]
+                : SplitTypeScan<Rest, Quote, `${Name}${Char}`>
+            : SplitTypeScan<Rest, Quote, `${Name}${Char}`>
+    : [Name, null]
+
+type SplitType<Source extends string> = Source extends `${string}:${string}`
+    ? SplitTypeScan<Source>
+    : [Source, null]
+
+type SplitDefaultScan<
+    Source extends string,
+    Quote extends string | null = null,
+    Head extends string = '',
+> = Source extends `${infer Char}${infer Rest}`
+    ? Char extends '\\'
+        ? Rest extends `${infer Next}${infer Tail}`
+            ? SplitDefaultScan<Tail, Quote, `${Head}${Char}${Next}`>
+            : [Head, false]
+        : Char extends '"' | "'"
+          ? Quote extends null
+              ? SplitDefaultScan<Rest, Char, `${Head}${Char}`>
+              : Char extends Quote
+                ? SplitDefaultScan<Rest, null, `${Head}${Char}`>
+                : SplitDefaultScan<Rest, Quote, `${Head}${Char}`>
+          : Char extends '='
+            ? Quote extends null
+                ? [Head, true]
+                : SplitDefaultScan<Rest, Quote, `${Head}${Char}`>
+            : SplitDefaultScan<Rest, Quote, `${Head}${Char}`>
+    : [Head, false]
+
+type SplitDefault<Source extends string> = Source extends `${string}=${string}`
+    ? SplitDefaultScan<Source>
+    : [Source, false]
+
 interface VariadicParam {
     readonly variadic: true
 }
 
-type ParamType<Head extends string> = Head extends `${string}:${infer Type}`
-    ? ParamValue<Type>
-    : string
+type ParamType<Head extends string> =
+    SplitType<Head> extends [unknown, infer Type]
+        ? Type extends string
+            ? ParamValue<Type>
+            : string
+        : string
 
 type ParamElement<Body extends string, Required extends boolean> =
     Trim<Body> extends `...${string}`
         ? VariadicParam
-        : Trim<Body> extends `${infer Head}=${string}`
-          ? ParamType<Head>
+        : SplitDefault<Trim<Body>> extends [infer Head, true]
+          ? Head extends string
+              ? ParamType<Head>
+              : never
           : Required extends true
             ? ParamType<Body>
             : ParamType<Body> | undefined

package/src/core/plugin/params/types.ts

@@ -35,14 +35,26 @@ export interface ParamLength {
     max?: number | undefined
 }
 
+export interface ParamTypeCandidate {
+    kind: 'type'
+    type: CommandParamType
+    length?: ParamLength | undefined
+}
+
+export interface ParamLiteralCandidate {
+    kind: 'literal'
+    value: string
+}
+
+export type ParamCandidate = ParamTypeCandidate | ParamLiteralCandidate
+
 export interface ParamSpec {
     name: string
-    type: CommandParamType
+    candidates: ParamCandidate[]
     required: boolean
     rest: boolean
     hasDefault: boolean
     defaultValue?: unknown
-    length?: ParamLength | undefined
 }
 
 export interface InputToken {
@@ -53,3 +65,35 @@ export interface InputToken {
 export function declarationError(message: string): Error {
     return new Error(`参数声明错误：${message}`)
 }
+
+export function candidateLabel(candidate: ParamCandidate): string {
+    if (candidate.kind === 'literal') {
+        const escaped = candidate.value
+            .replace(/\\/g, '\\\\')
+            .replace(/'/g, "\\'")
+
+        return `'${escaped}'`
+    }
+
+    const length = candidate.length
+
+    if (length === undefined) {
+        return candidate.type
+    }
+
+    return `${candidate.type}(${length.min ?? ''}..${length.max ?? ''})`
+}
+
+export function joinCandidateLabels(
+    candidates: readonly ParamCandidate[]
+): string {
+    const labels = candidates.map(candidateLabel)
+
+    if (labels.length === 0) return ''
+
+    if (labels.length === 1) return labels[0] ?? ''
+
+    const last = labels[labels.length - 1] ?? ''
+
+    return `${labels.slice(0, -1).join('、')} 或 ${last}`
+}