# Copyright (c) 2010 Aldo Cortesi
# Copyright (c) 2010, 2014 dequis
# Copyright (c) 2012 Randall Ma
# Copyright (c) 2012-2014 Tycho Andersen
# Copyright (c) 2012 Craig Barnes
# Copyright (c) 2013 horsik
# Copyright (c) 2013 Tao Sauvage
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.

from libqtile import bar, layout, widget
from libqtile.config import Click, Drag, Group, Key, Match, Screen
from libqtile.lazy import lazy
from libqtile.utils import guess_terminal
from qtile_extras import widget
from qtile_extras.widget.decorations import PowerLineDecoration
import subprocess
import os
import logging
import random
from libqtile import hook, qtile

@hook.subscribe.startup
def dbus_register():
    id = os.environ.get('DESKTOP_AUTOSTART_ID')
    if not id:
        return
    subprocess.Popen(['dbus-send',
                      '--session',
                      '--print-reply',
                      '--dest=org.gnome.SessionManager',
                      '/org/gnome/SessionManager',
                      'org.gnome.SessionManager.RegisterClient',
                      'string:qtile',
                      'string:' + id])

mod = "mod4"
terminal = "alacritty"
browser = "microsoft-edge"

@hook.subscribe.startup_once
def autostart_once():
    subprocess.run('/home/danmac/.config/qtile/autostart.sh')

@hook.subscribe.startup
def autostart():
    subprocess.run('/home/danmac/.config/qtile/refresh.sh')

def get_random_wallpaper():
    wallpaper_folder = '/home/shared/Pictures/wallpapers/'
    wallpapers = os.listdir(wallpaper_folder)
    return f'{wallpaper_folder}{random.choice(wallpapers)}'

def change_wallpaper():
    for screen in screens:
        lazy.screen.set_background(get_random_wallpaper())

colors = {
    "orange": "F8954F",
    "teal": "34C2D4",
    "purple": "6659AA",
    "blue": "3E86BC",
    "red": "EB6286",
} 

