aryx-cli 1.0.0

package/src/services/firestoreRest.ts

@@ -0,0 +1,119 @@
+import type { UsageData } from './ai.js';
+import { FIREBASE_PROJECT_ID } from '../constants.js';
+
+const BASE = `https://firestore.googleapis.com/v1/projects/${FIREBASE_PROJECT_ID}/databases/(default)/documents`;
+
+type FVal = string | number | boolean | Date;
+
+const toValue = (v: FVal): object => {
+  if (v instanceof Date) return { timestampValue: v.toISOString() };
+  if (typeof v === 'string') return { stringValue: v };
+  if (typeof v === 'boolean') return { booleanValue: v };
+  if (Number.isInteger(v)) return { integerValue: String(v) };
+
+  return { doubleValue: v };
+};
+
+const patch = async (
+  path: string,
+  fields: Record<string, FVal>,
+  token: string
+): Promise<void> => {
+  const mask = Object.keys(fields)
+    .map(k => `updateMask.fieldPaths=${k}`)
+    .join('&');
+
+  const fieldsEntries = Object.entries(fields)
+    .map(([k, v]) => [k, toValue(v)]);
+
+  await fetch(`${BASE}/${path}?${mask}`, {
+    method: 'PATCH',
+    headers: { 
+      Authorization: `Bearer ${token}`, 
+      'Content-Type': 'application/json' 
+    },
+    body: JSON.stringify({ 
+      fields: Object.fromEntries(fieldsEntries) 
+    }),
+  });
+};
+
+export type SessionTotals = {
+  inputTokens: number;
+  outputTokens: number;
+  totalTokens: number;
+  cost: number;
+  totalResponseMs: number;
+  messagesCount: number;
+};
+
+export type UserPlan = { 
+  planMessages: number; 
+  messageCount: number; 
+  subscriptionPlan: string 
+};
+
+export const getUserPlan = async (uid: string, token: string): Promise<UserPlan | null> => {
+  const res = await fetch(`${BASE}/users/${uid}`, {
+    headers: { 
+      Authorization: `Bearer ${token}` 
+    },
+  });
+
+  if (!res.ok) return null;
+
+  const data = await res.json() as { 
+    fields?: Record<string, { integerValue?: string; stringValue?: string }> 
+  };
+  const f = data.fields ?? {};
+
+  return {
+    planMessages: Number(f.planMessages?.integerValue ?? 0),
+    messageCount: Number(f.messageCount?.integerValue ?? 0),
+    subscriptionPlan: f.subscriptionPlan?.stringValue ?? 'free',
+  };
+};
+
+export const incrementMessageCount = (uid: string, count: number, token: string): Promise<void> =>
+  patch(`users/${uid}`, { messageCount: count }, token);
+
+export const saveMessageToFirestore = async (
+  uid: string,
+  sessionId: string,
+  sessionCreatedAt: Date,
+  messageId: string,
+  prompt: string,
+  response: string,
+  usage: UsageData,
+  totals: SessionTotals,
+  token: string
+): Promise<void> => {
+  const now = new Date();
+
+  await Promise.all([
+    patch(`users/${uid}/sessions/${sessionId}`, {
+      createdAt: sessionCreatedAt,
+      lastUpdated: now,
+      model: usage.model,
+      totalInputTokens: totals.inputTokens,
+      totalOutputTokens: totals.outputTokens,
+      totalTokens: totals.totalTokens,
+      totalCost: totals.cost,
+      totalResponseMs: totals.totalResponseMs,
+      messagesCount: totals.messagesCount,
+    }, token),
+
+    patch(`users/${uid}/sessions/${sessionId}/messages/${messageId}`, {
+      createdAt: now,
+      prompt,
+      response,
+      inputTokens: usage.inputTokens,
+      outputTokens: usage.outputTokens,
+      totalTokens: usage.totalTokens,
+      cost: usage.cost,
+      model: usage.model,
+      responseTimeMs: usage.responseTimeMs,
+    }, token),
+  ]);
+};
+

package/src/services/loginServer.ts

