claude-controller 0.1.2 → 0.3.0

package/web/static/utils.js

@@ -0,0 +1,131 @@
+/* ═══════════════════════════════════════════════
+   Utility Functions
+   ═══════════════════════════════════════════════ */
+
+function showToast(message, type = 'success') {
+  const container = document.getElementById('toastContainer');
+  const toast = document.createElement('div');
+  const duration = type === 'error' ? 6000 : 3000;
+  toast.className = `toast ${type}`;
+  toast.style.setProperty('--toast-duration', `${duration}ms`);
+  const icon = type === 'success'
+    ? '<svg width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2"><polyline points="20 6 9 17 4 12"/></svg>'
+    : '<svg width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2"><circle cx="12" cy="12" r="10"/><line x1="15" y1="9" x2="9" y2="15"/><line x1="9" y1="9" x2="15" y2="15"/></svg>';
+  toast.innerHTML = `${icon} <span class="toast-msg">${escapeHtml(message)}</span><span class="toast-close">&times;</span>`;
+  toast.addEventListener('click', () => {
+    toast.style.animation = 'toastOut 0.2s ease forwards';
+    setTimeout(() => { if (toast.parentNode) toast.remove(); }, 200);
+  });
+  container.appendChild(toast);
+  setTimeout(() => { if (toast.parentNode) toast.remove(); }, duration);
+}
+
+function escapeHtml(str) {
+  if (!str) return '';
+  const d = document.createElement('div');
+  d.textContent = str;
+  return d.innerHTML;
+}
+
+/** JS 문자열 이스케이프 — onclick 핸들러 내 싱글쿼트 문자열에서 사용 */
+function escapeJsStr(str) {
+  if (!str) return '';
+  return str.replace(/\\/g, '\\\\').replace(/'/g, "\\'");
+}
+
+function truncate(str, len = 60) {
+  if (!str) return '-';
+  return str.length > len ? str.slice(0, len) + '...' : str;
+}
+
+function renderPromptHtml(prompt) {
+  if (!prompt) return '-';
+  const text = truncate(prompt, 200);
+  const escaped = escapeHtml(text);
+  return escaped.replace(/@(\/[^\s,]+|image\d+)/g, (match, ref) => {
+    const isImage = ref.startsWith('image');
+    const label = isImage ? ref : ref.split('/').pop();
+    const icon = '<svg viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2"><path d="M13 2H6a2 2 0 0 0-2 2v16a2 2 0 0 0 2 2h12a2 2 0 0 0 2-2V9z"/><polyline points="13 2 13 9 20 9"/></svg>';
+    return `<span class="prompt-img-chip" title="${escapeHtml('@' + ref)}">${icon}${escapeHtml(label)}</span>`;
+  });
+}
+
+function formatTime(ts) {
+  if (!ts) return '-';
+  try {
+    const d = new Date(ts);
+    if (isNaN(d.getTime())) return ts;
+    const pad = n => String(n).padStart(2, '0');
+    return `${d.getFullYear()}-${pad(d.getMonth()+1)}-${pad(d.getDate())} ${pad(d.getHours())}:${pad(d.getMinutes())}:${pad(d.getSeconds())}`;
+  } catch { return ts; }
+}
+
+function formatCwd(cwd) {
+  if (!cwd) return '-';
+  const parts = cwd.replace(/\/$/, '').split('/');
+  return parts[parts.length - 1] || cwd;
+}
+
+function formatElapsed(startTs) {
+  if (!startTs) return '--:--';
+  const start = new Date(startTs).getTime();
+  if (isNaN(start)) return '--:--';
+  const elapsed = Math.max(0, Math.floor((Date.now() - start) / 1000));
+  const h = Math.floor(elapsed / 3600);
+  const m = Math.floor((elapsed % 3600) / 60);
+  const s = elapsed % 60;
+  const pad = n => String(n).padStart(2, '0');
+  if (h > 0) return `${h}:${pad(m)}:${pad(s)}`;
+  return `${pad(m)}:${pad(s)}`;
+}
+
+function formatFileSize(bytes) {
+  if (bytes < 1024) return bytes + ' B';
+  if (bytes < 1024 * 1024) return (bytes / 1024).toFixed(1) + ' KB';
+  return (bytes / (1024 * 1024)).toFixed(1) + ' MB';
+}
+
+function getFileExt(filename) {
+  const dot = filename.lastIndexOf('.');
+  return dot >= 0 ? filename.slice(dot + 1).toUpperCase() : '?';
+}
+
+/* ── Desktop Notification ── */
+function requestNotificationPermission() {
+  if ('Notification' in window && Notification.permission === 'default') {
+    Notification.requestPermission();
+  }
+}
+
+function notifyJobDone(jobId, status, prompt) {
+  if (!('Notification' in window) || Notification.permission !== 'granted') return;
+  if (document.hasFocus()) return;
+  const title = status === 'done' ? `Job #${jobId} 완료` : `Job #${jobId} 실패`;
+  const body = truncate(prompt || '', 80);
+  const icon = status === 'done'
+    ? 'data:image/svg+xml,<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 100 100"><text y="80" font-size="80">%E2%9C%85</text></svg>'
+    : 'data:image/svg+xml,<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 100 100"><text y="80" font-size="80">%E2%9D%8C</text></svg>';
+  try {
+    const n = new Notification(title, { body, icon, tag: `job-${jobId}` });
+    n.onclick = () => { window.focus(); toggleJobExpand(String(jobId)); n.close(); };
+    setTimeout(() => n.close(), 8000);
+  } catch { /* silent */ }
+}
+
+/* ── Duration formatting ── */
+function formatDuration(durationMs) {
+  if (durationMs == null) return '';
+  const sec = durationMs / 1000;
+  return sec < 60 ? `${sec.toFixed(1)}s` : `${Math.floor(sec / 60)}m ${Math.round(sec % 60)}s`;
+}
+
+/* ── Theme ── */
+function applyTheme(theme) {
+  document.documentElement.setAttribute('data-theme', theme);
+  localStorage.setItem('theme', theme);
+}
+
+function toggleTheme() {
+  const current = document.documentElement.getAttribute('data-theme') || 'dark';
+  applyTheme(current === 'dark' ? 'light' : 'dark');
+}

package/web/webhook.py

@@ -0,0 +1,210 @@
+"""
+Controller Service — Webhook 전달 모듈
+
+작업 완료/실패 시 등록된 URL로 결과를 POST한다.
+bash(lib/jobs.sh)에서 직접 호출 가능:
+  python3 /path/to/webhook.py <job_id> <status>
+"""
+
+import hashlib
+import hmac
+import json
+import os
+import sys
+import time
+import urllib.request
+import urllib.error
+from pathlib import Path
+
+
+# 경로 설정 — web/ 기준
+_WEB_DIR = Path(__file__).resolve().parent
+_CONTROLLER_DIR = _WEB_DIR.parent
+_DATA_DIR = _CONTROLLER_DIR / "data"
+_SETTINGS_FILE = _DATA_DIR / "settings.json"
+_LOGS_DIR = _CONTROLLER_DIR / "logs"
+_WEBHOOK_SENT_DIR = _DATA_DIR / "webhook_sent"
+
+
+def _load_settings():
+    try:
+        if _SETTINGS_FILE.exists():
+            return json.loads(_SETTINGS_FILE.read_text("utf-8"))
+    except (json.JSONDecodeError, OSError):
+        pass
+    return {}
+
+
+def _build_payload(job_id, status):
+    """작업의 메타 + 결과를 읽어서 웹훅 페이로드를 생성한다."""
+    meta_file = _LOGS_DIR / f"job_{job_id}.meta"
+    out_file = _LOGS_DIR / f"job_{job_id}.out"
+
+    meta = {}
+    if meta_file.exists():
+        try:
+            for line in meta_file.read_text().splitlines():
+                if "=" in line:
+                    k, v = line.split("=", 1)
+                    meta[k.strip()] = v.strip().strip("'\"")
+        except OSError:
+            pass
+
+    # 결과 추출
+    result_text = None
+    cost_usd = None
+    duration_ms = None
+    session_id = None
+    is_error = False
+
+    if out_file.exists():
+        try:
+            for line in out_file.read_text().splitlines():
+                if '"type":"result"' not in line:
+                    continue
+                try:
+                    obj = json.loads(line)
+                    if obj.get("type") == "result":
+                        result_text = obj.get("result")
+                        cost_usd = obj.get("total_cost_usd")
+                        duration_ms = obj.get("duration_ms")
+                        session_id = obj.get("session_id")
+                        is_error = obj.get("is_error", False)
+                except json.JSONDecodeError:
+                    continue
+        except OSError:
+            pass
+
+    return {
+        "event": f"job.{status}",
+        "timestamp": time.strftime("%Y-%m-%dT%H:%M:%SZ", time.gmtime()),
+        "job": {
+            "job_id": job_id,
+            "status": status,
+            "prompt": meta.get("PROMPT", ""),
+            "cwd": meta.get("CWD") or None,
+            "created_at": meta.get("CREATED_AT", ""),
+            "session_id": session_id or meta.get("SESSION_ID") or None,
+            "result": result_text,
+            "cost_usd": cost_usd,
+            "duration_ms": duration_ms,
+            "is_error": is_error,
+        },
+    }
+
+
+def _sign_payload(payload_bytes, secret):
+    """HMAC-SHA256 서명을 생성한다."""
+    return hmac.new(secret.encode("utf-8"), payload_bytes, hashlib.sha256).hexdigest()
+
+
+def deliver_webhook(job_id, status):
+    """설정에 webhook_url이 있으면 결과를 POST한다.
+
+    Returns:
+        dict | None: 전송 결과 또는 None (미설정/중복)
+    """
+    settings = _load_settings()
+    webhook_url = (settings.get("webhook_url") or "").strip()
+    if not webhook_url:
+        return None
+
+    # 이벤트 필터: webhook_events가 설정되어 있으면 해당 이벤트만 전송
+    webhook_events = settings.get("webhook_events", "done,failed")
+    allowed_events = {e.strip() for e in webhook_events.split(",")}
+    if status not in allowed_events:
+        return None
+
+    # 중복 전송 방지
+    _WEBHOOK_SENT_DIR.mkdir(parents=True, exist_ok=True)
+    sent_marker = _WEBHOOK_SENT_DIR / f"{job_id}_{status}"
+    if sent_marker.exists():
+        return None
+
+    payload = _build_payload(job_id, status)
+    payload_bytes = json.dumps(payload, ensure_ascii=False).encode("utf-8")
+
+    headers = {
+        "Content-Type": "application/json; charset=utf-8",
+        "User-Agent": "Controller-Webhook/1.0",
+    }
+
+    # HMAC 서명
+    webhook_secret = (settings.get("webhook_secret") or "").strip()
+    if webhook_secret:
+        sig = _sign_payload(payload_bytes, webhook_secret)
+        headers["X-Webhook-Signature"] = f"sha256={sig}"
+
+    try:
+        req = urllib.request.Request(
+            webhook_url, data=payload_bytes, headers=headers, method="POST"
+        )
+        with urllib.request.urlopen(req, timeout=10) as resp:
+            resp_status = resp.status
+
+        # 성공 마커 기록
+        sent_marker.write_text(str(int(time.time())))
+
+        return {
+            "delivered": True,
+            "url": webhook_url,
+            "status_code": resp_status,
+            "event": payload["event"],
+        }
+    except (urllib.error.URLError, OSError, ValueError) as e:
+        return {
+            "delivered": False,
+            "url": webhook_url,
+            "error": str(e),
+            "event": payload["event"],
+        }
+
+
+def cleanup_sent_markers(max_age_seconds=86400):
+    """오래된 전송 마커를 정리한다 (기본 24시간)."""
+    if not _WEBHOOK_SENT_DIR.exists():
+        return
+    now = time.time()
+    for f in _WEBHOOK_SENT_DIR.iterdir():
+        try:
+            if now - f.stat().st_mtime > max_age_seconds:
+                f.unlink()
+        except OSError:
+            pass
+
+
+# ════════════════════════════════════════════════
+#  DAG 디스패치 — 작업 완료 시 후속 pending 작업 자동 실행
+# ════════════════════════════════════════════════
+
+def _dispatch_pending_after_completion():
+    """작업 완료 후 의존성이 충족된 pending 작업을 디스패치한다."""
+    try:
+        sys.path.insert(0, str(_WEB_DIR))
+        from jobs import dispatch_pending_jobs
+        dispatched = dispatch_pending_jobs()
+        if dispatched:
+            print(f"DAG dispatch: {dispatched}", file=sys.stderr)
+    except Exception as e:
+        print(f"DAG dispatch error: {e}", file=sys.stderr)
+
+
+# CLI 진입점: python3 webhook.py <job_id> <status>
+if __name__ == "__main__":
+    if len(sys.argv) < 3:
+        print(f"Usage: {sys.argv[0]} <job_id> <done|failed>", file=sys.stderr)
+        sys.exit(1)
+
+    _job_id = sys.argv[1]
+    _status = sys.argv[2]
+
+    if _status not in ("done", "failed"):
+        print(f"Invalid status: {_status}", file=sys.stderr)
+        sys.exit(1)
+
+    result = deliver_webhook(_job_id, _status)
+    if result:
+        print(json.dumps(result, ensure_ascii=False))
+
+    # 작업 완료 후 DAG 의존성 체인 처리
+    _dispatch_pending_after_completion()