git-push-time 1.1.2 → 1.1.4

package/lib/config.js

@@ -1,220 +1,220 @@
-import fs from 'fs';
-import path from 'path';
-import os from 'os';
-import { spawnSync } from 'child_process';
-
-const CONFIG_DIR = path.join(os.homedir(), '.git-push-time');
-const CONFIG_FILE = path.join(CONFIG_DIR, 'config.json');
-
-function ensureConfigDir() {
-  if (!fs.existsSync(CONFIG_DIR)) {
-    try {
-      fs.mkdirSync(CONFIG_DIR, { recursive: true });
-    } catch {
-      // Ignore directory creation errors
-    }
-  }
-}
-
-/**
- * Đọc toàn bộ file cấu hình người dùng
- * @returns {Record<string, any>}
- */
-export function getConfig() {
-  try {
-    if (fs.existsSync(CONFIG_FILE)) {
-      const raw = fs.readFileSync(CONFIG_FILE, 'utf-8');
-      return JSON.parse(raw);
-    }
-  } catch {
-    // Ignore JSON parse errors
-  }
-  return {};
-}
-
-/**
- * Lưu cấu hình người dùng
- * @param {Record<string, any>} data 
- */
-export function saveConfig(data) {
-  try {
-    ensureConfigDir();
-    const current = getConfig();
-    const merged = { ...current, ...data };
-    fs.writeFileSync(CONFIG_FILE, JSON.stringify(merged, null, 2), 'utf-8');
-    try {
-      // Token nhạy cảm: giới hạn quyền đọc chỉ owner (chmod 600)
-      fs.chmodSync(CONFIG_FILE, 0o600);
-    } catch {
-      // Bỏ qua trên Windows hoặc FS không hỗ trợ chmod
-    }
-    return true;
-  } catch {
-    return false;
-  }
-}
-
-/**
- * Lấy danh sách tất cả các tài khoản GitHub đã lưu
- * @returns {Array<{ login: string, name: string, token: string, active: boolean, avatar_url?: string }>}
- */
-export function getAllGitHubAccounts() {
-  const cfg = getConfig();
-  const accountsMap = cfg.githubAccounts || {};
-  const activeLogin = cfg.activeGitHubLogin || (cfg.github?.user?.login) || null;
-
-  const result = [];
-  for (const [login, acc] of Object.entries(accountsMap)) {
-    if (acc && typeof acc.token === 'string' && acc.token.trim()) {
-      result.push({
-        login,
-        name: acc.user?.name || login,
-        avatar_url: acc.user?.avatar_url || '',
-        token: acc.token,
-        active: login === activeLogin
-      });
-    }
-  }
-
-  // Tương thích ngược với trường `github` cũ nếu chưa migrate
-  if (result.length === 0 && cfg.github && cfg.github.token) {
-    const login = cfg.github.user?.login || 'default';
-    result.push({
-      login,
-      name: cfg.github.user?.name || login,
-      avatar_url: cfg.github.user?.avatar_url || '',
-      token: cfg.github.token,
-      active: true
-    });
-  }
-
-  return result;
-}
-
-/**
- * Đặt tài khoản GitHub đang kích hoạt
- * @param {string} login 
- */
-export function setActiveGitHubAccount(login) {
-  const cfg = getConfig();
-  cfg.activeGitHubLogin = login;
-  if (cfg.githubAccounts && cfg.githubAccounts[login]) {
-    cfg.github = {
-      token: cfg.githubAccounts[login].token,
-      user: cfg.githubAccounts[login].user,
-      savedAt: new Date().toISOString()
-    };
-  }
-  return saveConfig(cfg);
-}
-
-/**
- * Lấy GitHub token từ nhiều nguồn theo thứ tự ưu tiên:
- * 1. Biến môi trường GITHUB_TOKEN hoặc GH_TOKEN
- * 2. Tài khoản GitHub active trong file config
- * 3. Token từ GitHub CLI (`gh auth token`) nếu có
- * @returns {{ token: string | null, source: 'env' | 'config' | 'gh-cli' | null, user?: any }}
- */
-export function getGitHubToken() {
-  // 1. Kiểm tra Environment variables
-  if (process.env.GITHUB_TOKEN && process.env.GITHUB_TOKEN.trim()) {
-    return { token: process.env.GITHUB_TOKEN.trim(), source: 'env' };
-  }
-  if (process.env.GH_TOKEN && process.env.GH_TOKEN.trim()) {
-    return { token: process.env.GH_TOKEN.trim(), source: 'env' };
-  }
-
-  // 2. Kiểm tra File config đã lưu
-  const cfg = getConfig();
-  const activeLogin = cfg.activeGitHubLogin;
-  if (activeLogin && cfg.githubAccounts && cfg.githubAccounts[activeLogin]) {
-    const acc = cfg.githubAccounts[activeLogin];
-    if (acc && typeof acc.token === 'string' && acc.token.trim()) {
-      return { token: acc.token.trim(), source: 'config', user: acc.user };
-    }
-    return { token: null, source: null };
-  }
-  if (cfg.github && cfg.github.token && typeof cfg.github.token === 'string') {
-    return { token: cfg.github.token.trim(), source: 'config', user: cfg.github.user };
-  }
-
-  // 3. Kiểm tra GitHub CLI gh
-  try {
-    const r = spawnSync('gh', ['auth', 'token'], { encoding: 'utf-8', stdio: ['ignore', 'pipe', 'ignore'] });
-    if (r.status === 0 && r.stdout && r.stdout.trim()) {
-      return { token: r.stdout.trim(), source: 'gh-cli' };
-    }
-  } catch {
-    // gh not installed or error
-  }
-
-  return { token: null, source: null };
-}
-
-/**
- * Lưu tài khoản GitHub vào file config (hỗ trợ đa tài khoản)
- * @param {string} token 
- * @param {object} userInfo 
- */
-export function saveGitHubToken(token, userInfo = null) {
-  if (!token || typeof token !== 'string' || !token.trim()) return false;
-  const current = getConfig();
-  if (!current.githubAccounts) {
-    current.githubAccounts = {};
-  }
-
-  const login = userInfo?.login || current.github?.user?.login || 'default';
-  current.githubAccounts[login] = {
-    token: token.trim(),
-    savedAt: new Date().toISOString(),
-    user: userInfo ? {
-      login: userInfo.login,
-      name: userInfo.name,
-      avatar_url: userInfo.avatar_url,
-      email: userInfo.email,
-      bio: userInfo.bio,
-      public_repos: userInfo.public_repos,
-      total_private_repos: userInfo.total_private_repos,
-      html_url: userInfo.html_url
-    } : (current.githubAccounts[login]?.user || null)
-  };
-
-  current.activeGitHubLogin = login;
-  current.github = {
-    token: token.trim(),
-    user: current.githubAccounts[login].user,
-    savedAt: new Date().toISOString()
-  };
-
-  return saveConfig(current);
-}
-
-/**
- * Xóa tài khoản GitHub khỏi file config
- * @param {string} [login] nếu không truyền sẽ xóa tài khoản active
- */
-export function removeGitHubToken(login = null) {
-  const current = getConfig();
-  const targetLogin = login || current.activeGitHubLogin || current.github?.user?.login;
-
-  if (targetLogin && current.githubAccounts && current.githubAccounts[targetLogin]) {
-    delete current.githubAccounts[targetLogin];
-  }
-
-  if (current.activeGitHubLogin === targetLogin) {
-    const remaining = Object.keys(current.githubAccounts || {});
-    current.activeGitHubLogin = remaining.length > 0 ? remaining[0] : null;
-    if (current.activeGitHubLogin && current.githubAccounts[current.activeGitHubLogin]) {
-      current.github = {
-        token: current.githubAccounts[current.activeGitHubLogin].token,
-        user: current.githubAccounts[current.activeGitHubLogin].user
-      };
-    } else {
-      delete current.github;
-    }
-  } else if (!targetLogin && current.github) {
-    delete current.github;
-  }
-
-  return saveConfig(current);
-}
+import fs from 'fs';
+import path from 'path';
+import os from 'os';
+import { spawnSync } from 'child_process';
+
+const CONFIG_DIR = path.join(os.homedir(), '.git-push-time');
+const CONFIG_FILE = path.join(CONFIG_DIR, 'config.json');
+
+function ensureConfigDir() {
+  if (!fs.existsSync(CONFIG_DIR)) {
+    try {
+      fs.mkdirSync(CONFIG_DIR, { recursive: true });
+    } catch {
+      // Ignore directory creation errors
+    }
+  }
+}
+
+/**
+ * Read the user's complete configuration file.
+ * @returns {Record<string, any>}
+ */
+export function getConfig() {
+  try {
+    if (fs.existsSync(CONFIG_FILE)) {
+      const raw = fs.readFileSync(CONFIG_FILE, 'utf-8');
+      return JSON.parse(raw);
+    }
+  } catch {
+    // Ignore JSON parse errors
+  }
+  return {};
+}
+
+/**
+ * Save the user's configuration.
+ * @param {Record<string, any>} data 
+ */
+export function saveConfig(data) {
+  try {
+    ensureConfigDir();
+    const current = getConfig();
+    const merged = { ...current, ...data };
+    fs.writeFileSync(CONFIG_FILE, JSON.stringify(merged, null, 2), 'utf-8');
+    try {
+      // Restrict access to sensitive tokens to the owner (chmod 600).
+      fs.chmodSync(CONFIG_FILE, 0o600);
+    } catch {
+      // Ignore on Windows or filesystems that do not support chmod.
+    }
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+/**
+ * Get all saved GitHub accounts.
+ * @returns {Array<{ login: string, name: string, token: string, active: boolean, avatar_url?: string }>}
+ */
+export function getAllGitHubAccounts() {
+  const cfg = getConfig();
+  const accountsMap = cfg.githubAccounts || {};
+  const activeLogin = cfg.activeGitHubLogin || (cfg.github?.user?.login) || null;
+
+  const result = [];
+  for (const [login, acc] of Object.entries(accountsMap)) {
+    if (acc && typeof acc.token === 'string' && acc.token.trim()) {
+      result.push({
+        login,
+        name: acc.user?.name || login,
+        avatar_url: acc.user?.avatar_url || '',
+        token: acc.token,
+        active: login === activeLogin
+      });
+    }
+  }
+
+  // Support the legacy `github` field if it has not been migrated.
+  if (result.length === 0 && cfg.github && cfg.github.token) {
+    const login = cfg.github.user?.login || 'default';
+    result.push({
+      login,
+      name: cfg.github.user?.name || login,
+      avatar_url: cfg.github.user?.avatar_url || '',
+      token: cfg.github.token,
+      active: true
+    });
+  }
+
+  return result;
+}
+
+/**
+ * Set the active GitHub account.
+ * @param {string} login 
+ */
+export function setActiveGitHubAccount(login) {
+  const cfg = getConfig();
+  cfg.activeGitHubLogin = login;
+  if (cfg.githubAccounts && cfg.githubAccounts[login]) {
+    cfg.github = {
+      token: cfg.githubAccounts[login].token,
+      user: cfg.githubAccounts[login].user,
+      savedAt: new Date().toISOString()
+    };
+  }
+  return saveConfig(cfg);
+}
+
+/**
+ * Get a GitHub token from the available sources, in priority order:
+ * 1. GITHUB_TOKEN or GH_TOKEN environment variable
+ * 2. Active GitHub account in the config file
+ * 3. GitHub CLI token (`gh auth token`), if available
+ * @returns {{ token: string | null, source: 'env' | 'config' | 'gh-cli' | null, user?: any }}
+ */
+export function getGitHubToken() {
+  // 1. Check environment variables.
+  if (process.env.GITHUB_TOKEN && process.env.GITHUB_TOKEN.trim()) {
+    return { token: process.env.GITHUB_TOKEN.trim(), source: 'env' };
+  }
+  if (process.env.GH_TOKEN && process.env.GH_TOKEN.trim()) {
+    return { token: process.env.GH_TOKEN.trim(), source: 'env' };
+  }
+
+  // 2. Check the saved config file.
+  const cfg = getConfig();
+  const activeLogin = cfg.activeGitHubLogin;
+  if (activeLogin && cfg.githubAccounts && cfg.githubAccounts[activeLogin]) {
+    const acc = cfg.githubAccounts[activeLogin];
+    if (acc && typeof acc.token === 'string' && acc.token.trim()) {
+      return { token: acc.token.trim(), source: 'config', user: acc.user };
+    }
+    return { token: null, source: null };
+  }
+  if (cfg.github && cfg.github.token && typeof cfg.github.token === 'string') {
+    return { token: cfg.github.token.trim(), source: 'config', user: cfg.github.user };
+  }
+
+  // 3. Check the GitHub CLI (`gh`).
+  try {
+    const r = spawnSync('gh', ['auth', 'token'], { encoding: 'utf-8', stdio: ['ignore', 'pipe', 'ignore'] });
+    if (r.status === 0 && r.stdout && r.stdout.trim()) {
+      return { token: r.stdout.trim(), source: 'gh-cli' };
+    }
+  } catch {
+    // gh not installed or error
+  }
+
+  return { token: null, source: null };
+}
+
+/**
+ * Save a GitHub account to the config file (supports multiple accounts).
+ * @param {string} token 
+ * @param {object} userInfo 
+ */
+export function saveGitHubToken(token, userInfo = null) {
+  if (!token || typeof token !== 'string' || !token.trim()) return false;
+  const current = getConfig();
+  if (!current.githubAccounts) {
+    current.githubAccounts = {};
+  }
+
+  const login = userInfo?.login || current.github?.user?.login || 'default';
+  current.githubAccounts[login] = {
+    token: token.trim(),
+    savedAt: new Date().toISOString(),
+    user: userInfo ? {
+      login: userInfo.login,
+      name: userInfo.name,
+      avatar_url: userInfo.avatar_url,
+      email: userInfo.email,
+      bio: userInfo.bio,
+      public_repos: userInfo.public_repos,
+      total_private_repos: userInfo.total_private_repos,
+      html_url: userInfo.html_url
+    } : (current.githubAccounts[login]?.user || null)
+  };
+
+  current.activeGitHubLogin = login;
+  current.github = {
+    token: token.trim(),
+    user: current.githubAccounts[login].user,
+    savedAt: new Date().toISOString()
+  };
+
+  return saveConfig(current);
+}
+
+/**
+ * Remove a GitHub account from the config file.
+ * @param {string} [login] If omitted, removes the active account.
+ */
+export function removeGitHubToken(login = null) {
+  const current = getConfig();
+  const targetLogin = login || current.activeGitHubLogin || current.github?.user?.login;
+
+  if (targetLogin && current.githubAccounts && current.githubAccounts[targetLogin]) {
+    delete current.githubAccounts[targetLogin];
+  }
+
+  if (current.activeGitHubLogin === targetLogin) {
+    const remaining = Object.keys(current.githubAccounts || {});
+    current.activeGitHubLogin = remaining.length > 0 ? remaining[0] : null;
+    if (current.activeGitHubLogin && current.githubAccounts[current.activeGitHubLogin]) {
+      current.github = {
+        token: current.githubAccounts[current.activeGitHubLogin].token,
+        user: current.githubAccounts[current.activeGitHubLogin].user
+      };
+    } else {
+      delete current.github;
+    }
+  } else if (!targetLogin && current.github) {
+    delete current.github;
+  }
+
+  return saveConfig(current);
+}