107 lines
4.4 KiB
Python
Executable File
107 lines
4.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import os
|
|
from functools import partial
|
|
from http import HTTPStatus
|
|
from http.server import SimpleHTTPRequestHandler, ThreadingHTTPServer
|
|
from pathlib import Path
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
PUBLIC = ROOT / 'public'
|
|
DIGEST_ROOT = Path(os.environ.get('DORIS_DIGEST_DIR', '/home/fizzlepoof/.openclaw/workspace/doris-digest'))
|
|
DIGEST_FEEDBACK = DIGEST_ROOT / 'data' / 'feedback.json'
|
|
PAPERLESS_STATE = ROOT / 'data' / 'paperless_review_state.json'
|
|
HOST = os.environ.get('HOST', '0.0.0.0')
|
|
PORT = int(os.environ.get('PORT', '8787'))
|
|
|
|
|
|
def load_json(path: Path, default):
|
|
try:
|
|
return json.loads(path.read_text()) if path.exists() else default
|
|
except Exception:
|
|
return default
|
|
|
|
|
|
def save_json(path: Path, value) -> None:
|
|
path.parent.mkdir(parents=True, exist_ok=True)
|
|
path.write_text(json.dumps(value, indent=2) + '\n')
|
|
|
|
|
|
class Handler(SimpleHTTPRequestHandler):
|
|
def __init__(self, *args, directory=None, **kwargs):
|
|
super().__init__(*args, directory=str(PUBLIC), **kwargs)
|
|
|
|
def do_POST(self):
|
|
length = int(self.headers.get('Content-Length', '0') or '0')
|
|
raw = self.rfile.read(length) if length else b'{}'
|
|
try:
|
|
payload = json.loads(raw.decode('utf-8') or '{}')
|
|
except Exception:
|
|
return self._json(HTTPStatus.BAD_REQUEST, {'ok': False, 'error': 'invalid_json'})
|
|
|
|
if self.path == '/api/topic-feedback':
|
|
return self._topic_feedback(payload)
|
|
if self.path == '/api/topic-feedback/reset':
|
|
return self._topic_reset()
|
|
if self.path == '/api/paperless-action':
|
|
return self._paperless_action(payload)
|
|
return self._json(HTTPStatus.NOT_FOUND, {'ok': False, 'error': 'not_found'})
|
|
|
|
def _topic_feedback(self, payload: dict):
|
|
term = str(payload.get('term', '')).strip()
|
|
vote = str(payload.get('vote', '')).strip().lower()
|
|
if not term or vote not in {'up', 'down'}:
|
|
return self._json(HTTPStatus.BAD_REQUEST, {'ok': False, 'error': 'bad_payload'})
|
|
items = load_json(DIGEST_FEEDBACK, [])
|
|
items = [x for x in items if not (str(x.get('origin', '')) == 'dashboard' and str(x.get('term', '')).strip().lower() == term.lower())]
|
|
items.append({
|
|
'term': term,
|
|
'vote': vote,
|
|
'note': f"Dashboard {'hide' if vote == 'down' else 'prefer'} topic: {term}",
|
|
'origin': 'dashboard',
|
|
'source': str(payload.get('source', '')).strip(),
|
|
'domain': str(payload.get('domain', '')).strip(),
|
|
})
|
|
save_json(DIGEST_FEEDBACK, items)
|
|
return self._json(HTTPStatus.OK, {'ok': True, 'count': len(items)})
|
|
|
|
def _topic_reset(self):
|
|
items = load_json(DIGEST_FEEDBACK, [])
|
|
items = [x for x in items if str(x.get('origin', '')) != 'dashboard']
|
|
save_json(DIGEST_FEEDBACK, items)
|
|
return self._json(HTTPStatus.OK, {'ok': True, 'count': len(items)})
|
|
|
|
def _paperless_action(self, payload: dict):
|
|
item_id = str(payload.get('id', '')).strip()
|
|
action = str(payload.get('action', '')).strip().lower()
|
|
if not item_id or action not in {'needs_attention', 'good_to_go'}:
|
|
return self._json(HTTPStatus.BAD_REQUEST, {'ok': False, 'error': 'bad_payload'})
|
|
state = load_json(PAPERLESS_STATE, {})
|
|
current = state.get(item_id, {}) if isinstance(state, dict) else {}
|
|
current_status = str(current.get('status') or '').lower()
|
|
if action == 'needs_attention':
|
|
new_status = 'active' if current_status == 'needs_attention' else 'needs_attention'
|
|
else:
|
|
new_status = 'good_to_go'
|
|
state[item_id] = {'status': new_status}
|
|
save_json(PAPERLESS_STATE, state)
|
|
return self._json(HTTPStatus.OK, {'ok': True, 'id': item_id, 'status': new_status})
|
|
|
|
def _json(self, status: HTTPStatus, payload: dict):
|
|
body = json.dumps(payload).encode('utf-8')
|
|
self.send_response(status)
|
|
self.send_header('Content-Type', 'application/json; charset=utf-8')
|
|
self.send_header('Content-Length', str(len(body)))
|
|
self.end_headers()
|
|
self.wfile.write(body)
|
|
|
|
def log_message(self, fmt, *args):
|
|
print('%s - - [%s] %s' % (self.address_string(), self.log_date_time_string(), fmt % args))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
handler = partial(Handler, directory=str(PUBLIC))
|
|
ThreadingHTTPServer((HOST, PORT), handler).serve_forever()
|