keys = [
    # A list of available commands that can be bound to keys can be found
    # at https://docs.qtile.org/en/latest/manual/config/lazy.html
    # Switch between windows
    Key([mod], "h", lazy.layout.left(), desc="Move focus to left"),
    Key([mod], "l", lazy.layout.right(), desc="Move focus to right"),
    Key([mod], "j", lazy.layout.down(), desc="Move focus down"),
    Key([mod], "k", lazy.layout.up(), desc="Move focus up"),
    Key([mod], "space", lazy.layout.next(), desc="Move window focus to other window"),
    Key([mod], "Left", lazy.layout.left(), desc="Move focus to left"),
    Key([mod], "Right", lazy.layout.right(), desc="Move focus to right"),
    Key([mod], "Down", lazy.layout.down(), desc="Move focus down"),
    Key([mod], "Up", lazy.layout.up(), desc="Move focus up"),
    Key([mod, "shift"], "Left", lazy.next_screen(), desc="Move to next screen"),
    Key([mod, "shift"], "Right", lazy.prev_screen(), desc="Move to previous screen"),
    # Move windows between left/right columns or move up/down in current stack.
    # Moving out of range in Columns layout will create new column.
    Key([mod, "shift"], "h", lazy.layout.shuffle_left(), desc="Move window to the left"),
    Key([mod, "shift"], "l", lazy.layout.shuffle_right(), desc="Move window to the right"),
    Key([mod, "shift"], "j", lazy.layout.shuffle_down(), desc="Move window down"),
    Key([mod, "shift"], "k", lazy.layout.shuffle_up(), desc="Move window up"),
    # Grow windows. If current window is on the edge of screen and direction
    # will be to screen edge - window would shrink.
    Key([mod, "control"], "h", lazy.layout.grow_left(), desc="Grow window to the left"),
    Key([mod, "control"], "l", lazy.layout.grow_right(), desc="Grow window to the right"),
    Key([mod, "control"], "j", lazy.layout.grow_down(), desc="Grow window down"),
    Key([mod, "control"], "k", lazy.layout.grow_up(), desc="Grow window up"),
    Key([mod], "n", lazy.layout.normalize(), desc="Reset all window sizes"),
    Key([mod], "f", lazy.window.toggle_floating(), desc="Toggle floating"),
    # Toggle between split and unsplit sides of stack.
    # Split = all windows displayed
    # Unsplit = 1 window displayed, like Max layout, but still with
    # multiple stack panes
    Key(
        [mod, "shift"],
        "Return",
        lazy.layout.toggle_split(),
        desc="Toggle between split and unsplit sides of stack",
    ),
    # launch things
    Key([mod], "Return", lazy.spawn(terminal), desc="Launch terminal"),
    Key([mod, "control"], "b", lazy.spawn(browser), desc="Launch Browser"), 
    Key([mod, "control"], "f", lazy.spawn("firefox"), desc="Launch Firefox"),
    Key([mod, "control"], "t", lazy.spawn("gtk-launch msedge-cifhbcnohmdccbgoicgdjpfamggdegmo-Default.desktop"), desc="Launch Teams"),
    Key([mod, "control"], "o", lazy.spawn("gtk-launch msedge-faolnafnngnfdaknnbpnkhgohbobgegn-Default.desktop"), desc="Launch Teams"),
    Key([mod, "control"], "w", lazy.spawn("VirtualBoxVM --startvm rnet-windows"), desc="Launch Windows VM"),
    Key([mod, "control"], "y", lazy.spawn("gtk-launch FFPWA-01HATKWJX09JCYK2HQ0E2MGSRY.desktop"), desc="Launch YouTube"),
    Key([mod, "control"], "v", lazy.spawn("VirtualBox"), desc="Launch VirtualBox"),
    Key([mod, "control"], "c", lazy.spawn("gtk launch arch-codium.dekstop"), desc="Launch VSCodium"),
    Key([], "Print", lazy.spawn("flatpak run org.flameshot.Flameshot gui"), desc="Screenshot"),
    # Toggle between different layouts as defined below
    Key([mod], "Tab", lazy.next_layout(), desc="Toggle between layouts"),
    Key([mod, "shift"], "Tab", lazy.prev_layout(), desc="Toggle between layouts"),
    Key([mod], "q", lazy.window.kill(), desc="Kill focused window"),
    Key([mod, "shift"], "b", lazy.screen.set_wallpaper(get_random_wallpaper()), desc="Change desktop wallpapers"),
    Key([mod, "control"], "r", lazy.reload_config(), desc="Reload the config"),
    Key([mod, "control"], "q", lazy.shutdown(), desc="Shutdown Qtile"),
    Key([mod], "r", lazy.spawncmd(), desc="Spawn a command using a prompt widget"),
    Key([], "XF86AudioPlay", lazy.spawn("playerctl play-pause"), desc="Play Pause media"),
    Key([], "XF86AudioMute", lazy.spawn("amixer set Master 1+ toggle"), desc="Mute sound"),
    Key([], "XF86AudioLowerVolume", lazy.spawn("amixer set Master 5%-"), desc="Reduce volume"),
    Key([], "XF86AudioRaiseVolume", lazy.spawn("amixer set Master 5%+"), desc="Increase volume"),
    Key([mod], "s", lazy.screen.next_group(skip_managed=True), desc="Switch to Next Group"),
    Key([mod], "w", lazy.screen.prev_group(skip_managed=True), desc="Switch to Previous Group"),
    Key([mod, "mod1"], "l", lazy.spawn("light-locker-command -l"), desc="Lock Screen"),
]

groups = [Group(i) for i in "123456789"]

for i in groups:
    keys.extend(
        [
            # mod1 + letter of group = switch to group
            Key(
                [mod],
                i.name,
                lazy.group[i.name].toscreen(),
                desc="Switch to group {}".format(i.name),
            ),
            # # mod1 + shift + letter of group = switch to & move focused window to group
            # Key(
            #     [mod, "shift"],
            #     i.name,
            #     lazy.window.togroup(i.name, switch_group=True),
            #     desc="Switch to & move focused window to group {}".format(i.name),
            # ),
            # Or, use below if you prefer not to switch to that group.
            # mod1 + shift + letter of group = move focused window to group
            Key([mod, "shift"], i.name, lazy.window.togroup(i.name),
                desc="move focused window to group {}".format(i.name)),
        ]
    )

layout_defaults = {
    "border_width": 2,
    "margin": [0, 4, 4, 4],
    "border_focus": "#ffffff",
}

