scripts/modeset2.py

#!/usr/bin/env python3
#
#       Ferit Yiğit BALABAN <f@fybx.dev>, 2022
#
from datetime import datetime as dt
from subprocess import run
import asyncio
import random
import sys
import os


START_NIGHT = "22.30"
START_DAY = "8.20"
PATH_SCPT_KEYBRD = "$HOME/scripts/keyboard"
PATH_RESC_VOLUME = "$HOME/.config/navi/volume"
PATH_RESC_KBDLGT = "$HOME/.config/navi/keyboard"
PATH_RESC_SCRLGT = "$HOME/.config/navi/screen"
PATH_RESC_LIGHTW = "$HOME/sources/wallpapers/light/"
PATH_RESC_DARKW  = "$HOME/sources/wallpapers/dark/"
PATH_RESC_WALLPS = "$HOME/.config/navi/wallpapers"
VAR_KBDNAME = "asus::kbd_backlight"


async def set_brightness(device: int, value: int, save_state = False):
    state_kbdlgt = get_brightness(1)
    state_scrlgt = get_brightness(0)
    if value == -1:
        with open(os.path.expandvars(PATH_RESC_SCRLGT if device == 0 else PATH_RESC_KBDLGT), 'r') as f:
            value = int(f.read())
            f.close()
    command = f"brightnessctl set {value}%" if device == 0 else f"brightnessctl --device {VAR_KBDNAME} set {value}%"
    await open_subprocess(command)
    if save_state:
        with open(os.path.expandvars(PATH_RESC_SCRLGT if device == 0 else PATH_RESC_KBDLGT), 'w') as f:
            f.write(str(state_scrlgt if device == 0 else state_kbdlgt))
            f.close()


async def connect_keyboard():
    command = 'bash ' + os.path.expandvars(PATH_SCRIPT_KEYBRD)
    await open_subprocess(command)


async def set_volume(value: int, save_state = False):
    state = get_volume()
    if value == -1:
        with open(os.path.expandvars(PATH_RESC_VOLUME), 'r') as f:
            value = int(f.read())
            f.close()
    value = 100 if value > 100 else 0 if value < 0 else value
    command = f'pactl set-sink-volume @DEFAULT_SINK@ {value}%'
    await open_subprocess(command)
    if save_state:
        with open(os.path.expandvars(PATH_RESC_VOLUME), 'w') as f:
            f.write(str(state))
            f.close()


async def open_subprocess(cmd: str):
    p = await asyncio.create_subprocess_shell(
            cmd,
            stdout=asyncio.subprocess.PIPE,
            stderr=asyncio.subprocess.PIPE)
    stdout, stderr = await p.communicate()
    return p.returncode, stdout, stderr


async def change_wallpaper(mode: int, cringe = False):
    if not os.path.exists(PATH_RESC_WALLPS):
        get_wallpapers():    
    region = f"{mode}{1 if cringe else 0}"
    with open(PATH_RESC_WALLPS, 'r') as f:
        file = f.read()
        f.close()
    list = file.split(region)[1].split("EOR")[0].lstrip('\n').splitlines()
    command = f"wal -i {random.choice(list)}"
    await open_subprocess(command)


def get_wallpapers():
    l0 = os.listdir(PATH_RESC_LIGHTW)
    d0 = os.listdir(PATH_RESC_DARKW)
    l0.remove('cringe')
    d0.remove('cringe')
    l0 = [PATH_RESC_LIGHTW + x + '\n' for x in l0]
    d0 = [PATH_RESC_DARKW + x + '\n' for x in d0]
    l1 = os.listdir(PATH_RESC_LIGHTW + "cringe/")
    d1 = os.listdir(PATH_RESC_DARKW + "cringe/")
    l1 = [PATH_RESC_LIGHTW + "cringe/" + x + '\n' for x in l1]
    d1 = [PATH_RESC_DARKW + "cringe/" + x + '\n' for x in d1]
    with open(PATH_RESC_WALLPS, 'w') as f:
        f.write("00:\n")
        f.writelines(l0)
        f.write("EOR\n01:\n")
        f.writelines(l1)
        f.write("EOR\n10:\n")
        f.writelines(d0)
        f.write("EOR\n11:\n")
        f.writelines(d1)
        f.write("EOR")
        f.close()


