import urllib.request
import json
import sys

def verify_code(sms_code):
    url = 'https://api-video.goodline.info/ords/mobile/vc2/auth/token/sms'

    # Данные из твоего скриншота
    payload = {
        "phone": "79134374202",
        "code": str(sms_code)
    }

    data = json.dumps(payload).encode('utf-8')
    headers = {
        'Content-Type': 'application/json',
        'origin': 'https://video.online-dozor.ru',
        'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64)'
    }

    req = urllib.request.Request(url, data=data, headers=headers, method='POST')

    try:
        with urllib.request.urlopen(req, timeout=10) as response:
            resp_json = json.loads(response.read().decode('utf-8'))
            new_token = resp_json.get('TOKEN')

            if new_token:
                # Записываем свежий токен в текстовый файл
                with open('/config/scripts/goodline_token.txt', 'w') as f:
                    f.write(new_token)
                print(f"Успех! Новый токен сохранен: {new_token[:10]}...")
            else:
                print("Ошибка: Токен не найден в ответе сервера!")

    except Exception as e:
        print(f"Ошибка при отправке кода: {e}")

if __name__ == "__main__":
    # Скрипт принимает код из аргумента командной строки
    if len(sys.argv) > 1:
        verify_code(sys.argv[1])
    else:
        print("Ошибка: Не передан СМС код!")