import copy
import time
from datetime import datetime, timedelta

rfid_type = ["TEXT", "TEXT", "FLOAT", "INT32"]
air_type = ["TEXT", "FLOAT", "FLOAT", "INT32"]
else_type = ["TEXT", "FLOAT", "INT32"]
rfid_measurement = ["iccid", "RFID", "temperature"  "type"]
air_measurement = ["iccid", "air_temperature", "air_humidity", "type"]
else_measurement = ["iccid", "value", "type"]
dataTypes = {
    0: rfid_type,
    1: air_type,
    2: else_type,
    3: else_type,
    4: else_type,
    5: else_type,
}
measurements = {
    0: rfid_measurement,
    1: air_measurement,
    2: else_measurement,
    3: else_measurement,
    4: else_measurement,
    5: else_measurement,
}
temperature_threshold = [37, 41]
air_temperature_threshold = [0, 40]
air_humidity_threshold = [30, 90]
danqi_threshold = [1.24]
jiawan_threshold = [3000]
zaoyin_threshold = [75]
yanwu_threshold = [200]
# example
# RFID
send1 = {
    "m": ["cid", "r", "v"],
    "v": ["abcd", "rfid", 10.62],
    "t": 0,
    "l": 63
}
# 空气
send2 = {
    "m": ["cid", "at", "ah"],
    "v": ["test2", 10.62, 50.22],
    "t": 1,
    "l": 65
}
# 普通
send3 = {
    "m": ["cid", "v"],
    "v": ["test1", 10.62],
    "t": 2,
    "l": 50
}


def has_rfid_sql(rfid):
    sql = f"select deviceId from root.rfid where rfid='{rfid}'"
    send_json = {
        "sql": sql
    }
    return send_json


def get_device_online_sql(iccid, deviceId):
    sql = "select * from root.farm.clientId where time >= 0"
    # 检查iccid是否有值，如果有，添加到SQL语句中
    if iccid:
        sql += f" and iccid = '{iccid}'"
    # 检查deviceId是否有值，如果有，添加到SQL语句中
    if deviceId:
        sql += f" and deviceId = '{deviceId}'"
    send_json = {
        "sql": sql
    }
    return send_json


def get_device_sql(iccid, deviceId):
    sql = "SELECT * FROM root.device where time>=0"
    # 检查iccid是否有值，如果有，添加到SQL语句中
    if iccid:
        sql += f" and iccid = '{iccid}'"
    # 检查deviceId是否有值，如果有，添加到SQL语句中
    if deviceId:
        sql += f" and deviceId = '{deviceId}'"
    send_json = {
        "sql": sql
    }
    return send_json


def data_query_sql(deviceId):
    sql = f"select last * from root.farm.{deviceId}"
    send_json = {
        "sql": sql
    }
    return send_json


def get_warning_sql(deviceId, limit, offset):
    sql = f"select * from root.warning.{deviceId}"
    if limit:
        sql += f" limit {limit}"
    if offset:
        sql += f" offset {offset}"
    send_json = {
        "sql": sql
    }
    return send_json


def rfid_query_sql(rfid):
    sql = f"select deviceId from root.rfid where rfid='{rfid}'"
    send_json = {
        "sql": sql
    }
    return send_json


def rfid_last_query_sql(rfid, deviceId):
    sql = f"select * from root.farm.{deviceId} where RFID='{rfid}' order by time desc limit 1"
    send_json = {
        "sql": sql
    }
    return send_json


def update_register_sql(timestamp, iccid, deviceId, type):
    send_json = {
        "sql": f"insert into root.device(timestamp, iccid, deviceId, type) values({timestamp}, '{iccid}', '{deviceId}', {type})"
    }
    return send_json


def create_register_sql(iccid, deviceId, type):
    send_json = {
        "sql": f"insert into root.device(iccid, deviceId, type) values('{iccid}', '{deviceId}', {type})"
    }
    return send_json


def insert_device_status_sql(iccid, clientid, deviceId):
    send_json = {
        "sql": f"insert into root.farm.clientId(iccid, clientId, is_online, deviceId) values('{iccid}','{clientid}', True, '{deviceId}')"
    }
    return send_json


def message_publish_sql(deviceId, m, t, v):
    send_json = {
        "devices": ["root.farm." + deviceId],
        "timestamps": [int(time.time() * 1000)],
        "measurements_list": [m],
        "data_types_list": [dataTypes[t]],
        "values_list": [v],
        "is_aligned": False
    }
    return send_json