@@ -0,0 +1,91 @@
+import { createServer, IncomingMessage, ServerResponse } from 'node:http';
+import { URL } from 'node:url';
+import { WEB_URL } from '../constants.js';
+
+export type CallbackData = {
+  uid: string;
+  email: string;
+  displayName: string;
+  photoURL: string;
+  idToken: string;
+  refreshToken: string;
+};
+
+export async function startCallbackServer(): Promise<{
+  port: number;
+  waitForCallback: Promise<CallbackData>;
+}> {
+  let resolveCallback: (data: CallbackData) => void;
+  let rejectCallback: (err: Error) => void;
+
+  const waitForCallback = new Promise<CallbackData>((resolve, reject) => {
+    resolveCallback = resolve;
+    rejectCallback = reject;
+  });
+
+  const server = createServer((req: IncomingMessage, res: ServerResponse) => {
+    if (!req.url) return;
+
+    const parsed = new URL(req.url, 'http://localhost');
+
+    if (parsed.pathname !== '/callback') {
+      res.writeHead(404);
+      res.end();
+      return;
+    }
+
+    const uid = parsed.searchParams.get('uid');
+    const email = parsed.searchParams.get('email');
+    const displayName = parsed.searchParams.get('displayName') ?? '';
+    const photoURL = parsed.searchParams.get('photoURL') ?? '';
+    const idToken = parsed.searchParams.get('idToken');
+    const refreshToken = parsed.searchParams.get('refreshToken');
+
+    if (!uid || !email || !idToken || !refreshToken) {
+      res.writeHead(302, { 
+        Location: `${WEB_URL}/auth/error` 
+      });
+      res.end();
+
+      rejectCallback(new Error('Missing required params in callback'));
+      server.close();
+
+      return;
+    }
+
+    res.writeHead(302, { 
+      Location: `${WEB_URL}/auth/success` 
+    });
+    res.end();
+
+    server.close();
+    resolveCallback({ 
+      uid, 
+      email, 
+      displayName, 
+      photoURL, 
+      idToken, 
+      refreshToken 
+    });
+  });
+
+  await new Promise<void>((resolve) => {
+    server.listen(0, '127.0.0.1', resolve);
+  });
+
+  const addr = server.address() as { port: number };
+  const port = addr.port;
+
+  // 5-minute timeout
+  const timeout = setTimeout(() => {
+    server.close();
+    rejectCallback(new Error('Login timed out. Please try again.'));
+  }, 5 * 60 * 1000);
+
+  const waitWithCleanup = waitForCallback.finally(() => {
+    clearTimeout(timeout);
+  });
+
+  return { port, waitForCallback: waitWithCleanup };
+}
+

package/src/services/platform.ts

@@ -0,0 +1,26 @@
+import { exec } from 'node:child_process';
+
+/** Open a URL in the user's default browser (cross-platform). */
+export const openBrowser = (url: string): void => {
+  const cmd =
+    process.platform === 'win32' 
+      ? `start "" "${url}"` 
+      : process.platform === 'darwin' 
+      ? `open "${url}"` 
+      : `xdg-open "${url}"`;
+
+  exec(cmd);
+};
+
+/** Copy text to the system clipboard (cross-platform). */
+export const copyToClipboard = (text: string): void => {
+  const cmd =
+    process.platform === 'win32' 
+      ? `powershell -command "Set-Clipboard '${text}'"` 
+      : process.platform === 'darwin' 
+      ? `printf '%s' "${text}" | pbcopy` 
+      : `printf '%s' "${text}" | xclip -selection clipboard`;
+
+  exec(cmd);
+};
+

package/src/theme.ts

@@ -0,0 +1,46 @@
+import { readConfig, writeConfig } from './services/config.js';
+
+// ─── Config Persistence ──────────────────────────────────────────────────────
+const load = (): 1 | 2 | 3 | 4 => (readConfig().diffTheme as 1 | 2 | 3 | 4) ?? 1;
+
+export const saveTheme = (n: 1 | 2 | 3 | 4) => writeConfig({ ...readConfig(), diffTheme: n });
+
+export const isSetupDone = (): boolean => readConfig().setupDone === true;
+export const markSetupDone = () => writeConfig({ ...readConfig(), setupDone: true });
+
+// ─── Diff Colors ─────────────────────────────────────────────────────────────
+export const DIFF_COLORS = {
+  1: { add: '#07224b', rm: '#480c0b' }, // Modern (background)
+  2: { add: '#083418', rm: '#480c0b' }, // Classic (background)
+  3: { add: '#609ee9', rm: '#e34242' }, // Modern (text only)
+  4: { add: '#29af42', rm: '#e34242' }, // Classic (text only)
+} as const;
+
+// ─── App Theme ───────────────────────────────────────────────────────────────
+export const theme = {
+  colors: { primary: '#70b0ff', highlight: '#bcff92' },
+  diffTheme: load(),
+};
+
+// ─── Subscription Plan Colors ────────────────────────────────────────────────
+export const PLAN_COLORS: Record<string, string> = {
+  free: '#c9c9c9',
+  pro: '#19c8b4',
+  max: '#ff9f43',
+};
+
+// ─── Message Colors ──────────────────────────────────────────────────────────
+export const USER_MESSAGE_BG = '#3a3a3a';
+
+// ─── Loader Shimmer Colors ──────────────────────────────────────────────────
+export const SHIMMER_COLORS = {
+  bright: '#ebffeb',
+  mid: '#eaffea',
+  dim: '#d6ffd6',
+};
+
+// ─── Gradient Palettes ───────────────────────────────────────────────────────
+/** Gradient used for brand text (IAmAryx, exit message). */
+export const GRADIENT_BRAND = ['#ff6b6b', '#6bcb77', '#4d96ff', '#ff9f43', '#ffd93d', '#a29bfe'];
+/** Gradient used for command suggestions (/aryx). */
+export const GRADIENT_COMMAND = ['#ff6b6b', '#ff9f43', '#ffd93d', '#6bcb77', '#4d96ff', '#a29bfe'];