import base64
import json
import os
import sys

import requests

server_id = 'aef5ac0d-d5a3-49ee-b02f-c31eeb063f9b'
url = f"ksowpscloudsvr://start=RelayHttpServer&serverId={server_id}"

os.startfile(url)


def base64_encode(data):
    return base64.b64encode(json.dumps(data).encode('utf-8')).decode('utf-8')


if __name__ == '__main__':

    if len(sys.argv) < 2:
        sys.exit(1)

    command = sys.argv[1].lower()

    if command == "install":
        cmd = 'enable'

    elif command == "uninstall":
        cmd = 'disable'

    else:
        print(f"未知命令: {command}")

    payload = {'data': {"cmd": cmd, "name": "jx_office", "url": "http://120.195.49.22:7212/", "addonType": "wps",
                        "online": "true"},
               'serverId': server_id}

    payload['data'] = base64_encode(payload['data'])

    headers = {
        'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/133.0.0.0 Safari/537.36',
        'content-type': 'text/plain;charset=UTF-8'
    }

    resp = requests.post('http://localhost:58890/deployaddons/runParams', data=json.dumps(payload), headers=headers)

    print(resp.text)