@myvillage/cli 1.5.0 → 1.6.0

package/src/utils/soulprint-workspace.js

@@ -0,0 +1,158 @@
+import { existsSync, mkdirSync, readFileSync, writeFileSync, readdirSync } from 'fs';
+import { join } from 'path';
+import { execSync } from 'child_process';
+import { getConfigDir } from './config.js';
+import { parse as parseYaml, stringify as stringifyYaml } from 'yaml';
+
+const SOULPRINT_DIR_NAME = 'soulprint';
+
+// ── Directory Helpers ──────────────────────────────────
+
+export function getSoulprintDir() {
+  return join(getConfigDir(), SOULPRINT_DIR_NAME);
+}
+
+export function getDatasetsDir() {
+  return join(getSoulprintDir(), 'datasets');
+}
+
+export function getModelsDir() {
+  return join(getSoulprintDir(), 'models');
+}
+
+export function getConfigsDir() {
+  return join(getSoulprintDir(), 'configs');
+}
+
+export function getLogsDir() {
+  return join(getSoulprintDir(), 'logs');
+}
+
+export function getVenvDir() {
+  return join(getSoulprintDir(), 'venv');
+}
+
+export function getScriptsDir() {
+  return join(getSoulprintDir(), 'scripts');
+}
+
+// ── Workspace State ────────────────────────────────────
+
+export function isWorkspaceInitialized() {
+  const dir = getSoulprintDir();
+  return existsSync(dir) && existsSync(join(dir, 'workspace.yaml'));
+}
+
+export function ensureWorkspace() {
+  return isWorkspaceInitialized();
+}
+
+export function readWorkspaceConfig() {
+  const configPath = join(getSoulprintDir(), 'workspace.yaml');
+  if (!existsSync(configPath)) return null;
+  try {
+    return parseYaml(readFileSync(configPath, 'utf-8'));
+  } catch {
+    return null;
+  }
+}
+
+export function writeWorkspaceConfig(config) {
+  const configPath = join(getSoulprintDir(), 'workspace.yaml');
+  writeFileSync(configPath, stringifyYaml(config, { lineWidth: 0 }));
+}
+
+// ── Dataset Helpers ────────────────────────────────────
+
+export function getLocalDatasetDir(slug, version) {
+  return join(getDatasetsDir(), slug, `v${version}`);
+}
+
+export function isDatasetDownloaded(slug, version) {
+  const dir = getLocalDatasetDir(slug, version);
+  return existsSync(dir) && existsSync(join(dir, 'manifest.json'));
+}
+
+export function listLocalDatasets() {
+  const dir = getDatasetsDir();
+  if (!existsSync(dir)) return [];
+  return readdirSync(dir, { withFileTypes: true })
+    .filter(e => e.isDirectory())
+    .map(e => e.name);
+}
+
+// ── Python Environment ─────────────────────────────────
+
+export function getPythonPath() {
+  const venvPython = join(getVenvDir(), 'bin', 'python');
+  if (existsSync(venvPython)) return venvPython;
+  return null;
+}
+
+export function isPythonAvailable() {
+  try {
+    execSync('python3 --version', { stdio: 'pipe' });
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+export function getPythonVersion() {
+  try {
+    return execSync('python3 --version', { stdio: 'pipe' }).toString().trim();
+  } catch {
+    return null;
+  }
+}
+
+export function detectGPU() {
+  try {
+    const nvidiaSmi = execSync('nvidia-smi --query-gpu=name,memory.total --format=csv,noheader', { stdio: 'pipe' }).toString().trim();
+    if (nvidiaSmi) {
+      const [name, vram] = nvidiaSmi.split(', ');
+      return { type: 'cuda', name, vram };
+    }
+  } catch { /* no CUDA */ }
+
+  try {
+    const sysctl = execSync('sysctl -n machdep.cpu.brand_string', { stdio: 'pipe' }).toString().trim();
+    if (sysctl.includes('Apple')) {
+      const mem = execSync('sysctl -n hw.memsize', { stdio: 'pipe' }).toString().trim();
+      const memGB = Math.round(parseInt(mem) / 1073741824);
+      return { type: 'mps', name: sysctl, vram: `${memGB}GB unified` };
+    }
+  } catch { /* not Apple Silicon */ }
+
+  return { type: 'cpu', name: 'CPU only', vram: null };
+}
+
+export function getMachineInfo() {
+  const gpu = detectGPU();
+  return {
+    gpu: gpu.name,
+    gpuType: gpu.type,
+    vram: gpu.vram,
+    os: `${process.platform} ${process.arch}`,
+    python: getPythonVersion(),
+    nodeVersion: process.version,
+  };
+}
+
+// ── Training Job Local State ───────────────────────────
+
+export function getJobOutputDir(jobId) {
+  return join(getModelsDir(), jobId);
+}
+
+export function getJobLogFile(jobId) {
+  const dir = getLogsDir();
+  if (!existsSync(dir)) mkdirSync(dir, { recursive: true });
+  return join(dir, `${jobId}.jsonl`);
+}
+
+export function appendJobLog(jobId, entry) {
+  const logFile = getJobLogFile(jobId);
+  const line = JSON.stringify({ ...entry, ts: new Date().toISOString() }) + '\n';
+  writeFileSync(logFile, line, { flag: 'a' });
+}

package/src/utils/village-resolver.js

@@ -0,0 +1,104 @@
+import chalk from 'chalk';
+import inquirer from 'inquirer';
+import { loadCredentials } from './auth.js';
+import { brand, villageSpinner } from './brand.js';
+import {
+  getVillagerVillages,
+  getVillagerByVillagerId,
+  getVillageCommunities,
+  joinCommunity,
+} from './api.js';
+
+/**
+ * Resolves a villager's village context: which village they're acting on behalf of
+ * and the corresponding MAN community.
+ *
+ * Returns { villagerUuid, village, community } or null if resolution fails.
+ */
+export async function resolveVillageContext() {
+  const creds = loadCredentials();
+  if (!creds) return null;
+
+  const spinner = villageSpinner('Resolving village context...').start();
+
+  try {
+    // 1. Get villager UUID
+    let villagerUuid = creds.villager_uuid;
+
+    if (!villagerUuid && creds.villager_id) {
+      const result = await getVillagerByVillagerId(creds.villager_id);
+      villagerUuid = result?.data?.id || result?.id;
+    }
+
+    if (!villagerUuid) {
+      spinner.fail('Could not resolve villager profile.');
+      console.log(chalk.dim('  Try logging out and back in: myvillage logout && myvillage login'));
+      return null;
+    }
+
+    // 2. Get associated villages
+    const villagesResult = await getVillagerVillages(villagerUuid);
+    const villages = villagesResult?.data || [];
+
+    if (villages.length === 0) {
+      spinner.fail('You are not associated with any villages.');
+      console.log(chalk.dim('  Join a village at portal.myvillageproject.ai to get started.'));
+      return null;
+    }
+
+    spinner.stop();
+
+    // 3. Select village
+    let village;
+    if (villages.length === 1) {
+      village = villages[0];
+      console.log(brand.teal(`  Village: ${village.name}`));
+    } else {
+      const { selected } = await inquirer.prompt([{
+        type: 'list',
+        name: 'selected',
+        message: 'Which village are you representing?',
+        choices: villages.map((v) => ({
+          name: `${v.name} (${v.city}, ${v.state})`,
+          value: v,
+        })),
+      }]);
+      village = selected;
+    }
+
+    // 4. Get village's dedicated community
+    const commSpinner = villageSpinner('Finding village community...').start();
+    const commResult = await getVillageCommunities(village.id);
+    const communities = commResult?.data || [];
+
+    // Find the community linked to this village
+    const community = communities.find((c) => c.villageId === village.id) || communities[0];
+
+    if (!community) {
+      commSpinner.fail('No MAN community found for this village.');
+      console.log(chalk.dim('  A community needs to be set up for your village first.'));
+      return null;
+    }
+
+    // 5. Auto-join community (ignore 409 = already member)
+    try {
+      await joinCommunity(community.slug);
+    } catch (err) {
+      if (err.response?.status !== 409 && err.response?.status !== 400) {
+        // Only throw if it's not "already a member"
+        const msg = err.response?.data?.error || '';
+        if (!msg.toLowerCase().includes('already')) {
+          throw err;
+        }
+      }
+    }
+
+    commSpinner.succeed(`Community: ${community.name}`);
+
+    return { villagerUuid, village, community };
+  } catch (err) {
+    spinner.stop();
+    console.log(chalk.red(`  \u2717 Failed to resolve village context: ${err.message}`));
+    return null;
+  }
+}