from argparse import ArgumentParser
from getpass import getpass
from hashlib import sha1 as _sha1
from json import dumps as _dumps, load
from math import ceil
from random import random
from urllib.parse import urlencode
from requests import Session


def stringify(json):
    return _dumps(json, ensure_ascii=False, separators=(",", ":"))


def sha1(data: str):
    f = _sha1()
    f.update(data.encode())
    return f.hexdigest()


class Client():
    MYSHOP_FORAPP_KEY = "987654321"
    SOFTWARE = "0"
    LOGINDEVICETYPE = "android"
    FROMANDROID = "1"
    VERSION = "131"
    DEVICEID = "设备Id获取失败,null值!版本\u003d131"

    def __init__(self):
        self._ses = Session()
        self._ses.headers.update({
            "User-Agent": "okhttp/3.6.0",
            "Accept-Encoding": "gzip",
        })
        self._baseUrl = "http://39.104.60.22:6023/"
        self._token = ""
        self._fromteacher = False
        self._portalUrl = None

    def check_in(self, longitude: float, latitude: float, localname, note):
        if self._portalUrl is None:
            raise Exception("请先选择医院")
        if not self._token:
            raise Exception("请先登录")
        url = self._portalUrl + "doctor_train/rest/ningboschedule/signstudent.do"  # noqa: E501
        url += f"?{self.get_base_params()}"
        d = stringify({
            "logindevicetype": self.LOGINDEVICETYPE,
            "longitude": f"{longitude:.06f}",
            "latitude": f"{latitude:.06f}",
            "fromandroid": self.FROMANDROID,
            "note": note,
            "fromteacher": "1" if self._fromteacher else "0",
            "localname": localname,
        })
        re = self._ses.post(url, data={"data": d})
        if re.status_code != 200:
            raise Exception(f"HTTP {re.status_code} {re.reason}")
        return self.check_error2(re.json())

    def check_error(self, r):
        if r["code"] != "1" or "data" not in r or r["data"] is None:
            raise Exception(f"Error: {r['msg']}")
        return r["data"]

    def check_error2(self, r):
        if r["code"] != "1":
            raise Exception(f"Error: {r['msg']}")
        return r

    def get_base_params(self):
        d = {
            "token": self._token,
            "myshop_app_key": self.MYSHOP_FORAPP_KEY,
            "software": self.SOFTWARE,
        }
        return urlencode(d)

    def get_ningbo_schedule_init_data(self):
        if self._portalUrl is None:
            raise Exception("请先选择医院")
        if not self._token:
            raise Exception("请先登录")
        url = self._portalUrl + "doctor_train/rest/ningboschedule/getInitData.do"  # noqa: E501
        url += f"?{self.get_base_params()}"
        d = stringify({
            "logindevicetype": self.LOGINDEVICETYPE,
            "fromteacher": "1" if self._fromteacher else "0",
            "fromandroid": self.FROMANDROID,
        })
        re = self._ses.post(url, data={"data": d})
        if re.status_code != 200:
            raise Exception(f"HTTP {re.status_code} {re.reason}")
        return self.check_error2(re.json())

    def login_check(self, loginid: str, password: str):
        if self._portalUrl is None:
            raise Exception("请先选择医院")
        url = self._portalUrl + "doctor_portal/rest/loginCheck.do"
        url += f"?{self.get_base_params()}"
        d = stringify({
            "logindevicetype": self.LOGINDEVICETYPE,
            "password": sha1(password),
            "fromandroid": self.FROMANDROID,
            "devicetype": self.LOGINDEVICETYPE,
            "deviceid": self.DEVICEID,
            "fromteacher": "1" if self._fromteacher else "0",
            "loginid": loginid,
            "mac": self.DEVICEID,
        })
        re = self._ses.post(url, data={"data": d})
        if re.status_code != 200:
            raise Exception(f"HTTP {re.status_code} {re.reason}")
        return self.check_error2(re.json())

    def init_with_hospital(self, hospital):
        self._portalUrl = f'http://{hospital["portalurl"]}/'

    def init_with_user(self, user):
        self._token = user["token"]

    def query_train_hospital(self):
        url = self._baseUrl + "cloud_doctor_train/rest/trainHospital/query.do"
        url += f"?{self.get_base_params()}"
        d = stringify({
            "logindevicetype": self.LOGINDEVICETYPE,
            "fromteacher": "1" if self._fromteacher else "0",
            "fromandroid": self.FROMANDROID,
        })
        re = self._ses.post(url, data={"data": d})
        if re.status_code != 200:
            raise Exception(f"HTTP {re.status_code} {re.reason}")
        return self.check_error(re.json())