def get_has_clientId_sql(clientid):
    send_json = {
        "sql": f"select timestamp from root.farm.clientId where clientId='{clientid}'"
    }
    return send_json


# 获取空气设备历史7天的报警数据sql
def air_device_past_seven_days_sql(deviceId):
    current_date = datetime.now().date()
    tomorrow = current_date + timedelta(days=1)
    past_6_days_date = current_date - timedelta(days=6)
    sql = f"select MAX_VALUE(air_temperature),MAX_VALUE(air_humidity) from root.warning.{deviceId} group by ([{past_6_days_date}, {tomorrow}), 1d)"
    send_json = {
        "sql": sql
    }
    return send_json

# 获取普通设备历史7天的报警数据sql
def common_device_past_seven_days_sql(deviceId):
    current_date = datetime.now().date()
    tomorrow = current_date + timedelta(days=1)
    past_6_days_date = current_date - timedelta(days=6)
    sql = f"select MAX_VALUE(value) from root.warning.{deviceId} group by ([{past_6_days_date}, {tomorrow}), 1d)"
    send_json = {
        "sql": sql
    }
    return send_json


# 获取设备所有报警数量的sql
def get_device_all_warning_sql(deviceId):
    return {"sql": f"select count(iccid) from root.warning.{deviceId}"}


# 获取设备当天报警数量的sql
def get_device_today_warning_sql(deviceId):
    now = datetime.now().date()
    return {
        "sql": f"select count(iccid) from root.warning.{deviceId} where time >= {now}T00:00:00"
    }


# 改变设备在线状态表的sql
def get_client_change_status_sql(timestamp, status):
    if status:
        return [f"insert into root.farm.clientId(timestamp, is_online) values({timestamp}, True)"]
    else:
        return [f"insert into root.farm.clientId(timestamp, is_online) values({timestamp}, False)"]


# 创建RFID设备的sql
def RFID_template(type, deviceId):
    return [
        f"CREATE TIMESERIES root.{type}.{deviceId}.iccid(cid) WITH datatype=TEXT,ENCODING=PLAIN",
        f"CREATE TIMESERIES root.{type}.{deviceId}.temperature(v) WITH datatype=FLOAT,ENCODING=PLAIN",
        f"CREATE TIMESERIES root.{type}.{deviceId}.RFID(r) WITH datatype=TEXT,ENCODING=PLAIN",
        f"CREATE TIMESERIES root.{type}.{deviceId}.type(t) WITH datatype=INT32,ENCODING=PLAIN",
    ]


# 创建普通设备的sql
def common_template(type, deviceId):
    return [
        f"CREATE TIMESERIES root.{type}.{deviceId}.iccid(cid) WITH datatype=TEXT,ENCODING=PLAIN",
        f"CREATE TIMESERIES root.{type}.{deviceId}.value(v) WITH datatype=FLOAT,ENCODING=PLAIN",
        f"CREATE TIMESERIES root.{type}.{deviceId}.type(t) WITH datatype=INT32,ENCODING=PLAIN",
    ]


# 创建空气设备的sql
def air_template(type, deviceId):
    return [
        f"CREATE TIMESERIES root.{type}.{deviceId}.iccid(cid) WITH datatype=TEXT,ENCODING=PLAIN",
        f"CREATE TIMESERIES root.{type}.{deviceId}.air_temperature(at) WITH datatype=FLOAT,ENCODING=PLAIN",
        f"CREATE TIMESERIES root.{type}.{deviceId}.air_humidity(ah) WITH datatype=FLOAT,ENCODING=PLAIN",
        f"CREATE TIMESERIES root.{type}.{deviceId}.type(t) WITH datatype=INT32,ENCODING=PLAIN",
    ]


# 创建warning表的sql
def warning_sql(deviceId, type):
    template = "warning"
    match type:
        case 0:
            rfid = RFID_template(template, deviceId)
            rfid.append(f"CREATE TIMESERIES root.{template}.{deviceId}.reason(rs) WITH datatype=TEXT,ENCODING=PLAIN")
            rfid.append(f"CREATE TIMESERIES root.{template}.{deviceId}.solve(s) WITH datatype=TEXT,ENCODING=PLAIN")
            return rfid
        case 1:
            air = air_template(template, deviceId)
            air.append(f"CREATE TIMESERIES root.{template}.{deviceId}.reason(rs) WITH datatype=TEXT,ENCODING=PLAIN")
            air.append(f"CREATE TIMESERIES root.{template}.{deviceId}.solve(s) WITH datatype=TEXT,ENCODING=PLAIN")
            return air
        case _:
            common = common_template(template, deviceId)
            common.append(f"CREATE TIMESERIES root.{template}.{deviceId}.reason(rs) WITH datatype=TEXT,ENCODING=PLAIN")
            common.append(f"CREATE TIMESERIES root.{template}.{deviceId}.solve(s) WITH datatype=TEXT,ENCODING=PLAIN")
            return common


