import os
import chromedriver_autoinstaller
from logging import getLogger
from pyrogram import filters
from misskaty import app
from misskaty.vars import COMMAND_HANDLER
from misskaty.core.decorator.errors import capture_err
from selenium import webdriver

LOGGER = getLogger(__name__)


@app.on_message(filters.command(["pahe"], COMMAND_HANDLER))
@capture_err
async def pahe(_, msg):
    # chromedriver_autoinstaller.install()
    os.chmod("/MissKaty/chromedriver", 755)
    chrome_options = webdriver.ChromeOptions()
    chrome_options.binary_location = "/usr/bin/google-chrome"
    chrome_options.add_argument("--headless")
    chrome_options.add_argument("--no-sandbox")
    chrome_options.add_argument("--disable-dev-shm-usage")
    wd = webdriver.Chrome("/MissKaty/chromedriver", chrome_options=chrome_options)
    wd.get("https://pahe.li/")
    LOGGER.info(wd.page_source)