Arcs-MCP

from dataclasses import dataclass
from typing import Annotated, Dict, Any
from base import AbstractClient
from config import HALO_PUBLISH_TOKEN
from environment import get_cookies_from_chromium
from utils.field_metadata_util import Description, ExampleValue


@dataclass
class HaloClient(AbstractClient):
    """ Halo发布参数说明：
        _host: str = None, 主机IP或域名
        _attachment_publish_url: str = None, 附件API
        _pre_publish_url: str = None, 创建并发布API
        _publish_url: str = None, 发布API
        _cookies: Dict = None, cookies
        _attachment_headers: Dict = None, 附件请求头
        _headers: Dict = None, 发布请求头
        _json_data: Dict = None, 发布请求data
    """
    _host: Annotated[
        str,
        Description("Halo主机IP或域名"),
        ExampleValue("192.168.1.189"),
    ] = None

    _cookies: Annotated[
        Dict[str, str],
        Description("Halo服务浏览器cookies")
    ] = None

    _publish_url: Annotated[
        str,
        Description("Halo发布文章API"),
        ExampleValue("http://192.168.1.189/apis/api.console.halo.run/v1alpha1/posts/<uuid>")
    ] = None

    _json_data: Annotated[
        Dict[str, Any],
        Description("Halo发布文章API请求json data格式")
    ] = None

    _token: Annotated[
        str,
        Description("Halo发布文章API的token")
    ] = HALO_PUBLISH_TOKEN

    @property
    def host(self):
        return self._host

    @host.setter
    def host(self, value: str):
        self._host = value

    @property
    def token(self):
        return self._token

    @property
    def attachment_publish_url(self) -> str:
        return 'http://' + self._host + '/apis/api.console.halo.run/v1alpha1/attachments/upload'

    @property
    def pre_publish_url(self) -> str:
        return 'http://' + self._host + '/apis/api.console.halo.run/v1alpha1/posts'

    @property
    def publish_url(self):
        return self._publish_url

    @publish_url.setter
    def publish_url(self, uuid: str):
        self._publish_url = 'http://' + self._host + '/apis/api.console.halo.run/v1alpha1/posts/' + uuid

    @property
    def cookies(self):
        return self._cookies

    @cookies.setter
    def cookies(self, value: str):
        self._cookies = get_cookies_from_chromium(value)

    @property
    def attachment_headers(self):
        return {
            'Accept': '*/*',
            'Accept-Language': 'zh-CN,zh;q=0.9',
            'Connection': 'keep-alive',
            'Authorization': 'Bearer ' + self._token,
            # 'Content-Type': 'multipart/form-data; boundary=----WebKitFormBoundary26w5KbBsjd2c3EwK',
            'Origin': 'http://' + self.host,
            'Referer': 'http://' + self.host + '/console/attachments',
            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/130.0.0.0 Safari/537.36',
        }

    @property
    def headers(self):
        return {
            'Accept': 'application/json, text/plain, */*',
            'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8,en-GB;q=0.7,en-US;q=0.6',
            'Connection': 'keep-alive',
            'Authorization': 'Bearer ' + self._token,
            'Content-Type': 'application/json',
            'Origin': 'http://' + self.host,
            'Referer': 'http://' + self.host + '/console/posts/editor',
            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/130.0.0.0 Safari/537.36',
            'X-Requested-With': 'XMLHttpRequest',
        }

    @property
    def json_data(self):
        return self._json_data

    @json_data.setter
    def json_data(self, value: dict):
        self._json_data = {
            'post': {
                'spec': {
                    'title': value.get('title'),
                    'slug': value.get('slug'),  # 生成文章的固定链接，默认为文章标题的拼音用-分隔
                    'template': '',
                    'cover': '',
                    'deleted': False,
                    'publish': True,  # 是否发布，可以设置为True直接发布，也可以设置为False，在调用publish_url发布
                    'pinned': False,  # 是否置顶
                    'allowComment': True,  # 是否允许评论
                    'visible': 'PUBLIC',  # 可见性：公开
                    'priority': 0,
                    "publishTime": value.get('publish_time'),  # 发布时间，格式 -> "2024-10-30T09:39:10.763909062Z",
                    'excerpt': {
                        'autoGenerate': True,
                        'raw': '',
                    },
                    # 'categories': [  # 分类
                    #     '76514a40-6ef1-4ed9-b58a-e26945bde3ca',
                    # ],
                    # 'tags': [  # 标签
                    #     'c33ceabb-d8f1-4711-8991-bb8f5c92ad7c',
                    # ],
                    'htmlMetas': [],
                    # 'owner': 'lcyrus',
                },
                'apiVersion': 'content.halo.run/v1alpha1',
                'kind': 'Post',
                'metadata': {
                    'name': value.get('uuid'),
                    'annotations': {
                        'content.halo.run/preferred-editor': 'default',
                    },
                },
            },
            'content': {
                'raw': value.get('content'),
                'content': value.get('content'),
                'rawType': 'HTML',
            },
        }


# client = HaloClient()  # 打印类信息
# print(dir(client))