import os
import subprocess
import socket
from libqtile import layout, bar, widget, hook
from libqtile.config import Group, Screen
from libqtile.dgroups import simple_key_binder
from libqtile.config import Key
from libqtile.command import lazy
from libqtile.config import Click, Drag
from colors import dracula
    

# var
hostname = socket.gethostname()

# Groups and layouts
groups = []
group_names = ["1", "2", "3", "4", "5", "6", "7", "8", "9", "0", ]
group_labels = ["Ⅰ", "Ⅱ", "Ⅲ", "Ⅳ", "Ⅴ", "Ⅵ", "Ⅶ", "Ⅷ", "Ⅸ", "Ⅹ", ]
group_layouts = ["monadtall", "monadtall", "monadtall", "monadtall",
		 "monadtall", "monadtall", "monadtall", "monadtall",
		 "monadtall", "monadtall", ]

for i in range(len(group_names)):
    groups.append(
        Group(name=group_names[i],
              layout=group_layouts[i].lower(),
              label=group_labels[i],))


def init_layout_theme():
    """Layout default theme."""
    return {"margin": 5,
	    "border_width": 2,
	    "border_focus": dracula["cyan"],
	    "border_normal": dracula["bg"],
	    }

@hook.subscribe.startup_once
def autostart():
    autostart_script = os.path.expanduser('~/.config/autostart/autostart.sh')
    subprocess.Popen([autostart_script])

layout_theme = init_layout_theme()

layouts = [
    layout.MonadTall(**layout_theme),
    # layout.MonadWide(**layout_theme),
    # layout.Matrix(**layout_theme),
    # layout.Bsp(**layout_theme),
    layout.Floating(**layout_theme),
    # layout.RatioTile(**layout_theme),
    # layout.Max(**layout_theme),
    # layout.Spiral(**layout_theme),
    layout.TreeTab(bg_color=dracula["bg"],
		   active_bg=dracula["purple"],
		   active_fg=dracula["fg"],
		   inactive_fg=dracula["fg"],
		   inactive_bg=dracula["bg"],
		   font="JetBrains Mono",
		   fontshadow=None,
		   section_fontsize=14,
		   panel_width=135,),
    # layout.Zoomy(**layout_theme),
]

@hook.subscribe.client_new
def float_my_app(window):
    if window.window.get_name() == "thanos/pass-launcher":
        window.floating = True
        
# Widgets
def init_widgets_defaults():
    """Set widget default settings."""
    return dict(font="JetBrains Mono",
		fontsize=14,
		padding=2,
		foreground=dracula["fg"],
		background=dracula["bg"]
		)


widget_defaults = init_widgets_defaults()

def init_widgets_list():
    """My widget list."""
    widgets_list = [
	widget.Spacer(
	    length=7,
	),
	widget.Image(
	    filename="~/.config/qtile/Gentoo2.png",
	    scale="True",
	    margin=2,
	),
	widget.Spacer(
	    length=7,
	),
	widget.GroupBox(
	    font="FontAwesome",
	    fontsize=20,
	    padding=3,
	    margin=4,
	    borderwidth=1,
	    active=dracula["cyan"],
	    inactive=dracula["curr"],
	    rounded=False,
	    highlight_method="line",
	    this_current_screen_border=dracula["red"],
	    other_current_screen_border=dracula["curr"],
	),
	widget.CurrentLayout(
	    font="Noto Sans Bold",
	    foreground=dracula["fg"],
	    background="#5B4B8A",
	),
	widget.Spacer(
	    length=5
	),
	widget.WindowName(
	    font="JetBrains Mono Bold",
	    fontsize=12,
	    foreground=dracula["green"],
	    background=dracula["bg"],
	),
	widget.Spacer(
	    length=10,
	),
	widget.Image(
	    filename="~/.config/qtile/volume1.png",
	    scale=True,
	),
	widget.Spacer(
	    length=7,
	    padding=10,
	),
	widget.Volume(
	    fontsize=15,
	    fmt='{} ',
	    background=dracula["bg"],
	    foreground=dracula['orange'],
	    font="JetBrains Mono",
	    margin=10,
        # get_volume_command="pamixer --get-volume-human",
        # volume_app = "pamixer"
	),
	# widget.Mpris2(
	#     fontsize= 16,
	#     background = my_colors["red"],
	#     objname="org.mpris.MediaPlayer2.spotify"
	# ),
	widget.Spacer(
	    length=7,
	    padding=10,
	),
	widget.Image(
	    filename="~/.config/qtile/ram2.png",
	    scale=True,
	),
	widget.Memory(
	    measure_mem="G",
	    fontsize=15,
	    foreground=dracula["cyan"],
	    background=dracula["bg"],
	    font='JetBrains Mono Bold',
	),
	widget.MemoryGraph(
	    type='box',
	    graph_color=dracula["cyan"],
	    foreground=dracula["fg"],
	    background=dracula["bg"],
	),
	widget.Spacer(
	    length=7
	),
	widget.Image(
	    filename="~/.config/qtile/cpu1.png",
	    scale=True,
	),
	widget.ThermalSensor(
	    threshold=50,
	    fontsize=15,
	    font='JetBrains Mono Bold',
	    foreground=dracula["cyan"],
	    background=dracula["bg"],
	),
	widget.CPUGraph(
	    type='box',
	    graph_color=dracula["cyan"],
	    background=dracula["bg"],
	),
	widget.Spacer(
	    length=5
	),
	widget.TextBox(
	    font="FontAwesome",
	    text="  ",
	    foreground=dracula["orange"],
	    background=dracula["bg"],
	    padding=2,
	    fontsize=18
	),
	widget.Clock(
	    font='JetBrains Mono Bold',
	    foreground=dracula["cyan"],
	    background=dracula["bg"],
	    fontsize=16,
	    format="%A %d/%m | %H:%M"
	),
	widget.Spacer(
	    length=6
	),
	widget.Systray(
	    background=dracula["bg"],
	    icon_size=20,
	    padding=10,
	    margin=10,
	),
	widget.Spacer(
	    length=5
	),
    ]
    return widgets_list

