63 lines
3.4 KiB
Python
63 lines
3.4 KiB
Python
import json, os, re, subprocess, sys
|
|
from pathlib import Path
|
|
PROTO='jarvis.skill.invoke.v1'
|
|
NAMES=('compose.yaml','compose.yml','docker-compose.yaml','docker-compose.yml')
|
|
def fail(c,m): return {'protocol':PROTO,'success':False,'error':{'code':c,'message':m}}
|
|
def ok(d,m='OK',mut=False): return {'protocol':PROTO,'success':True,'data':d,'message':m,'mutated':mut}
|
|
def root():
|
|
p=Path(os.getenv('DOCKGE_STACKS_DIR','/opt/stacks')).resolve()
|
|
if not p.exists() or not p.is_dir(): raise RuntimeError(f'Dockge-Stacks-Verzeichnis fehlt: {p}')
|
|
return p
|
|
def stack_dir(name):
|
|
if not re.fullmatch(r'[A-Za-z0-9_.-]+',name or ''): raise RuntimeError('Ungültiger Stack-Name')
|
|
p=(root()/name).resolve(); r=root()
|
|
if p.parent!=r or not p.is_dir(): raise RuntimeError('Stack nicht gefunden')
|
|
if not any((p/n).is_file() for n in NAMES): raise RuntimeError('Keine Compose-Datei im Stack gefunden')
|
|
return p
|
|
def allowed(name):
|
|
vals=[x.strip() for x in os.getenv('DOCKGE_ALLOWED_STACKS','').split(',') if x.strip()]
|
|
return '*' in vals or name in vals
|
|
def require_allowed(name):
|
|
if not allowed(name): raise RuntimeError('Mutation verweigert: Stack ist nicht in DOCKGE_ALLOWED_STACKS freigegeben')
|
|
def compose_cmd():
|
|
# Prefer modern plugin; fall back to legacy docker-compose.
|
|
try:
|
|
subprocess.run(['docker','compose','version'],stdout=subprocess.DEVNULL,stderr=subprocess.DEVNULL,check=True,timeout=5)
|
|
return ['docker','compose']
|
|
except Exception:
|
|
return ['docker-compose']
|
|
def run(name,args,timeout=90):
|
|
p=stack_dir(name); cmd=compose_cmd()+args
|
|
cp=subprocess.run(cmd,cwd=p,text=True,capture_output=True,timeout=timeout)
|
|
if cp.returncode!=0: raise RuntimeError((cp.stderr or cp.stdout or f'Exit {cp.returncode}')[-1800:])
|
|
return (cp.stdout or '').strip()
|
|
def main(inv):
|
|
a=inv.get('action'); x=inv.get('input') or {}
|
|
if a=='list_stacks':
|
|
out=[]
|
|
for p in sorted(root().iterdir()):
|
|
if not p.is_dir(): continue
|
|
cf=next((n for n in NAMES if (p/n).is_file()),None)
|
|
if cf: out.append({'name':p.name,'compose_file':cf,'mutation_allowed':allowed(p.name)})
|
|
return ok({'stacks':out,'count':len(out)},'Dockge-Stacks geladen.')
|
|
name=x['stack']
|
|
if a=='stack_status':
|
|
raw=run(name,['ps','--format','json'],30); items=[]
|
|
for line in raw.splitlines():
|
|
try:
|
|
v=json.loads(line)
|
|
items.extend(v if isinstance(v,list) else [v])
|
|
except Exception: pass
|
|
return ok({'stack':name,'services':items,'count':len(items)},'Stack-Status geladen.')
|
|
require_allowed(name)
|
|
if a=='start_stack': run(name,['up','-d','--remove-orphans']); return ok({'stack':name,'action':'start'},'Stack gestartet.',True)
|
|
if a=='stop_stack': run(name,['stop']); return ok({'stack':name,'action':'stop'},'Stack gestoppt.',True)
|
|
if a=='restart_stack': run(name,['restart']); return ok({'stack':name,'action':'restart'},'Stack neu gestartet.',True)
|
|
if a=='update_stack':
|
|
run(name,['pull'],120); run(name,['up','-d','--remove-orphans'],120); return ok({'stack':name,'action':'update'},'Stack aktualisiert.',True)
|
|
return fail('UNKNOWN_ACTION',f'Unbekannte Aktion: {a}')
|
|
try: res=main(json.load(sys.stdin))
|
|
except subprocess.TimeoutExpired: res=fail('DOCKGE_TIMEOUT','Docker-Compose-Aktion hat das Zeitlimit überschritten')
|
|
except Exception as e: res=fail('DOCKGE_ERROR',str(e))
|
|
json.dump(res,sys.stdout,ensure_ascii=False)
|