feat: 워크스페이스 시스템 + 통합 프롬프트 + Docs 기록 관리
- workspace.py: 채널별 워크스페이스 모델 + JSON 영속 저장 - discord_bot.py: /workspace 슬래시 커맨드 (set/git/vikunja/info/remove/list) - 등록 채널만 자동 응답, 미등록 채널 무시 - Git/Vikunja 미설정 시 작업 차단 + 안내 - 통합 프롬프트 1회 호출 (router+planner+chat 통합) - docs_manager.py: Wiki 인덱스, 세션 기록, Changelog 자동 업데이트 - task_pipeline.py: 모든 Gemini 호출에 docs 컨텍스트 주입, 완료 시 기록 - unified.md: 분류+즉답/계획 통합 프롬프트
This commit is contained in:
180
core/workspace.py
Normal file
180
core/workspace.py
Normal file
@@ -0,0 +1,180 @@
|
||||
"""Workspace Manager — 채널별 프로젝트 설정 관리.
|
||||
|
||||
워크스페이스는 JSON 파일로 영속 저장됩니다.
|
||||
각 디스코드 채널은 하나의 워크스페이스에 바인딩됩니다.
|
||||
"""
|
||||
|
||||
import json
|
||||
import logging
|
||||
from pathlib import Path
|
||||
from dataclasses import dataclass, field, asdict
|
||||
from typing import Optional
|
||||
|
||||
logger = logging.getLogger("variet.workspace")
|
||||
|
||||
# 워크스페이스 설정 파일 경로
|
||||
WORKSPACES_FILE = Path(__file__).parent.parent / "workspaces.json"
|
||||
|
||||
|
||||
@dataclass
|
||||
class GitConfig:
|
||||
url: str = ""
|
||||
token: str = ""
|
||||
repo: str = "" # "Owner/RepoName"
|
||||
branch: str = "main"
|
||||
|
||||
@property
|
||||
def is_configured(self) -> bool:
|
||||
return bool(self.url and self.token)
|
||||
|
||||
|
||||
@dataclass
|
||||
class VikunjaConfig:
|
||||
url: str = ""
|
||||
token: str = ""
|
||||
project_id: int = 0
|
||||
|
||||
@property
|
||||
def is_configured(self) -> bool:
|
||||
return bool(self.url and self.token and self.project_id)
|
||||
|
||||
|
||||
@dataclass
|
||||
class Workspace:
|
||||
name: str
|
||||
path: str
|
||||
channel_id: int
|
||||
git: GitConfig = field(default_factory=GitConfig)
|
||||
vikunja: VikunjaConfig = field(default_factory=VikunjaConfig)
|
||||
docs_path: str = "docs/wiki"
|
||||
|
||||
@property
|
||||
def is_ready(self) -> bool:
|
||||
"""Git + Vikunja 모두 설정되었는지."""
|
||||
return self.git.is_configured and self.vikunja.is_configured
|
||||
|
||||
@property
|
||||
def missing_configs(self) -> list[str]:
|
||||
"""미설정 항목 목록."""
|
||||
missing = []
|
||||
if not self.git.is_configured:
|
||||
missing.append("Git")
|
||||
if not self.vikunja.is_configured:
|
||||
missing.append("Vikunja")
|
||||
return missing
|
||||
|
||||
@property
|
||||
def abs_docs_path(self) -> Path:
|
||||
return Path(self.path) / self.docs_path
|
||||
|
||||
def to_dict(self) -> dict:
|
||||
return {
|
||||
"name": self.name,
|
||||
"path": self.path,
|
||||
"channel_id": self.channel_id,
|
||||
"git": asdict(self.git),
|
||||
"vikunja": asdict(self.vikunja),
|
||||
"docs_path": self.docs_path,
|
||||
}
|
||||
|
||||
@classmethod
|
||||
def from_dict(cls, data: dict) -> "Workspace":
|
||||
return cls(
|
||||
name=data["name"],
|
||||
path=data["path"],
|
||||
channel_id=data["channel_id"],
|
||||
git=GitConfig(**data.get("git", {})),
|
||||
vikunja=VikunjaConfig(**data.get("vikunja", {})),
|
||||
docs_path=data.get("docs_path", "docs/wiki"),
|
||||
)
|
||||
|
||||
|
||||
class WorkspaceManager:
|
||||
"""워크스페이스 CRUD + 영속 저장."""
|
||||
|
||||
def __init__(self, config_path: Path = WORKSPACES_FILE):
|
||||
self.config_path = config_path
|
||||
self.workspaces: dict[int, Workspace] = {} # channel_id → Workspace
|
||||
self._load()
|
||||
|
||||
def _load(self):
|
||||
"""JSON에서 워크스페이스 로드."""
|
||||
if not self.config_path.exists():
|
||||
self.workspaces = {}
|
||||
return
|
||||
|
||||
try:
|
||||
data = json.loads(self.config_path.read_text(encoding="utf-8"))
|
||||
self.workspaces = {
|
||||
int(ch_id): Workspace.from_dict(ws)
|
||||
for ch_id, ws in data.items()
|
||||
}
|
||||
logger.info(f"워크스페이스 {len(self.workspaces)}개 로드됨")
|
||||
except Exception as e:
|
||||
logger.error(f"워크스페이스 로드 실패: {e}")
|
||||
self.workspaces = {}
|
||||
|
||||
def _save(self):
|
||||
"""JSON으로 워크스페이스 저장."""
|
||||
data = {
|
||||
str(ch_id): ws.to_dict()
|
||||
for ch_id, ws in self.workspaces.items()
|
||||
}
|
||||
self.config_path.write_text(
|
||||
json.dumps(data, ensure_ascii=False, indent=2),
|
||||
encoding="utf-8",
|
||||
)
|
||||
|
||||
def set_workspace(self, channel_id: int, name: str, path: str) -> Workspace:
|
||||
"""채널에 워크스페이스 등록."""
|
||||
ws = Workspace(name=name, path=path, channel_id=channel_id)
|
||||
self.workspaces[channel_id] = ws
|
||||
self._save()
|
||||
logger.info(f"워크스페이스 설정: #{channel_id} → {name} ({path})")
|
||||
return ws
|
||||
|
||||
def get_workspace(self, channel_id: int) -> Optional[Workspace]:
|
||||
"""채널의 워크스페이스 조회."""
|
||||
return self.workspaces.get(channel_id)
|
||||
|
||||
def is_workspace_channel(self, channel_id: int) -> bool:
|
||||
"""이 채널이 워크스페이스로 등록되어 있는지."""
|
||||
return channel_id in self.workspaces
|
||||
|
||||
def set_git(self, channel_id: int, url: str, token: str,
|
||||
repo: str = "", branch: str = "main") -> bool:
|
||||
"""워크스페이스에 Git 설정."""
|
||||
ws = self.workspaces.get(channel_id)
|
||||
if not ws:
|
||||
return False
|
||||
|
||||
ws.git = GitConfig(url=url, token=token, repo=repo, branch=branch)
|
||||
self._save()
|
||||
logger.info(f"Git 설정: {ws.name} → {url}")
|
||||
return True
|
||||
|
||||
def set_vikunja(self, channel_id: int, url: str, token: str,
|
||||
project_id: int) -> bool:
|
||||
"""워크스페이스에 Vikunja 설정."""
|
||||
ws = self.workspaces.get(channel_id)
|
||||
if not ws:
|
||||
return False
|
||||
|
||||
ws.vikunja = VikunjaConfig(url=url, token=token, project_id=project_id)
|
||||
self._save()
|
||||
logger.info(f"Vikunja 설정: {ws.name} → {url} (project {project_id})")
|
||||
return True
|
||||
|
||||
def remove_workspace(self, channel_id: int) -> bool:
|
||||
"""워크스페이스 제거."""
|
||||
if channel_id in self.workspaces:
|
||||
name = self.workspaces[channel_id].name
|
||||
del self.workspaces[channel_id]
|
||||
self._save()
|
||||
logger.info(f"워크스페이스 제거: {name}")
|
||||
return True
|
||||
return False
|
||||
|
||||
def list_all(self) -> list[Workspace]:
|
||||
"""전체 워크스페이스 목록."""
|
||||
return list(self.workspaces.values())
|
||||
Reference in New Issue
Block a user