Upload app_v2_entry.py

app_v2_entry.py

@@ -10,10 +10,10 @@ except Exception as e:
 from fastapi.responses import HTMLResponse, JSONResponse, FileResponse, Response
 from fastapi.staticfiles import StaticFiles
 from starlette.routing import Mount
-from fastapi import Query, Request
+from fastapi import Query, Request, UploadFile, File, Form
 import requests as req
 from bs4 import BeautifulSoup
-import re, html as html_lib, json, threading, time
+import re, html as html_lib, json, threading, time, uuid
 from concurrent.futures import ThreadPoolExecutor, as_completed
 from urllib.parse import quote
 
@@ -236,7 +236,7 @@ def _s_vietnamnet(topic,limit=6):
     try:
         r=req.get(f"https://vietnamnet.vn/tim-kiem?q={quote(topic)}",headers={'User-Agent':'Mozilla/5.0'},timeout=10);soup=BeautifulSoup(r.text,'lxml')
         for a in soup.select('h3 a[href], .vnn-title a')[:limit*2]:
-            t=_clean(a.get_text(strip=True));href=a.get('href','')
+            t=_clean(a.get('title','') or a.get_text(strip=True));href=a.get('href','')
             if t and len(t)>15 and _has_kw(topic,t):
                 if not href.startswith('http'):href='https://vietnamnet.vn'+href
                 items.append({'title':t,'url':href,'via':'VietNamNet'})
@@ -415,7 +415,6 @@ _xlb_cache = {}
 _xlb_lock = threading.Lock()
 
 def _xlb_scrape(path):
-    """Scrape xemlaibongda.top and extract video articles. Multi-strategy parsing."""
     url = f"https://xemlaibongda.top/{path}"
     r = req.get(url, headers={"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36"}, timeout=15, allow_redirects=True)
     if r.status_code != 200:
@@ -430,7 +429,6 @@ def _xlb_scrape(path):
         seen.add(href)
         if not href.startswith('http'):
             href = 'https://xemlaibongda.top' + href
-        # Image: search <img> in anchor, then parent up to 3 levels
         img = a.select_one('img')
         p = a.parent
         for _ in range(4):
@@ -446,7 +444,6 @@ def _xlb_scrape(path):
                 img_src = 'https:' + img_src
             elif img_src.startswith('/'):
                 img_src = 'https://xemlaibongda.top' + img_src
-        # Title: try heading/title tags inside anchor first
         title = ''
         for sel in ['.title', 'h3', 'h2', '.name', '.post-title', '.entry-title', '.video-title']:
             t = a.select_one(sel)
@@ -459,14 +456,12 @@ def _xlb_scrape(path):
             img_alt = a.select_one('img')
             if img_alt:
                 title = _clean(img_alt.get('alt',''))
-        # Fallback: use parent's short text
         if not title:
             parent = a.parent
             if parent:
                 pt = _clean(parent.get_text(' ',strip=True))
                 if 5 < len(pt) < 120:
                     title = pt
-        # Skip if still no meaningful title
         if not title or len(title) < 3:
             continue
         vids.append({"link": href, "img": img_src, "title": title})
@@ -476,12 +471,11 @@ def _xlb_scrape(path):
 
 @app.get('/api/proxy/xlb')
 def proxy_xlb(path: str = Query(default="")):
-    """Proxy for xemlaibongda.top to avoid CORS in browser."""
     now = time.time()
     cache_key = f"xlb:{path}"
     with _xlb_lock:
         cached = _xlb_cache.get(cache_key)
-        if cached and now - cached['t'] < 120:  # 2 min cache
+        if cached and now - cached['t'] < 120:
             return JSONResponse(cached['d'])
     try:
         vids = _xlb_scrape(path)
@@ -514,8 +508,14 @@ def _wl(eid:int):return JSONResponse(scrape_lineups(eid))
 def _wm(eid:int):return JSONResponse(scrape_match_detail(eid))
 
 DATA_DIR='/data' if os.path.isdir('/data') else os.path.join(os.path.dirname(os.path.abspath(__file__)),'data')
-os.makedirs(DATA_DIR,exist_ok=True);IF=os.path.join(DATA_DIR,'interactions_v2.json');CF=os.path.join(DATA_DIR,'comments_v2.json')
-_il=threading.Lock();_cl=threading.Lock()
+os.makedirs(DATA_DIR,exist_ok=True)
+IF=os.path.join(DATA_DIR,'interactions_v2.json')
+CF=os.path.join(DATA_DIR,'comments_v2.json')
+WALL_FILE=os.path.join(DATA_DIR,'wall_posts.json')
+WALL_VIDEO_DIR=os.path.join(DATA_DIR,'wall_videos')
+os.makedirs(WALL_VIDEO_DIR,exist_ok=True)
+
+_il=threading.Lock();_cl=threading.Lock();_wl_lock=threading.Lock()
 def _lj(p):
     try:
         if os.path.exists(p):return json.load(open(p,'r',encoding='utf-8'))
@@ -548,6 +548,176 @@ async def _pc(request:Request):
     with _il:idb=_lj(IF);idb.setdefault(v,{'views':0,'likes':0,'comments':0});idb[v]['comments']=len(cms);_sj(IF,idb)
     return JSONResponse({'comments':cms})
 