battery_widget = widget.Battery()

widgets_list = init_widgets_list()


def init_widgets_screen1():
    """Init every widget for primary screen."""
    widgets_screen1 = init_widgets_list()
    if hostname == "hermes":
        widgets_screen1.append(battery_widget)
    return widgets_screen1


def init_widgets_screen2():
    """Remove systray and spacer for second screen."""
    widgets_screen2 = init_widgets_list()
    return widgets_screen2[:-2]


widgets_screen1 = init_widgets_screen1()
widgets_screen2 = init_widgets_screen2()


def init_screens():
    """Panel settings."""
    return [Screen(top=bar.Bar(widgets=init_widgets_screen1(), size=33,
			       margin=11),),
	    Screen(top=bar.Bar(widgets=init_widgets_screen2(), size=28,
			       margin=7))]


screens = init_screens()
follow_mouse_focus = True
bring_front_click = False
cursor_warp = True
auto_fullscreen = True
focus_on_window_activation = "smart"  # or focus
wmname = "Qtile-Apo11o"


## Keys

# Use this to change between workspaces
dgroups_key_binder = simple_key_binder("mod4")
mod = "mod4"
browser = "firefox"

keys = [
    Key([mod], "Return",
        lazy.spawn("emacsclient -c"),
        ),
    Key([mod, "shift"], "a",
        lazy.spawn('anki'),
        ),
    Key([mod, "shift"], "m",
        lazy.spawn("emacsclient -e '(thanos/pass-launcher)'"),
        ),
    Key([mod], "Tab",
        lazy.spawn("rofi -show drun"),
        ),
    Key([mod], "r",
        lazy.spawn("rofi -show run"),
        ),
    Key([mod], "w",
        lazy.spawn("rofi -show window"),
        ),
    Key([mod], "p",
        lazy.spawn("passmenu -p 'Password for: '"),
        ),
    Key([mod], "b",
        lazy.spawn(browser),
        ),
    Key([mod, "shift"], "c",
        lazy.next_layout(),
        ),
    Key([mod], "q",
        lazy.window.kill(),
        ),
    Key([mod, "shift"], "r",
        lazy.restart(),
        ),
    Key([mod, "shift"], "0",
        lazy.shutdown(),
        ),
    Key([mod, "shift"], "e",
        lazy.spawn("emacs"),
        ),
    # Switch focus of monitors
    Key([mod], "period",
        lazy.next_screen(),
        ),
    # Treetab controls
    Key([mod, "shift"], "h",
        lazy.layout.move_left(),
        ),
    Key([mod, "shift"], "l",
        lazy.layout.move_right(),
        ),
    # Window controls
    Key([mod], "n",
        lazy.layout.down(),
        ),
    Key([mod], "p",
        lazy.layout.up(),
        ),
    Key([mod, "shift"], "n",
        lazy.layout.shuffle_down(),
        lazy.layout.section_down(),
        ),
    Key([mod, "shift"], "p",
        lazy.layout.shuffle_up(),
        lazy.layout.section_up(),
        ),
    Key([mod], "s",
        lazy.layout.shrink(),
        lazy.layout.decrease_nmaster(),
        ),
    Key([mod], "l",
        lazy.layout.grow(),
        lazy.layout.increase_nmaster(),
        ),
    Key([mod], "o",
        lazy.layout.normalize(),
        ),
    Key([mod], "m",
        lazy.layout.maximize(),
        ),
    Key([mod, "shift"], "f",
        lazy.window.toggle_floating(),
        ),
    Key([mod], "f",
        lazy.window.toggle_fullscreen(),
        ),
    # Stack controls
    Key([mod, "shift"], "Tab",
        lazy.layout.rotate(),
        lazy.layout.flip(),
        ),
    Key([mod, "shift"], "space",
        lazy.layout.toggle_split(),
        ),
    # Volume controls
    Key([], "XF86AudioLowerVolume",
        lazy.spawn("amixer sset Master 5%-")
        ),
    Key([], "XF86AudioRaiseVolume",
        lazy.spawn("amixer sset Master 5%+")
        ),
    ]

mouse = [
    Drag([mod], "Button1", lazy.window.set_position_floating(),
	 start=lazy.window.get_position()),
    Drag([mod], "Button3", lazy.window.set_size_floating(),
	 start=lazy.window.get_size()),
    Click([mod], "Button2",
	  lazy.window.bring_to_front()),
 ]