# -*- coding: utf-8 -*-
import logging
import os
import requests

from sqlalchemy import text

from website import consts
from website import db
from website import errors
from website import settings
from website.handler import APIHandler, authenticated

class ListHandler(APIHandler):
    """
    - 描述： 模型运行库列表
    - 请求方式：post
    - 请求参数：
    > - pageNo, int
    > - pageSize, int
    > - name, string, 名称
    - 返回值：
    ```
        {
            "count": 123,
            "data": [
                {
                    "id": 123,
                    "name": "xx",
                    "create_time": "xxx",
                    "update_time": "xxx"
                },
                ...
            ]
        }
    ```
    """
    @authenticated
    def post(self):
        pageNo = self.get_int_argument("pageNo", 1)
        pageSize = self.get_int_argument("pageSize", consts.PAGE_SIZE)
        name = self.get_escaped_argument("name", "")
        
        result = []
        count = 0
        
        with self.app_mysql.connect() as conn:
            sql = "select id, name, create_at, update_at from model_hub where 1=1"
            param = {}

            sql_count = "select count(id) from model where 1=1"
            param_count = {}

            if name:
                sql += "and m.name like :name"
                param["name"] = "%{}%".format(name)

                sql_count += "and m.name like :name"
                param_count["name"] = "%{}%".format(name)

            sql += " order by m.id desc limit :pageSize offset :offset"
            param["pageSize"] = pageSize
            param["offset"] = (pageNo - 1) * pageSize

            cur = conn.execute(text(sql), param)
            result = db.to_json_list(cur)

            count = conn.execute(text(sql_count), param_count).fetchone()[0]

        data = []
        for item in result:
            data.append({
                "id": item["id"],
                "name": item["name"],
                "create_time": item["create_at"].strftime("%Y-%m-%d %H:%M:%S"),
                "update_time": item["update_at"].strftime("%Y-%m-%d %H:%M:%S")
            })

        self.finish({"count": count, "data": data})
        


class SyncHandler(APIHandler):
    """
    - 描述： 查询docker registry中的镜像
    - 请求方式：post
    - 请求参数：
    > - host, string, ip地址
    > - port, int, 端口
    - 返回值：
    ```
        {
            "data": [
                "xxx",    # docker registry中docker images的地址
                "xxx",
                ...
            ]
        }
```
    """
    @authenticated
    def post(self):
        host = self.get_escaped_argument("host", "")
        port = self.get_int_argument("port")
        if not host or not port:
            raise errors.HTTPAPIError(errors.ERROR_BAD_REQUEST, "host and port must be provided.")
        
        images = []
        # 查询docker registry中的镜像
        repositories = requests.get("http://{}:{}/v2/_catalog".format(host, port)).json()["repositories"]
        for repository in repositories:
            # 查询docker registry中的镜像的tag
            tags = requests.get("http://{}:{}/v2/{}/tags/list".format(host, port, repository)).json()["tags"]
            for tag in tags:
                image_name = "{}:{}/{}:{}".format(host, port, repository, tag)
                images.append(image_name)
        self.finish({"data": images})



class AddHandler(APIHandler):
    """
    - 描述： 新建模型运行库
    - 请求方式：post
    - 请求参数：
    > - name, string, 名称
    > - host, string,
    > - port, int
    > - path, string, 镜像路径
    > - comment, string, 备注
    - 返回值：无
    """
    @authenticated
    def post(self):
        name = self.get_escaped_argument("name", "")
        host = self.get_escaped_argument("host", "")
        port = self.get_int_argument("port")
        path = self.get_escaped_argument("path", "")
        comment = self.get_escaped_argument("comment", "")
        if not name or not host or not port:
            raise errors.HTTPAPIError(errors.ERROR_BAD_REQUEST, "name and host and port must be provided.")
        
        with self.app_mysql.connect() as conn:
            conn.execute(text("""insert into model_hub (name, host, port, path, comment, create_at, update_at) 
                              values (:name, :host, :port, :path, :comment, NOW(), NOW())"""),
                          {"name": name, "host": host, "port": port, "path": path, "comment": comment})

            conn.commit()

        self.finish()


class EditHandler(APIHandler):
    """
    - 描述： 编辑模型运行库
    - 请求方式：post
    - 请求参数：
    > - id, int
    > - name, string, 名称
    > - host, string,
    > - port, int
    > - path, string, 镜像路径
    > - comment, string, 备注
    - 返回值：无
    """
    @authenticated
    def post(self):
        id = self.get_int_argument("id")
        name = self.get_escaped_argument("name", "")
        host = self.get_escaped_argument("host", "")
        port = self.get_int_argument("port")
        path = self.get_escaped_argument("path", "")
        comment = self.get_escaped_argument("comment", "")
        if not id or not name or not host or not port or path:
            raise errors.HTTPAPIError(errors.ERROR_BAD_REQUEST, "parameter error")
        with self.app_mysql.connect() as conn:
            conn.execute(text("""update model_hub set name=:name, host=:host, port=:port, path=:path, comment=:comment, update_at=NOW() 
                              where id=:id"""), {"id": id, "name": name, "host": host, "port": port, "path": path, "comment": comment})
            conn.commit()
        self.finish()


class InfoHandler(APIHandler):
    """
    - 描述： 模型运行库信息
    - 请求方式：post
    - 请求参数：
    > - id, int
    - 返回值：
    ```
    {
        "name": "xxx", 
        "host": "xxx", 
        "port": 123, 
        "path": "xxx", 
        "comment": "xxx", 
    }
    ```
    """
    @authenticated
    def post(self):
        hid = self.get_int_argument("id")
        if not id:
            raise errors.HTTPAPIError(errors.ERROR_BAD_REQUEST, "parameter error")

        result = {}
        with self.app_mysql.connect() as conn:
            cur = conn.execute(text("""select name, host, port, path, comment from model_hub where id=:id"""), {"id": hid})
            result = db.to_json(cur)
            if not result:
                raise errors.HTTPAPIError(errors.ERROR_BAD_REQUEST, "model hub not found")
        self.finish({"data": result})


class DeleteHandler(APIHandler):
    """
    - 描述： 删除模型运行库
    - 请求方式：post
    - 请求参数：
    > - id, int
    - 返回值：无
    """
    @authenticated
    def post(self):
        hid = self.get_int_argument("id")
        if not id:
            raise errors.HTTPAPIError(errors.ERROR_BAD_REQUEST, "parameter error")
        with self.app_mysql.connect() as conn:
            conn.execute(text("""delete from model_hub where id=:id"""), {"id": hid})
            conn.commit()
        self.finish()