main.py

# main.py
import os
from shutil import Error

from dotenv import load_dotenv
from textual.binding import Binding
from textual.widgets import Footer

from ui.add_key_view import AddKeyView
from ui.base_screen import BaseScreen
from ui.keys_view import KeysView

load_dotenv()
from textual.app import App
#from ui.login_view import LoginView
#from ui.main_view import MainView
from ui.message_view import MessageView
from vpn_interface.outline_manager import get_outline_access_keys
from managers.data_manager import load_data, save_data



class AdminApp(App):
    TITLE = "Outline Admin Interface"
    BINDINGS = [
        Binding("ctrl+a","add", "Add new key"),
        Binding("ctrl+c", "quit", "Quit app"),
        #Binding("down", "focus_next", "Go down", show=False)
    ]


    def action_quit(self):
        self.exit()

    def action_add(self):
        self.push_screen(AddKeyView())

    def action_keys(self):
        if self.screen == KeysView():
            pass
        self.push_screen(KeysView())

    def compose(self):
        yield Footer()

    async def on_ready(self):
        print("Приложение начало работу")
        from helper.parser import parse_env_json
        api_url, cert = parse_env_json("Json")
        if not api_url or not cert:
            raise Error("api_url and cert weren't found in .env file.")
            #await self.push_screen(LoginView())
        else:
            keys = get_outline_access_keys()
            if keys is None:
                print("Error getting keys.")
                await self.push_screen(
                    MessageView("Error", "Error getting keys. Check API URL and CERT."))
            else:
                self.update_users_yaml_with_keys(keys)
                print("Success")
                await self.push_screen(KeysView())


    def update_users_yaml_with_keys(self, keys):
        data = load_data()
        user = None
        for u in data['users']:
            if u['user_id'] == 'admin':
                user = u
                break
        if not user:
            user = {'user_id': 'admin', 'name': 'admin_user', 'devices': []}
            data['users'].append(user)
        user['devices'].clear()
        for k in keys:
            device = {
                'device_id': k.key_id,
                'device_name': k.name,
                'outline_key': k.access_url
            }
            user['devices'].append(device)
        save_data(data)



if __name__ == "__main__":
    app = AdminApp()
    app.run()