# 插入到rfid和deviceId映射关系表
def rfid_deviceId(rfid, deviceId):
    return [
        f"insert into root.rfid(rfid, deviceId) values('{rfid}', '{deviceId}')"
    ]


# 判断是否为报警数据的函数
def is_warning(deviceId, v, t):
    match t:
        case 0:
            temperature = v[1]
            if temperature_threshold[1] <= temperature or temperature <= temperature_threshold[0]:
                return True
        case 1:
            air_temperature = v[1]
            air_humidity = v[2]
            if air_temperature_threshold[1] <= air_temperature or air_temperature <= air_temperature_threshold[0]:
                return True
            if air_humidity_threshold[1] <= air_humidity or air_humidity <= air_humidity_threshold[0]:
                return True
        case 2:
            danqi = v[1]
            if danqi >= danqi_threshold[0]:
                return True
        case 3:
            jiawan = v[1]
            if jiawan > jiawan_threshold[0]:
                return True
        case 4:
            zaoyin = v[1]
            if zaoyin >= zaoyin_threshold[0]:
                return True
        case 5:
            yanwu = v[1]
            if yanwu >= yanwu_threshold[0]:
                return True
    return False


# 插入到warning表的sql
def insert_to_warning_sql(deviceId, v, t):
    ml = copy.deepcopy(measurements[t])
    dt = copy.deepcopy(dataTypes[t])
    ml.append("reason")
    ml.append("solve")
    dt.append("TEXT")
    dt.append("TEXT")
    # 设置原因和解决方案
    # 解决方案可以调用AI进行生成回答
    if t == 0:
        # RFID 温度
        temperature = v[1]
        prompt = f"动物体温为{temperature}，正常吗？如果体温过高或过低，如何解决？"
        if temperature_threshold[1] <= temperature:
            v.append("体温过高")
        elif temperature <= temperature_threshold[0]:
            v.append("体温过低")
    elif t == 1:
        # air 空气、湿度
        air_temperature = v[1]
        air_humidity = v[2]
        prompt = f"室外天气温度为{air_temperature}，室外天气湿度{air_humidity}，正常吗？如果过高或过低，如何解决？"
        if air_temperature_threshold[1] <= air_temperature:
            v.append("室外天气温度偏高")
        elif air_temperature <= air_temperature_threshold[0]:
            v.append("室外天气温度偏低")
        elif air_humidity_threshold[1] <= air_humidity:
            v.append("室外天气湿度偏高")
        elif air_humidity <= air_humidity_threshold[0]:
            v.append("室外天气湿度偏低")
    elif t == 2:
        # danqi 氮气
        danqi = v[1]
        prompt = f"空气中氮气浓度为{danqi}，正常吗？如果过高或过低，如何解决？"
        v.append("氮气浓度偏高")
    elif t == 3:
        # jiawan 甲烷
        jiawan = v[1]
        prompt = f"空气中甲烷浓度为{jiawan}，正常吗？如果过高或过低，如何解决？"
        v.append("甲烷浓度过高")
    elif t == 4:
        # zaoyin 噪音
        zaoyin = v[1]
        prompt = f"噪音分贝为{zaoyin}，正常吗？如果过高或过低，如何解决？"
        v.append("噪音强度过高")
    elif t == 5:
        # yanwu 烟雾
        yanwu = v[1]
        prompt = f"烟雾浓度为{yanwu}，正常吗？如果过高或过低，如何解决？"
        v.append("烟雾浓度偏高")

    v.append("无")
    return {
        "devices": ["root.warning." + deviceId],
        "timestamps": [int(time.time() * 1000)],
        "measurements_list": [ml],
        "data_types_list": [dt],
        "values_list": [v],
        "is_aligned": False
    }


# 数据库创建字段sql语句
def farm_sql(deviceId, type):
    template = "farm"
    match type:
        case 0:
            return RFID_template(template, deviceId)
        case 1:
            return air_template(template, deviceId)
        case _:
            return common_template(template, deviceId)


# 监控视频接口
def get_video_url(username):
    return f"http://rtsp.lihaink.cn/live/xumu_{username}.live.mp4"