layouts = [
    layout.Columns(border_focus_stack=["#ff33cc", "#cc33ff"], **layout_defaults),
    layout.Max(**layout_defaults),
    # Try more layouts by unleashing below layouts.
    layout.Stack(num_stacks=1, **layout_defaults),
    # layout.Bsp(),
    # layout.Matrix(),
    layout.MonadTall(ratio=0.75, **layout_defaults),
    layout.MonadThreeCol(**layout_defaults),
    # layout.MonadWide(),
    # layout.RatioTile(),
    layout.Tile(**layout_defaults),
    # layout.TreeTab(),
    # layout.VerticalTile(),
    # layout.Zoomy(),
]

# default layouts for groups
groups[1].layout = "monadthreecol"
groups[2].layout = "stack"

# groups[1].spawn = "flatpak run com.microsoft.Edge"

powerline = {
    "decorations": [
        PowerLineDecoration()
    ]
}

lb_progs = [
    ("microsoft-edge","microsoft-edge","Launch Micorosoft Edge"),
    ("msedge-cifhbcnohmdccbgoicgdjpfamggdegmo-Default", "gtk-launch msedge-cifhbcnohmdccbgoicgdjpfamggdegmo-Default.desktop", "Launch Microsoft Teams"),
    # ("FFPWA-01H62W1P23FQGJ1KW173SWKGX5", "gtk-launch arch-FFPWA-01H62W1P23FQGJ1KW173SWKGX5.desktop",""),
    ("firefox.png", "firefox", "Launch Firefox"),
    ("virtualbox.png", "VirtualBox", "Launch VirtualBox"),
    ("pycharm", "pycharm-professional", "Launch PyCharm"),
    ("org.xfce.thunar", "thunar", "Launch Files"),
    #("vscodium.png", "gtk-launch arch-codium.desktop", "Launch Codium"),
    ("FFPWA-01HATKWJX09JCYK2HQ0E2MGSRY", "gtk-launch FFPWA-01HATKWJX09JCYK2HQ0E2MGSRY.desktop",""),
    ("🔊", "pavucontrol", "Launch pavucontrol"),
]

app_launcher = widget.WidgetBox(
    widgets=[widget.LaunchBar(scale=0.7,progs=lb_progs, padding=12, background=colors["red"])],
    background=colors["red"],
    text_closed=" ▶ ",
    text_open=" ◀ ",
    **powerline
)
    

shutdown_progs = [
    ("Reboot", "systemctl reboot", "Reboot"),
    ("Shutdown", "systemctl halt", "Shutdown"),
    ("Logout", "qtile cmd-obj -o cmd -f shutdown", "Logout"),
    ("Lock", "light-locker-command -l", "Lock"),
]

widget_defaults = dict(
    font="sans",
    fontsize=14,
    padding=3,
)

extension_defaults = widget_defaults.copy()

groupbox_colors = {
    "highlight_method": "block",
    "block_highlight_text_color": "000000",
    "this_current_screen_border": "ffffff",
    "this_screen_border": "ffffff",
    "other_screen_border": "777777",
    "other_current_screen_border": "777777",
    "inactive": "777777",
}

