186 lines
7.5 KiB
Python
186 lines
7.5 KiB
Python
from __future__ import annotations
|
|
|
|
import html
|
|
|
|
from twisted.web import resource
|
|
|
|
from DB import Database
|
|
from Hatenatools import TMB
|
|
from hatena import NotFound, as_text, client_ip, request_headers
|
|
from reximemo import current_account, dialog, dsi_url, forward, html_page, keyboard_value
|
|
|
|
|
|
class PyResource(resource.Resource):
|
|
isLeaf = False
|
|
def getChild(self, name, request):
|
|
fsid = as_text(name).upper()
|
|
return CreatorNode(fsid) if Database.CreatorExists(fsid) else NotFound
|
|
|
|
def render(self, request):
|
|
request.setResponseCode(403); return b""
|
|
|
|
|
|
class CreatorNode(resource.Resource):
|
|
isLeaf = False
|
|
def __init__(self, fsid):
|
|
super().__init__(); self.fsid = fsid
|
|
|
|
def getChild(self, name, request):
|
|
text = as_text(name)
|
|
if "." in text:
|
|
base, ext = text.rsplit(".", 1)
|
|
if ext in {"ppm", "tmb", "info", "htm", "star", "dl", "reply"} and Database.FlipnoteExists(self.fsid, base):
|
|
return FlipnoteFile(self.fsid, base, ext)
|
|
if Database.FlipnoteExists(self.fsid, text):
|
|
return FlipnoteActionNode(self.fsid, text)
|
|
return NotFound
|
|
|
|
|
|
class FlipnoteActionNode(resource.Resource):
|
|
isLeaf = False
|
|
def __init__(self, fsid, filename):
|
|
super().__init__(); self.fsid = fsid; self.filename = filename
|
|
def getChild(self, name, request):
|
|
if as_text(name) == "keyboard.kbd":
|
|
return TextComment(self.fsid, self.filename)
|
|
return NotFound
|
|
|
|
|
|
class TextComment(resource.Resource):
|
|
isLeaf = True
|
|
def __init__(self, fsid, filename):
|
|
super().__init__(); self.fsid = fsid; self.filename = filename
|
|
def render_POST(self, request):
|
|
fn = Database.get_flipnote(self.fsid, self.filename)
|
|
if not fn:
|
|
return NotFound.render(request)
|
|
account = current_account(request)
|
|
try:
|
|
Database.add_text_comment(
|
|
fn["id"], keyboard_value(request), ip_address=client_ip(request),
|
|
account_id=(account["id"] if account else None),
|
|
)
|
|
except ValueError as exc:
|
|
return dialog(request, str(exc))
|
|
forward(request, dsi_url(request, f"movie/{self.fsid}/{self.filename}.htm"))
|
|
return b"OK"
|
|
def render_GET(self, request):
|
|
request.setResponseCode(405); return b""
|
|
|
|
|
|
class FlipnoteFile(resource.Resource):
|
|
isLeaf = True
|
|
def __init__(self, fsid, filename, ext):
|
|
super().__init__(); self.fsid = fsid; self.filename = filename; self.ext = ext
|
|
|
|
def render_GET(self, request):
|
|
fn = Database.get_flipnote(self.fsid, self.filename)
|
|
if not fn:
|
|
return NotFound.render(request)
|
|
if self.ext == "ppm":
|
|
Database.AddView(self.fsid, self.filename)
|
|
request.setHeader(b"content-type", b"application/octet-stream")
|
|
return Database.GetFlipnotePPM(self.fsid, self.filename)
|
|
if self.ext == "tmb":
|
|
request.setHeader(b"content-type", b"application/octet-stream")
|
|
return Database.GetFlipnoteTMB(self.fsid, self.filename)
|
|
if self.ext == "info":
|
|
request.setHeader(b"content-type", b"text/plain")
|
|
return b"0\n0\n"
|
|
if self.ext == "dl":
|
|
Database.AddDownload(self.fsid, self.filename)
|
|
return b"OK"
|
|
if self.ext == "star":
|
|
headers = request_headers(request)
|
|
try:
|
|
amount = int(headers.get("x-hatena-star-count", "1"))
|
|
except ValueError:
|
|
amount = 1
|
|
Database.AddStar(self.fsid, self.filename, amount)
|
|
return b"Success"
|
|
if self.ext == "htm":
|
|
return details_page(request, fn)
|
|
request.setResponseCode(405); return b""
|
|
|
|
def render_POST(self, request):
|
|
fn = Database.get_flipnote(self.fsid, self.filename)
|
|
if not fn:
|
|
return NotFound.render(request)
|
|
if self.ext == "reply":
|
|
data = request.content.read()
|
|
tmb = TMB().Read(data)
|
|
if not tmb:
|
|
return dialog(request, "That mini Flipnote is invalid.")
|
|
if int(getattr(tmb, "FrameCount", 1)) != 1:
|
|
return dialog(request, "Mini Flipnote comments must contain one frame.")
|
|
account = current_account(request)
|
|
try:
|
|
Database.add_memo_comment(
|
|
fn["id"], data, ip_address=client_ip(request),
|
|
account_id=(account["id"] if account else None),
|
|
)
|
|
except ValueError as exc:
|
|
return dialog(request, str(exc))
|
|
forward(request, dsi_url(request, f"movie/{self.fsid}/{self.filename}.htm"))
|
|
return b"OK"
|
|
request.setResponseCode(405); return b""
|
|
|
|
|
|
def details_page(request, fn):
|
|
try:
|
|
tmb = TMB().Read(Database.GetFlipnoteTMB(fn["creator_fsid"], fn["filename"]))
|
|
except Exception:
|
|
tmb = None
|
|
creator_name = fn.get("creator_name") or (getattr(tmb, "Username", "") if tmb else "Creator")
|
|
creator_url = dsi_url(request, f"creator/{fn['creator_room_id']}/profile.htm")
|
|
ppm_url = dsi_url(request, f"movie/{fn['creator_fsid']}/{fn['filename']}.ppm")
|
|
star_url = dsi_url(request, f"movie/{fn['creator_fsid']}/{fn['filename']}.star")
|
|
reply_url = dsi_url(request, f"movie/{fn['creator_fsid']}/{fn['filename']}.reply")
|
|
text_url = dsi_url(request, f"movie/{fn['creator_fsid']}/{fn['filename']}/keyboard.kbd")
|
|
extra = (
|
|
'<meta name="upperlink" content="%(ppm)s">'
|
|
'<meta name="savebutton" content="%(ppm)s">'
|
|
'<meta name="playcontrolbutton" content="">'
|
|
'<meta name="starbutton" content="%(star)s">'
|
|
'<meta name="commentbutton" content="%(reply)s">'
|
|
) % {
|
|
"ppm": html.escape(ppm_url, quote=True),
|
|
"star": html.escape(star_url, quote=True),
|
|
"reply": html.escape(reply_url, quote=True),
|
|
}
|
|
comments = Database.list_comments(fn["id"], limit=30)
|
|
comment_html = []
|
|
for c in comments:
|
|
if c["type"] == "memo":
|
|
npf = dsi_url(request, f"comment/{c['id']}.npf?w=128&h=64")
|
|
ppm = dsi_url(request, f"comment/{c['id']}.ppm")
|
|
content = '<a href="%s"><img src="%s" width="128" height="64"></a>' % (html.escape(ppm, quote=True), html.escape(npf, quote=True))
|
|
else:
|
|
content = html.escape(c.get("text_content") or "")
|
|
comment_html.append(
|
|
'<div class="comment"><strong>%s</strong><br>%s</div><div class="hr"></div>' % (
|
|
html.escape(c.get("creator_name") or "Guest"), content,
|
|
)
|
|
)
|
|
body = (
|
|
'<div class="title">%(title)s</div>'
|
|
'<table width="226" class="detail">'
|
|
'<tr><th>Creator</th><td><a href="%(creator_url)s">%(creator)s</a></td></tr>'
|
|
'<tr><th>Stars</th><td>%(stars)d</td></tr>'
|
|
'<tr><th>Views</th><td>%(views)d</td></tr>'
|
|
'</table>'
|
|
'<p><a href="%(ppm)s">Play Flipnote</a></p>'
|
|
'<p><a href="%(text)s">Text Comment</a></p>'
|
|
'<div class="hr"></div><div class="subtitle">Comments</div>%(comments)s'
|
|
) % {
|
|
"title": html.escape(fn.get("title") or "Untitled Flipnote"),
|
|
"creator_url": html.escape(creator_url, quote=True),
|
|
"creator": html.escape(creator_name),
|
|
"stars": int(fn.get("stars") or 0),
|
|
"views": int(fn.get("views") or 0),
|
|
"ppm": html.escape(ppm_url, quote=True),
|
|
"text": html.escape(text_url, quote=True),
|
|
"comments": "".join(comment_html) or '<div class="notice">No comments yet.</div>',
|
|
}
|
|
return html_page(request, fn.get("title") or "Flipnote", body, extra)
|