+# ===== WALL / SHORT AI ENDPOINTS =====
+
+def _load_wall_posts():
+    """Load wall posts from JSON file."""
+    with _wl_lock:
+        return _lj(WALL_FILE)
+
+def _save_wall_posts(posts):
+    """Save wall posts to JSON file."""
+    with _wl_lock:
+        _sj(WALL_FILE, posts)
+
+@app.get('/api/wall')
+def api_wall():
+    """Get all wall posts."""
+    posts = _load_wall_posts()
+    if not posts:
+        # Return empty list, not error
+        return JSONResponse({"posts": []})
+    return JSONResponse({"posts": posts})
+
+@app.post('/api/wall')
+async def api_wall_post(request: Request):
+    """
+    Create a wall post. Supports:
+    - JSON body: {title, text, img, source}
+    - Multipart form: title, text, source + video file upload
+    """
+    content_type = request.headers.get('content-type', '')
+
+    # Handle multipart upload (video file)
+    if 'multipart/form-data' in content_type:
+        try:
+            form = await request.form()
+        except Exception as e:
+            return JSONResponse({"error": f"Form parse error: {str(e)}"}, status_code=400)
+
+        title = form.get('title', 'Video mới') or 'Video mới'
+        text = form.get('text', '') or ''
+        source = form.get('source', 'vtv_recorder') or 'vtv_recorder'
+        video_file = form.get('video')
+
+        post_id = str(uuid.uuid4())[:12]
+        video_url = None
+
+        # Save video file if provided
+        if video_file and hasattr(video_file, 'filename') and video_file.filename:
+            # Determine extension
+            fname = video_file.filename.lower()
+            if fname.endswith('.mp4'):
+                ext = '.mp4'
+            elif fname.endswith('.webm'):
+                ext = '.webm'
+            else:
+                ext = '.webm'
+
+            video_filename = f"wall_{post_id}{ext}"
+            video_path = os.path.join(WALL_VIDEO_DIR, video_filename)
+
+            try:
+                # Read file content
+                content = await video_file.read()
+                if not content:
+                    return JSONResponse({"error": "Empty video file"}, status_code=400)
+
+                # Save to disk
+                with open(video_path, 'wb') as f:
+                    f.write(content)
+
+                file_size_mb = len(content) / 1024 / 1024
+                if file_size_mb > 50:
+                    os.remove(video_path)
+                    return JSONResponse({"error": f"Video quá lớn ({file_size_mb:.1f}MB). Tối đa 50MB."}, status_code=400)
+
+                # URL to access the video
+                video_url = f"/api/wall/video/{video_filename}"
+            except Exception as e:
+                return JSONResponse({"error": f"Lỗi lưu video: {str(e)}"}, status_code=500)
+
+        # Create post
+        post = {
+            "id": post_id,
+            "title": title[:200],
+            "text": text[:2000],
+            "source": source,
+            "video": video_url,
+            "img": None,
+            "images": [],
+            "created": int(time.time()),
+            "created_str": time.strftime('%H:%M %d/%m/%Y', time.localtime()),
+        }
+
+        # Save to wall
+        posts = _load_wall_posts()
+        if not isinstance(posts, list):
+            posts = []
+        posts.insert(0, post)
+        # Keep max 200 posts
+        posts = posts[:200]
+        _save_wall_posts(posts)
+
+        return JSONResponse({"post": post, "ok": True})
+
+    # Handle JSON body (text-only post)
+    try:
+        body = await request.json()
+    except:
+        body = {}
+
+    title = body.get('title', 'Bài mới') or 'Bài mới'
+    text = body.get('text', '') or ''
+    img = body.get('img', None)
+    source = body.get('source', 'user') or 'user'
+
+    post_id = str(uuid.uuid4())[:12]
+    post = {
+        "id": post_id,
+        "title": title[:200],
+        "text": text[:2000],
+        "source": source,
+        "video": None,
+        "img": img,
+        "images": [],
+        "created": int(time.time()),
+        "created_str": time.strftime('%H:%M %d/%m/%Y', time.localtime()),
+    }
+
+    posts = _load_wall_posts()
+    if not isinstance(posts, list):
+        posts = []
+    posts.insert(0, post)
+    posts = posts[:200]
+    _save_wall_posts(posts)
+
+    return JSONResponse({"post": post, "ok": True})
+
+@app.get('/api/wall/video/{filename}')
+def api_wall_video(filename: str):
+    """Serve a wall video file."""
+    # Security: prevent path traversal
+    if '..' in filename or '/' in filename:
+        return Response(status_code=403)
+    video_path = os.path.join(WALL_VIDEO_DIR, filename)
+    if not os.path.exists(video_path):
+        return Response(status_code=404)
+    ext = os.path.splitext(filename)[1].lower()
+    media_type = 'video/mp4' if ext == '.mp4' else 'video/webm'
+    return FileResponse(video_path, media_type=media_type)
+
+@app.delete('/api/wall/{post_id}')
+def api_wall_delete(post_id: str):
+    """Delete a wall post and its video."""
+    posts = _load_wall_posts()
+    if not isinstance(posts, list):
+        return JSONResponse({"error": "No posts"}, status_code=404)
+
+    for i, p in enumerate(posts):
+        if p.get('id') == post_id:
+            # Delete video file if exists
+            if p.get('video'):
+                video_name = p['video'].split('/')[-1]
+                video_path = os.path.join(WALL_VIDEO_DIR, video_name)
+                if os.path.exists(video_path):
+                    os.remove(video_path)
+            posts.pop(i)
+            _save_wall_posts(posts)
+            return JSONResponse({"ok": True})
+
+    return JSONResponse({"error": "Post not found"}, status_code=404)
+
 def _bg():
     time.sleep(15)
     while True: