from flask import Blueprint, request, abort

from utils import dbhelper
from utils.misc import internal_json2protobuf
from protobuf_files import niming_pb2

# prefix: /owner/
bl_owner_article = Blueprint('owner_article', __name__)


# 只有發文者可以看到的獲取指定文章
# 只有發文者可以做到的刪除文章
@bl_owner_article.route("/<type>/<key>", methods = ["GET", "DELETE"])
def owner_getarticle(type:str, key:str):
    # arguments
    sha256 = request.args.get("hash", None)
    if not sha256:
        return abort(400)
    sha256 = str(sha256)
    
    if type == 'article':
        if not (len(key) > 0 and key.isdigit()):
            return abort(400)
        type, key = 'a', int(key) # id
    elif type == 'comment':
        if not (len(key) > 0):
            return abort(400)
        type, key = 'c', str(key) # sha1
    else:
        return abort(404)

    # 獲取指定文章/留言
    if request.method == "GET":
        if type == 'a': # 文章
            resfn, code = dbhelper.solo_article_fetcher("owner", key=key, hash=sha256)
        elif type == 'c': # 留言
            resfn, code = dbhelper.solo_comment_fetcher("owner", key=key, hash=sha256)
        if code == 200: 
            return internal_json2protobuf(role="owner", otype=type, original=[resfn]), code
        return abort(code)
    # 刪除指定文章/留言
    elif request.method == "DELETE":
        if type == 'a':
            rtype = niming_pb2.FetchPostResponse
            result, code = dbhelper.solo_article_remover("owner", hash=sha256, id=key)
        elif type == 'c':
            rtype = niming_pb2.FetchCommentResponse
            result, code = dbhelper.solo_comment_remover("owner", hash=sha256, sha1=key)
        
        if not code == 200: # Exception
            return abort(code)
        
        if type == 'a':
            ret = rtype(posts=[ rtype.Message(id=result["id"]) ])
        elif type == 'c':
            ret = rtype(posts=[ rtype.Message(sha1=result["sha1"]) ])
            
        return ret.SerializeToString(), 200