screens = [
    Screen(
        top=bar.Bar(
            [
                widget.CurrentLayoutIcon(scale=0.7, padding=14),
                widget.GroupBox(**groupbox_colors), 
                # widget.Prompt(background="333333", **powerline),
                widget.WindowTabs(),
                widget.Chord(
                    chords_colors={
                        "launch": ("#ff0000", "#ffffff"),
                    },
                    name_transform=lambda name: name.upper(),
                ),
                # widget.TextBox("default config", name="default"),
                # widget.TextBox("Press <M-r> to spawn", foreground="#d75f5f"),
                # NB Systray is incompatible with Wayland, consider using StatusNotifier instead
                # widget.StatusNotifier(),
                # app_launcher,
                widget.WidgetBox(
                    widgets=[widget.LaunchBar(progs=lb_progs, padding=12),],
                    text_closed=" ▶ ",
                    text_open=" ◀ ",
                ),
                widget.Spacer(length=8),
                # widget.pulse_volume(),
                # widget.PulseVolume(background=colors["purple"], **powerline),
                widget.UPowerWidget(),
                widget.CapsNumLockIndicator(),
                widget.CPUGraph(border_width=0),
                widget.ThermalSensor(tag_sensor='Package id 0', format='CPU: {temp}{unit}', threshold=85), 
                widget.NvidiaSensors(threshold=85, format='GPU: {temp}°C'),
                widget.Wttr(location={'Adelaide, Australia':"Home"}),
                # widget.Bluetooth(),
                widget.Systray(),
                widget.Clock(format="%a %d %b %Y %I:%M %p"),
                # widget.Wallpaper(directory='/home/shared/Pictures/wallpapers/', random_selection=True, label="🖼️"),
                # widget.QuickExit(),
                widget.WidgetBox(
                    widgets=[
                        widget.LaunchBar(padding=12, progs=shutdown_progs),
                    ],
                    text_closed=" ⏻ ▶ ",
                    text_open=" ⏻ ◀ ",
                ),
            ],
            32,
            margin=4,
            background=["#00000099"],
            # border_width=[2, 0, 2, 0],  # Draw top and bottom borders
            # border_color=["ff00ff", "000000", "ff00ff", "000000"]  # Borders are magenta
        ),
        wallpaper=get_random_wallpaper(),
        wallpaper_mode="fill"
    ),
    Screen(
        top=bar.Bar(
            [
                widget.CurrentLayoutIcon(scale=0.7,padding=14), # , background=colors["teal"], **powerline),
                widget.GroupBox(**groupbox_colors), 
                widget.Prompt(), # background=colors["purple"], **powerline),
                widget.WindowTabs(), # background=colors["blue"], **powerline),
                widget.Spacer(), # background=colors["blue"], **powerline),
                widget.Clock(format="%a %d %b %Y %I:%M %p",), # background=colors["orange"]),
            ],
            32,
            margin=4,
            background=["#00000099"],
            # border_width=[2, 0, 2, 0],  # Draw top and bottom borders
            # border_color=["ff00ff", "000000", "ff00ff", "000000"]  # Borders are magenta
        ),
        wallpaper=get_random_wallpaper(),
        wallpaper_mode="fill"
    ),
    Screen(
        top=bar.Bar(
            [
                widget.CurrentLayoutIcon(scale=0.7,padding=14),
                widget.GroupBox(**groupbox_colors),
                widget.Prompt(),
                widget.WindowTabs(),
                widget.Spacer(),
                widget.Clock(format="%a %d %b %Y %I:%M %p"),
            ],
            32,
            margin=4,
            background=["#00000099"],
            # border_width=[2, 0, 2, 0],  # Draw top and bottom borders
            # border_color=["ff00ff", "000000", "ff00ff", "000000"]  # Borders are magenta
        ),
        wallpaper=get_random_wallpaper(),
        wallpaper_mode="fill"
    ),
]





# Drag floating layouts.
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()),
]

dgroups_key_binder = None
dgroups_app_rules = []  # type: list
follow_mouse_focus = False
bring_front_click = True
cursor_warp = False
floating_layout = layout.Floating(
    border_width=1,
    float_rules=[
        # Run the utility of `xprop` to see the wm class and name of an X client.
        *layout.Floating.default_float_rules,
        Match(wm_class="confirmreset"),  # gitk
        Match(wm_class="makebranch"),  # gitk
        Match(wm_class="maketag"),  # gitk
        Match(wm_class="ssh-askpass"),  # ssh-askpass
        Match(title="branchdialog"),  # gitk
        Match(title="pinentry"),  # GPG key password entry
        Match(title="Sticky Notes"),
        Match(wm_class="xpad"),
        Match(wm_class="gnome-calculator"),
        Match(wm_class="blueman-manager"),
        Match(wm_class="pavucontrol"),
    ]
)
auto_fullscreen = True
focus_on_window_activation = "smart"
reconfigure_screens = True

# If things like steam games want to auto-minimize themselves when losing
# focus, should we respect this or not?
auto_minimize = True

# When using the Wayland backend, this can be used to configure input devices.
wl_input_rules = None

# XXX: Gasp! We're lying here. In fact, nobody really uses or cares about this
# string besides java UI toolkits; you can see several discussions on the
# mailing lists, GitHub issues, and other WM documentation that suggest setting
# this string if your java app doesn't work correctly. We may as well just lie
# and say that we're a working one by default.
#
# We choose LG3D to maximize irony: it is a 3D non-reparenting WM written in
# java that happens to be on java's whitelist.
wmname = "LG3D"