class Main():
    def __init__(self, config, arg):
        self._config = config
        self._arg = arg
        self._client = Client()

    def ask_choice(self, choices: list, prompt="请选择：", fn=None, extra=None):
        count = len(choices)
        total_pages = ceil(count / self._arg.page_size)
        page = 1

        def show_page():
            nonlocal page
            base = (page - 1) * self._arg.page_size
            if total_pages > 1:
                print(f"第{page}/{total_pages}页")
            for i in range(self._arg.page_size):
                index = base + i
                if index >= count:
                    break
                s = fn(choices[index]) if fn else choices[index]
                print(f"{i}. {s}")
            if page > 1:
                print("f. 第一页")
                print("p. 上一页")
            if page < total_pages:
                print("n. 下一页")
                print("l. 最后一页")
            if extra is not None:
                for t in extra:
                    print(f"{t[0]}. {t[1]}")

        while True:
            show_page()
            s = input(prompt)
            if s == "f":
                page = 1
            elif s == "p":
                page = max(1, page - 1)
            elif s == "n":
                page = min(total_pages, page + 1)
            elif s == "l":
                page = total_pages
            else:
                if extra is not None:
                    for t in extra:
                        if t[0] == s:
                            return t[2]
                try:
                    index = int(s)
                except Exception:
                    continue
                base = (page - 1) * self._arg.page_size
                index += base
                if index < 0 or index >= count:
                    continue
                return choices[index]

    def checkIn(self):
        self.init_client()
        if "location" not in self._config:
            self._config["location"] = []
        location = self.ask_choice(self._config["location"], "请选择位置：", lambda x: f'{x["name"]} ({x["longitude"]},{x["latitude"]},{x["localname"]})', [("a", "添加位置", "add")])  # noqa: E501
        if location == "add":
            name = input("请输入位置名称（仅用作标识，不会上传到服务器）：")
            longitude = float(input("请输入经度："))
            latitude = float(input("请输入纬度："))
            localname = input("请输入地名（会在签到时上传到服务器）：")
            location = {
                "name": name,
                "longitude": longitude,
                "latitude": latitude,
                "localname": localname,
                "longitude_radius": 0.00001,
                "latitude_radius": 0.00001,
            }
            while True:
                self.show_location(location)
                s = input("是否需要进行调整？(y/n)")
                if s == "n":
                    break
                elif s == "y":
                    t = {
                        "name": "位置名称",
                        "longitude": "经度",
                        "latitude": "纬度",
                        "localname": "地名",
                        "longitude_radius": "随机化经度半径",
                        "latitude_radius": "随机化纬度半径",
                    }
                    c = self.ask_choice(["name", "longitude", "latitude",
                                         "localname", "longitude_radius",
                                         "latitude_radius"], "请选择要修改的内容：",
                                        lambda x: t[x])
                    if c == "name":
                        location["name"] = input("请输入位置名称：")
                    elif c == "longitude":
                        location["longitude"] = float(input("请输入经度："))
                    elif c == "latitude":
                        location["latitude"] = float(input("请输入纬度："))
                    elif c == "localname":
                        location["localname"] = input("请输入地名：")
                    elif c == "longitude_radius":
                        location["longitude_radius"] = float(
                            input("请输入随机化经度半径："))
                    elif c == "latitude_radius":
                        location["latitude_radius"] = float(
                            input("请输入随机化纬度半径："))
                else:
                    continue
            self._config["location"].append(location)
        print("获取签到信息...")
        data = self._client.get_ningbo_schedule_init_data()
        print("已签到列表：")
        for x in data["signlist"]:
            print(x)
        prompt = f"要进行{data['buttonname']}吗？(y/n)"
        while True:
            s = input(prompt)
            if s == "y":
                break
            elif s == "n":
                return
        localname = location["localname"]
        longitude = round(location["longitude"] + (random() - 0.5) * location["longitude_radius"], 6)  # noqa: E501
        latitude = round(location["latitude"] + (random() - 0.5) * location["latitude_radius"], 6)  # noqa: E501
        note = ""
        if self._arg.note:
            note = self._arg.note
        while True:
            print("经度：", longitude)
            print("纬度：", latitude)
            print("地名：", localname)
            print("备注：", note)
            s = input("是否需要进行调整？(y/n)")
            if s == "n":
                break
            elif s == "y":
                t = {
                    "longitude": "经度",
                    "latitude": "纬度",
                    "localname": "地名",
                    "note": "备注",
                }
                c = self.ask_choice(["longitude", "latitude", "localname",
                                     "note"], "请选择要修改的内容：", lambda x: t[x])
                if c == "longitude":
                    longitude = float(input("请输入经度："))
                elif c == "latitude":
                    latitude = float(input("请输入纬度："))
                elif c == "localname":
                    localname = input("请输入地名：")
                elif c == "note":
                    note = input("请输入备注：")
        print("签到中...")
        self._client.check_in(longitude, latitude, localname, note)
        print("签到成功")

    def get_hospital(self):
        print("获取医院列表...")
        data = self._client.query_train_hospital()
        if self._arg.hospital_id:
            for x in data:
                if x["trainhospitalid"] == self._arg.hospital_id:
                    return x
            raise Exception(f"医院ID {self._arg.hospital_id} 不存在")
        if self._arg.hospital_name:
            for x in data:
                if x["name"] == self._arg.hospital_name:
                    return x
            raise Exception(f"医院名称 {self._arg.hospital_name} 不存在")
        if self._arg.hospital_code:
            for x in data:
                if x["hospitalcode"] == self._arg.hospital_code:
                    return x
            raise Exception(f"医院编码 {self._arg.hospital_code} 不存在")
        return self.ask_choice(data, "请选择医院：", lambda x: x["name"])

    def init_client(self):
        if "hospital" not in self._config or "user" not in self._config:
            self.login()
        else:
            self._client.init_with_hospital(self._config["hospital"])
            self._client.init_with_user(self._config["user"])

    def login(self):
        if "hospital" not in self._config or self._arg.change_hospital:
            self._config["hospital"] = self.get_hospital()
        self._client.init_with_hospital(self._config["hospital"])
        if self._arg.login_id is None:
            self._arg.login_id = input("请输入登录ID：")
        if self._arg.password is None:
            if self._arg.no_getpass:
                self._arg.password = input("请输入密码：")
            else:
                self._arg.password = getpass("请输入密码：")
        print("登录中...")
        self._config["user"] = self._client.login_check(
            self._arg.login_id, self._arg.password)
        self._client.init_with_user(self._config["user"])
        print("登录成功")

    def show_location(self, loc):
        print("位置名称：", loc["name"])
        print("经度：", loc["longitude"])
        print("纬度：", loc["latitude"])
        print("地名：", loc["localname"])
        print("随机化经度半径：", loc["longitude_radius"])
        print("随机化纬度半径：", loc["latitude_radius"])


