@0xkahi/pi-qol 0.0.1 → 0.0.3

package/src/schemas/model-select.config.schema.ts

@@ -0,0 +1,18 @@
+import z from 'zod';
+import { ModelConfigSchema } from './shared-config.schema';
+
+const FavouriteModelSchema = ModelConfigSchema.omit({ reasoning: true });
+
+const ModelSelectLayoutSchema = z.enum(['inline', 'overlay']);
+export type ModelSelectLayout = z.infer<typeof ModelSelectLayoutSchema>;
+
+export const ModelSelectConfigSchema = z.object({
+  enabled: z.boolean().default(false),
+  favourite: z.array(FavouriteModelSchema).default([]),
+  provider_filter: z.array(z.string().min(1)).default([]),
+  layout: ModelSelectLayoutSchema.default('inline'),
+});
+export type ModelSelectConfig = z.infer<typeof ModelSelectConfigSchema>;
+
+export const PartialModelSelectConfigSchema = ModelSelectConfigSchema.partial();
+export type PartialModelSelectConfig = z.infer<typeof PartialModelSelectConfigSchema>;

package/src/schemas/shared-config.schema.ts

@@ -1,11 +1,14 @@
 import z from 'zod';
+import { COLOR_HEX_REGEX } from '../constants';
 
 export const ReasoningLevelSchema = z.enum(['off', 'minimal', 'low', 'medium', 'high', 'xhigh']);
 export type ReasoningLevel = z.infer<typeof ReasoningLevelSchema>;
 
-export const ModelConfigschema = z.object({
+export const ModelConfigSchema = z.object({
   provider: z.string(),
   modelId: z.string(),
   reasoning: ReasoningLevelSchema,
 });
-export type ModelConfig = z.infer<typeof ModelConfigschema>;
+export type ModelConfig = z.infer<typeof ModelConfigSchema>;
+
+export const ColorHexSchema = z.string().regex(COLOR_HEX_REGEX, { message: 'Invalid color format. Must be a 7-character hex code (e.g., #RRGGBB).' });

package/src/types.ts

@@ -0,0 +1 @@
+export type ObjectValues<T> = T[keyof T];

package/src/utils/crayon.util.ts

@@ -0,0 +1,58 @@
+import { COLOR_HEX_REGEX } from '../constants';
+
+type ColorizeOptions = {
+  bg?: string;
+  fg?: string;
+};
+
+function hexToRgb(hex: string): { r: number; g: number; b: number } {
+  return {
+    r: Number.parseInt(hex.slice(1, 3), 16),
+    g: Number.parseInt(hex.slice(3, 5), 16),
+    b: Number.parseInt(hex.slice(5, 7), 16),
+  };
+}
+
+function fgAnsi(hex: string): string {
+  const { r, g, b } = hexToRgb(hex);
+  return `\x1b[38;2;${r};${g};${b}m`;
+}
+
+function bgAnsi(hex: string): string {
+  const { r, g, b } = hexToRgb(hex);
+  return `\x1b[48;2;${r};${g};${b}m`;
+}
+
+function colorize(text: string, { bg, fg }: ColorizeOptions = {}): string {
+  const hasFg = fg !== undefined && COLOR_HEX_REGEX.test(fg);
+  const hasBg = bg !== undefined && COLOR_HEX_REGEX.test(bg);
+
+  if (!hasFg && !hasBg) {
+    return text;
+  }
+
+  const prefix = `${hasBg ? bgAnsi(bg) : ''}${hasFg ? fgAnsi(fg) : ''}`;
+  const suffix = `${hasFg ? '\x1b[39m' : ''}${hasBg ? '\x1b[49m' : ''}`;
+
+  return `${prefix}${text}${suffix}`;
+}
+
+const ANSI_ESCAPE_REGEX = new RegExp(`${String.fromCharCode(27)}\\[[0-?]*[ -/]*[@-~]`, 'g');
+
+function stripAnsi(text: string): string {
+  return text.replace(ANSI_ESCAPE_REGEX, '');
+}
+
+export const crayon = {
+  colorize(text: string, opts: ColorizeOptions = {}): string {
+    return colorize(text, opts);
+  },
+
+  reverseVideo(text: string): string {
+    return `\x1b[7m${text}\x1b[27m`;
+  },
+
+  stripAnsi(text: string): string {
+    return stripAnsi(text);
+  },
+};

package/src/utils/model-resolver.util.ts

@@ -7,6 +7,7 @@ export type ResolvedModel = {
   apiKey?: string;
   headers?: Record<string, string>;
   reasoning?: ModelConfig['reasoning'];
+  isOauth?: boolean;
 };
 
 export type ResolveModelResult = {
@@ -50,6 +51,7 @@ export class ModelResolver {
             apiKey: auth.apiKey,
             headers: auth.headers,
             reasoning: candidate.reasoning,
+            isOauth: this.ctx.modelRegistry.isUsingOAuth(model),
           },
         };
       }

package/src/utils/path.util.ts

@@ -31,6 +31,10 @@ export class PathUtil {
     }
   }
 
+  static findPiAuthConfig(): FileSearchResult {
+    return PathUtil.findFile(path.join(getAgentDir(), 'auth.json'));
+  }
+
   private static getExtensionConfig(paths: string[]): string {
     return path.join(...paths, 'extensions', EXTENSION_ID, 'config.json');
   }

package/src/utils/raw-data-parser.util.ts

@@ -0,0 +1,14 @@
+export class RawDataParser {
+  static asRecord(value: unknown): Record<string, unknown> | undefined {
+    return value && typeof value === 'object' && !Array.isArray(value) ? (value as Record<string, unknown>) : undefined;
+  }
+
+  static stringValue(value: unknown): string | undefined {
+    return typeof value === 'string' && value.trim() ? value.trim() : undefined;
+  }
+
+  static numberValue(value: unknown): number | undefined {
+    const number = typeof value === 'number' ? value : Number(value);
+    return Number.isFinite(number) ? number : undefined;
+  }
+}