def get_brightness(device: int):
    if device == 0:
        cmd = ['brightnessctl']
    elif device == 1:
        cmd = ['brightnessctl', '--device', VAR_KBDNAME]
    return int(run(cmd, text=True, capture_output=True).stdout.split('(')[1].split(')')[0].replace('%', ''))
    

def get_volume():
    r = run(["pactl", "list"], text=True, capture_output=True)
    for x in r.stdout.split("Sink #0")[1].split("Base Volume:")[0].split(' '):
        if '%' in x:
            return int(x.replace('%', ''))


def log(message: str):
    with open(os.path.expandvars("$HOME/navi.log"), 'a') as f:
        f.write(f"[{dt.now().strftime('%m/%d/%y-%H.%M.%S')}] {message}\n")
        f.close()


def get_hour():
    if 0 <= dt.now().minute and dt.now().minute <= 9:
        return f"{dt.now().hour}.0{dt.now().minute}"
    return f"{dt.now().hour}.{dt.now().minute}"


def get_hour_spec(hour_str = None):
    if hour_str != None:
        return (int(hour_str.split('.')[0]) * 60) + int(hour_str.split('.')[0])
    else:
        return (dt.now().hour * 60) + dt.now().minute


def get_mode():
    low = get_hour_spec(START_DAY)
    now = get_hour_spec()
    hgh = get_hour_spec(START_NIGHT)
    return 0 if low <= now and now < hgh else 1


def expand_vars():
    global PATH_SCPT_KEYBRD
    global PATH_RESC_VOLUME
    global PATH_RESC_KBDLGT
    global PATH_RESC_SCRLGT
    global PATH_RESC_LIGHTW
    global PATH_RESC_DARKW 
    global PATH_RESC_WALLPS
    PATH_SCPT_KEYBRD = os.path.expandvars(PATH_SCPT_KEYBRD)
    PATH_RESC_VOLUME = os.path.expandvars(PATH_RESC_VOLUME)
    PATH_RESC_KBDLGT = os.path.expandvars(PATH_RESC_KBDLGT)
    PATH_RESC_SCRLGT = os.path.expandvars(PATH_RESC_SCRLGT)
    PATH_RESC_LIGHTW = os.path.expandvars(PATH_RESC_LIGHTW)
    PATH_RESC_DARKW = os.path.expandvars(PATH_RESC_DARKW)
    PATH_RESC_WALLPS = os.path.expandvars(PATH_RESC_WALLPS)


def main():
    sys.argv.remove(sys.argv[0])
    sys.argv.reverse()
    expand_vars()
    if len(sys.argv) == 1:
        if sys.argv[0] == "--login":
            log("modeset2 started with \"--login\"")
            asyncio.run(connect_keyboard())
            asyncio.run(set_volume(0))
            if mode == 0:
                set_brightness(0, 70)
                set_brightness(1, 0)
            else:
                set_brightness(0, 40)
                set_brightness(1, 100)
            asyncio.run(change_wallpaper(mode))
        elif sys.argv[0] == "--lock":
            log("modeset2 started with \"--lock\"")
            asyncio.run(set_volume(0, save_state = True))
            asyncio.run(set_brightness(0, 0, save_state = True))
            asyncio.run(set_brightness(1, 0, save_state = True))
        elif sys.argv[0] == "--unlock":
            log("modeset2 started with \"--unlock\"")
            print("Unlock")
        elif sys.argv[0] == "--shutdown":
            log("modeset2 started with \"--shutdown\"")
            print("Shutdown")
    elif len(sys.argv) == 0: 
        print("modeset2 by fyb")
        print(f"local machine time:  {get_hour()}")
        print(f"current mode is:     {get_mode()}")
        print(f"current sink volume: {get_volume()}")
        print("""Available options:
1. Login
2. Lock
3. Unlock
4. Shutdown""")
    else:
        print("Issuing more than 1 argument to modeset2 is not supported yet. Aborting...")
        exit(1)
    exit(0)


if __name__ == '__main__':
    main()