ACTIONS = ["login", "checkin"]
EACTIONS = ["login2"]


def get_action_str(action):
    if action == "login":
        return "登录"
    elif action == "login2":
        return "登录（重新选择医院）"
    elif action == "checkin":
        return "签到"


def main(args=None):
    p = ArgumentParser()
    p.add_argument("-c", "--config", default="zyys.json", help="配置文件")
    p.add_argument("-l", "--login-id", help="登录ID")
    p.add_argument("-p", "--password", help="密码")
    p.add_argument("--hospital-id", help="医院ID", type=int)
    p.add_argument("--hospital-name", help="医院名称")
    p.add_argument("--hospital-code", help="医院编码")
    p.add_argument("-P", "--page-size", default=10, help="每页数量", type=int)
    p.add_argument("-C", "--change-hospital", action="store_true",
                   help="重新选择医院")
    p.add_argument("--no-getpass", action="store_true", help="不使用getpass")
    p.add_argument("-n", "--note", help="备注")
    p.add_argument("action", choices=ACTIONS, help="操作", nargs="?")
    arg = p.parse_args(args)
    try:
        with open(arg.config) as f:
            config = load(f)
    except Exception:
        config = {}
    m = Main(config, arg)
    if arg.action is None:
        arg.action = m.ask_choice(ACTIONS + EACTIONS, "请选择操作：", get_action_str)
    try:
        if arg.action == "login":
            m.login()
        elif arg.action == "login2":
            arg.change_hospital = True
            m.login()
        elif arg.action == "checkin":
            m.checkIn()
    finally:
        with open(arg.config, "w") as f:
            f.write(stringify(config))


if __name__ == "__main__":
    main()