import os
import json
import time

import lark_oapi as lark

from flask import Flask
from dotenv import load_dotenv
from itertools import chain
from lark_oapi.adapter.flask import *
from lark_oapi.api.im.v1 import *
from lark_oapi.api.bitable.v1 import *

from utils import *


load_dotenv()


app_id = os.environ["APP_ID"]
app_secret = os.environ["APP_SECRET"]

encrypt_key = os.environ["ENCRYPT_KEY"]
verification_token = os.environ["VERIFICATION_TOKEN"]

bitable_token = os.environ["BITABLE_TOKEN"]
table_id = os.environ["TABLE_ID"]

app = Flask(__name__)

# 创建client
client = lark.Client.builder().app_id(app_id).app_secret(app_secret).build()

current_dir = os.path.dirname(os.path.abspath(__file__))
cache_dir = os.path.join(current_dir, "cache")

if not os.path.exists(cache_dir):
    os.makedirs(cache_dir)


def do_p2_im_message_receive_v1(data: P2ImMessageReceiveV1) -> None:
    print(lark.JSON.marshal(data))

    # record = {}
    # images = []
    # images_token = []
    # message_content = ""
    # message_type = data.event.message.message_type
    # content = json.loads(data.event.message.content)

    # if message_type == "post":
    #     content = content["content"]
    #     for item in chain.from_iterable(content):
    #         if item.get("tag") == "text":
    #             message_content += item["text"]

    #         if item.get("tag") == "img":
    #             images.append(item["image_key"])

    # if message_type == "text":
    #     message_content = content["text"].replace("@_user_1", "").strip()

    # sender_id = data.event.sender.sender_id.open_id
    # record["问题概述"] = message_content
    # record["创建人"] = sender_id
    # message_id = data.event.message.message_id

    # # 对 image_key 处理并并上传至多维表格

    # for image in images:
    #     image_path = get_message_resource(client, message_id, image, "image", cache_dir)
    #     print("下载文件成功:", image, image_path)
    #     file_token = upload_media(client, image_path, "bitable_image", bitable_token)
    #     print("上传文件成功:", file_token)
    #     images_token.append(file_token)

    # if images:
    #     record["附件"] = [{"file_token": image} for image in images_token]

    # record_id = add_record(client, bitable_token, table_id, record)

    # time.sleep(1)

    # update_data = {"创建人": {"id": sender_id}}
    # update_record(client, bitable_token, table_id, record_id, update_data)

    # # content = {"text": "反馈的问题已经记录到多维表格, 稍后有小伙伴跟进"}
    # # send_message(receive_id, content)

    # # 回复群组的消息
    # content = f'{{"text": "反馈的问题已经记录到多维表格, 稍后有小伙伴跟进"}}'
    # send_group_message(client, message_id, content)


def do_customized_event(data: lark.CustomizedEvent) -> None:
    print(lark.JSON.marshal(data))


handler = (
    lark.EventDispatcherHandler.builder(
        encrypt_key, verification_token, lark.LogLevel.DEBUG
    )
    .register_p2_im_message_receive_v1(do_p2_im_message_receive_v1)
    .register_p1_customized_event("message", do_customized_event)
    .build()
)


@app.route("/event", methods=["POST"])
def event():
    print("receive event: ", time.time())
    resp = handler.do(parse_req())
    return parse_resp(resp)


if __name__ == "__main__":
    app.run(port=8080, debug=True, host="0.0.0.0")