from nicegui import ui, app import os import asyncio from database import Category, FFmpegProfile, AppConfig from core.state import state # <--- ACESSAMOS O BOT POR AQUI AGORA # --- CONSTANTES --- TMDB_GENRES = { '28': 'Ação (Action)', '12': 'Aventura (Adventure)', '16': 'Animação (Animation)', '35': 'Comédia (Comedy)', '80': 'Crime', '99': 'Documentário', '18': 'Drama', '10751': 'Família (Family)', '14': 'Fantasia (Fantasy)', '36': 'História', '27': 'Terror (Horror)', '10402': 'Música (Music)', '9648': 'Mistério', '10749': 'Romance', '878': 'Ficção Científica (Sci-Fi)', '10770': 'Cinema TV', '53': 'Thriller', '10752': 'Guerra (War)', '37': 'Faroeste (Western)', '10759': 'Action & Adventure (TV)', '10762': 'Kids (TV)', '10765': 'Sci-Fi & Fantasy (TV)' } COMMON_COUNTRIES = { 'JP': '🇯🇵 Japão (Anime)', 'US': '🇺🇸 EUA (Hollywood)', 'BR': '🇧🇷 Brasil', 'KR': '🇰🇷 Coreia do Sul (Dorama)', 'GB': '🇬🇧 Reino Unido', 'ES': '🇪🇸 Espanha', 'FR': '🇫🇷 França', 'CN': '🇨🇳 China', 'IN': '🇮🇳 Índia' } ISO_LANGS = { 'por': 'Português (por)', 'eng': 'Inglês (eng)', 'jpn': 'Japonês (jpn)', 'spa': 'Espanhol (spa)', 'fra': 'Francês (fra)', 'ger': 'Alemão (ger)', 'ita': 'Italiano (ita)', 'rus': 'Russo (rus)', 'kor': 'Coreano (kor)', 'zho': 'Chinês (zho)', 'und': 'Indefinido (und)' } # --- SELETOR DE PASTAS FLEXÍVEL --- async def pick_folder_dialog(start_path='/', allowed_root='/'): """ allowed_root: '/' permite tudo, '/media' restringe. """ # Se start_path for vazio ou None, começa na raiz permitida if not start_path: start_path = allowed_root # Se o caminho salvo não começar com a raiz permitida, reseta if allowed_root != '/' and not start_path.startswith(allowed_root): start_path = allowed_root result = {'path': None} with ui.dialog() as dialog, ui.card().classes('w-96 h-[500px] flex flex-col'): ui.label(f'Selecionar Pasta ({allowed_root})').classes('font-bold text-lg mb-2') path_label = ui.label(start_path).classes('text-xs bg-gray-100 p-2 border rounded w-full break-all font-mono') scroll = ui.scroll_area().classes('flex-grow border rounded p-1 mt-2 bg-white') async def load_dir(path): # Segurança: Garante que não saiu da jaula (se houver jaula) if allowed_root != '/' and not path.startswith(allowed_root): path = allowed_root path_label.text = path scroll.clear() try: # Botão Voltar (..) # Só mostra se não estiver na raiz do sistema OU na raiz permitida show_back = True if path == '/': show_back = False if allowed_root != '/' and path == allowed_root: show_back = False if show_back: parent = os.path.dirname(path) with scroll: ui.button('.. (Voltar)', on_click=lambda: load_dir(parent)).props('flat dense icon=arrow_upward align=left w-full') # Lista Pastas with scroll: if os.path.exists(path): entries = sorted([e for e in os.scandir(path) if e.is_dir()], key=lambda x: x.name.lower()) for entry in entries: ui.button(entry.name, on_click=lambda p=entry.path: load_dir(p)).props('flat dense icon=folder align=left w-full color=amber-8') else: ui.label('Caminho não encontrado.').classes('text-gray-400 italic p-2') except Exception as e: with scroll: ui.label(f'Acesso negado: {e}').classes('text-red text-xs') def select_this(): result['path'] = path_label.text dialog.close() with ui.row().classes('w-full justify-between mt-auto pt-2'): ui.button('Cancelar', on_click=dialog.close).props('flat color=grey') ui.button('Selecionar Esta', on_click=select_this).props('flat icon=check color=green') await load_dir(start_path) await dialog return result['path'] # --- TELA PRINCIPAL --- def show(): # Helper para pegar o bot de forma segura via State def get_bot(): if state.watcher and state.watcher.bot: return state.watcher.bot return None with ui.column().classes('w-full p-6'): ui.label('Configurações').classes('text-3xl font-light text-gray-800 mb-4') with ui.tabs().classes('w-full') as tabs: tab_ident = ui.tab('Identificação', icon='search') tab_cats = ui.tab('Categorias', icon='category') tab_deploy = ui.tab('Deploy & Caminhos', icon='move_to_inbox') tab_ffmpeg = ui.tab('Motor (FFmpeg)', icon='movie') tab_telegram = ui.tab('Telegram', icon='send') with ui.tab_panels(tabs).classes('w-full bg-gray-50 p-4 rounded border'): # --- ABA 1: IDENTIFICAÇÃO --- with ui.tab_panel(tab_ident): with ui.card().classes('w-full max-w-2xl mx-auto p-6'): ui.label('🔍 Configuração do Identificador').classes('text-2xl font-bold mb-4 text-indigo-600') tmdb_key = AppConfig.get_val('tmdb_api_key', '') ui.input('API Key do TMDb', value=tmdb_key, password=True, on_change=lambda e: AppConfig.set_val('tmdb_api_key', e.value)).classes('w-full mb-4') with ui.grid(columns=2).classes('w-full gap-4'): ui.input('Idioma', value=AppConfig.get_val('tmdb_language', 'pt-BR'), on_change=lambda e: AppConfig.set_val('tmdb_language', e.value)) ui.number('Confiança Auto (%)', value=int(AppConfig.get_val('min_confidence', '90')), min=50, max=100, on_change=lambda e: AppConfig.set_val('min_confidence', str(int(e.value)))) # --- ABA 2: CATEGORIAS (COM EDIÇÃO) --- with ui.tab_panel(tab_cats): ui.label('Regras de Organização').classes('text-xl text-gray-700 mb-2') editing_id = {'val': None} cats_container = ui.column().classes('w-full gap-2') form_card = ui.card().classes('w-full mb-4 bg-gray-100 p-4') def load_cats(): cats_container.clear() cats = list(Category.select()) for cat in cats: g_ids = cat.genre_filters.split(',') if cat.genre_filters else [] c_codes = cat.country_filters.split(',') if cat.country_filters else [] g_names = [TMDB_GENRES.get(gid, gid) for gid in g_ids if gid] c_names = [COMMON_COUNTRIES.get(cc, cc) for cc in c_codes if cc] desc = f"{cat.content_type.upper()}" if g_names: desc += f" | {', '.join(g_names)}" if c_names: desc += f" | {', '.join(c_names)}" color_cls = 'bg-white border-gray-200' if 'Animação' in str(g_names) and 'Japão' in str(c_names): color_cls = 'bg-purple-50 border-purple-200' with cats_container, ui.card().classes(f'w-full flex-row items-center justify-between p-3 border {color_cls}'): with ui.row().classes('items-center gap-4'): icon = 'movie' if cat.content_type == 'movie' else ('tv' if cat.content_type == 'series' else 'shuffle') ui.icon(icon).classes('text-gray-600') with ui.column().classes('gap-0'): ui.label(cat.name).classes('font-bold text-lg') ui.label(desc).classes('text-xs text-gray-500') with ui.row(): ui.button(icon='edit', on_click=lambda c=cat: start_edit(c)).props('flat dense color=blue') ui.button(icon='delete', on_click=lambda c=cat: delete_cat(c)).props('flat dense color=red') def start_edit(cat): editing_id['val'] = cat.id name_input.value = cat.name type_select.value = cat.content_type genre_select.value = cat.genre_filters.split(',') if cat.genre_filters else [] country_select.value = cat.country_filters.split(',') if cat.country_filters else [] form_label.text = f"Editando: {cat.name}" btn_save.text = "Atualizar Regra" btn_save.props('color=blue icon=save') btn_cancel.classes(remove='hidden') form_card.classes(replace='bg-blue-50') def cancel_edit(): editing_id['val'] = None name_input.value = '' genre_select.value = [] country_select.value = [] form_label.text = "Nova Biblioteca" btn_save.text = "Adicionar Regra" btn_save.props('color=green icon=add') btn_cancel.classes(add='hidden') form_card.classes(replace='bg-gray-100') def save_cat(): if not name_input.value: return g_str = ",".join(genre_select.value) if genre_select.value else "" c_str = ",".join(country_select.value) if country_select.value else "" try: if editing_id['val']: Category.update( name=name_input.value, content_type=type_select.value, genre_filters=g_str, country_filters=c_str ).where(Category.id == editing_id['val']).execute() ui.notify('Categoria atualizada!', type='positive') else: Category.create( name=name_input.value, target_path=f"/media/{name_input.value}", content_type=type_select.value, genre_filters=g_str, country_filters=c_str ) ui.notify('Categoria criada!', type='positive') cancel_edit() load_cats() except Exception as e: ui.notify(f'Erro: {e}', type='negative') def delete_cat(cat): cat.delete_instance() if editing_id['val'] == cat.id: cancel_edit() load_cats() with form_card: with ui.row().classes('w-full justify-between items-center'): form_label = ui.label('Nova Biblioteca').classes('font-bold text-gray-700') btn_cancel = ui.button('Cancelar', on_click=cancel_edit).props('flat dense color=red').classes('hidden') with ui.grid(columns=4).classes('w-full gap-2'): name_input = ui.input('Nome') type_select = ui.select({'mixed': 'Misto', 'movie': 'Só Filmes', 'series': 'Só Séries'}, value='mixed', label='Tipo') genre_select = ui.select(TMDB_GENRES, multiple=True, label='Gêneros').props('use-chips') country_select = ui.select(COMMON_COUNTRIES, multiple=True, label='Países').props('use-chips') btn_save = ui.button('Adicionar Regra', on_click=save_cat).props('icon=add color=green').classes('mt-2 w-full') load_cats() # --- ABA 3: DEPLOY (DESTINOS VS ORIGEM) --- with ui.tab_panel(tab_deploy): # ORIGEM (Permite Raiz /) with ui.card().classes('w-full mb-6 border-l-4 border-amber-500 bg-amber-50'): ui.label('📡 Origem dos Arquivos').classes('text-lg font-bold mb-2 text-amber-900') monitor_path = AppConfig.get_val('monitor_path', '/downloads') async def pick_source(): # AQUI: allowed_root='/' permite navegar em tudo (incluindo /downloads) p = await pick_folder_dialog(mon_input.value, allowed_root='/') if p: mon_input.value = p with ui.row().classes('w-full items-center gap-2'): mon_input = ui.input('Pasta Monitorada', value=monitor_path).classes('flex-grow font-mono bg-white rounded px-2') ui.button(icon='folder', on_click=pick_source).props('flat dense color=amber-9') ui.button('Salvar Origem', on_click=lambda: (AppConfig.set_val('monitor_path', mon_input.value), ui.notify('Salvo!'))).classes('mt-2 bg-amber-600 text-white') # DESTINOS (Restrito a /media para organização) ui.label('📂 Mapeamento de Destinos (/media)').classes('text-xl text-gray-700 mt-4') paths_container = ui.column().classes('w-full gap-2') def load_deploy_paths(): paths_container.clear() for cat in list(Category.select()): with paths_container, ui.card().classes('w-full p-4 flex-row items-center gap-4'): ui.label(cat.name).classes('font-bold w-32') path_input = ui.input(value=cat.target_path).classes('flex-grow font-mono') async def pick_dest(i=path_input): # AQUI: allowed_root='/media' trava a navegação p = await pick_folder_dialog(i.value, allowed_root='/media') if p: i.value = p ui.button(icon='folder', on_click=pick_dest).props('flat dense color=amber-8') ui.button(icon='save', on_click=lambda c=cat, p=path_input: (setattr(c, 'target_path', p.value), c.save(), ui.notify('Salvo!'))).props('flat round color=green') load_deploy_paths() # --- ABA 4: FFMPEG --- with ui.tab_panel(tab_ffmpeg): ui.label('Perfis de Conversão').classes('text-xl text-gray-700') profiles_query = list(FFmpegProfile.select()) profiles_dict = {p.id: p.name for p in profiles_query} active_profile = next((p for p in profiles_query if p.is_active), None) def set_active_profile(e): if not e.value: return FFmpegProfile.update(is_active=False).execute() FFmpegProfile.update(is_active=True).where(FFmpegProfile.id == int(e.value)).execute() ui.notify(f'Perfil Ativado!', type='positive') select_profile = ui.select(profiles_dict, value=active_profile.id if active_profile else None, label='Perfil Ativo', on_change=set_active_profile).classes('w-64 mb-6') for p in profiles_query: with ui.expansion(f"{p.name}", icon='tune').classes('w-full bg-white mb-2 border rounded'): with ui.column().classes('p-4 w-full'): with ui.grid(columns=2).classes('w-full gap-4'): c_name = ui.input('Nome', value=p.name) c_codec = ui.select({'h264_vaapi':'VAAPI','libx264':'CPU','copy':'Copy'}, value=p.video_codec, label='Codec') c_preset = ui.select(['fast', 'medium', 'slow'], value=p.preset, label='Preset') c_crf = ui.number('CRF', value=p.crf) c_audio = ui.select(ISO_LANGS, value=p.audio_langs.split(','), multiple=True, label='Áudios').props('use-chips') c_sub = ui.select(ISO_LANGS, value=p.subtitle_langs.split(','), multiple=True, label='Legendas').props('use-chips') def save_profile(prof=p, n=c_name, c=c_codec, pr=c_preset, cr=c_crf, a=c_audio, s=c_sub): prof.name = n.value; prof.video_codec = c.value; prof.preset = pr.value prof.crf = int(cr.value); prof.audio_langs = ",".join(a.value); prof.subtitle_langs = ",".join(s.value) prof.save() profiles_dict[prof.id] = prof.name select_profile.options = profiles_dict select_profile.update() ui.notify('Salvo!') ui.button('Salvar', on_click=save_profile).classes('mt-2') # --- ABA 5: TELEGRAM --- with ui.tab_panel(tab_telegram): with ui.card().classes('w-full max-w-lg mx-auto p-6'): ui.label('🤖 Integração Telegram').classes('text-2xl font-bold mb-4 text-blue-600') token_input = ui.input('Bot Token', value=AppConfig.get_val('telegram_token', '')).props('password').classes('w-full mb-2') chat_input = ui.input('Chat ID', value=AppConfig.get_val('telegram_chat_id', '')).classes('w-full mb-6') async def save_and_restart(): AppConfig.set_val('telegram_token', token_input.value) AppConfig.set_val('telegram_chat_id', chat_input.value) ui.notify('Salvando...', type='warning') bot = get_bot() if bot: await bot.restart() ui.notify('Bot Reiniciado!', type='positive') else: ui.notify('Salvo, mas Bot não está rodando.', type='warning') async def test_connection(): bot = get_bot() if bot: ui.notify('Enviando mensagem...') success = await bot.send_test_msg() if success: ui.notify('Mensagem enviada!', type='positive') else: ui.notify('Falha no envio.', type='negative') else: ui.notify('Bot offline.', type='negative') with ui.row().classes('w-full gap-2'): ui.button('Salvar e Reiniciar', on_click=save_and_restart).props('icon=save color=blue').classes('flex-grow') ui.button('Testar', on_click=test_connection).props('icon=send color=green')