#!/bin/python3
# 这个脚本的作用是在部署helm chart时动态生成adapter的配置文件，保存在configmap中
# 需要动态生成的原因是core服务的DNS名称是动态的，无法在adapter服务的配置文件中提前确定
# 一些与k8s现有资源冲突的配置也会在这里重置

import os
import toml
import base64
from kubernetes import client, config
from datetime import datetime, timezone

config.load_incluster_config()
core_api = client.CoreV1Api()
apps_api = client.AppsV1Api()

# 读取部署的关键信息
namespace = os.getenv("NAMESPACE")
release_name = os.getenv("RELEASE_NAME")
data_b64 = os.getenv("DATA_B64")

# 解析并覆盖关键配置
# 这里被覆盖的配置应当在helm chart中针对对应的k8s资源来灵活修改
data = toml.loads(base64.b64decode(data_b64).decode("utf-8"))
data.setdefault('napcat_server', {})
data['napcat_server']['host'] = '0.0.0.0'
data['napcat_server']['port'] = 8095
data.setdefault('maibot_server', {})
data['maibot_server']['host'] = f'{release_name}-maibot-core'  # 根据release名称动态拼接core服务的DNS名称
data['maibot_server']['port'] = 8000

# 创建/修改configmap
cm_name = f'{release_name}-maibot-adapter'
cm = client.V1ConfigMap(
    metadata=client.V1ObjectMeta(name=cm_name),
    data={'config.toml': toml.dumps(data)}
)
try:
    core_api.create_namespaced_config_map(namespace, cm)
    print(f"ConfigMap `{cm_name}` created successfully")
except client.exceptions.ApiException as e:
    if e.status == 409:  # 已存在，更新
        core_api.replace_namespaced_config_map(cm_name, namespace, cm)
        print(f"ConfigMap `{cm_name}` replaced successfully")
    else:
        raise

# 重启adapter和core的statefulset
now = datetime.now(timezone.utc).isoformat()
body = {
    "spec": {
        "template": {
            "metadata": {
                "annotations": {
                    "kubectl.kubernetes.io/restartedAt": now
                }
            }
        }
    }
}
apps_api.patch_namespaced_stateful_set(
    name=f'{release_name}-maibot-adapter',
    namespace=namespace,
    body=body,
)
apps_api.patch_namespaced_stateful_set(
    name=f'{release_name}-maibot-core',
    namespace=namespace,
    body=body,
)
print(f"StatefulSet `{release_name}-maibot-adapter` restarted successfully")

print('Job succeed.')