pcall(require, "luarocks.loader")
local gears = require("gears")
local awful = require("awful")
require("awful.autofocus")
local wibox = require("wibox")
local beautiful = require("beautiful")
local naughty = require("naughty")
local menubar = require("menubar")
require("awful.hotkeys_popup.keys")

-- {{{ Error handling
-- Check if awesome encountered an error during startup and fell back to
-- another config (This code will only ever execute for the fallback config)
if awesome.startup_errors then
	naughty.notify({
		preset = naughty.config.presets.critical,
		title = "Oops, there were errors during startup!",
		text = awesome.startup_errors,
	})
end

-- Handle runtime errors after startup
do
	local in_error = false
	awesome.connect_signal("debug::error", function(err)
		-- Make sure we don't go into an endless error loop
		if in_error then
			return
		end
		in_error = true

		naughty.notify({
			preset = naughty.config.presets.critical,
			title = "Oops, an error happened!",
			text = tostring(err),
		})
		in_error = false
	end)
end
-- }}}

-- {{{ Variable definitions
-- Themes define colours, icons, font and wallpapers.
beautiful.init(gears.filesystem.get_themes_dir() .. "default/theme.lua")

-- This is used later as the default terminal and editor to run.
terminal = "kitty"
editor = os.getenv("EDITOR") or "nvim"
editor_cmd = terminal .. " -e " .. editor

mod = "Mod4"

awful.layout.layouts = {
	-- awful.layout.suit.floating,
	awful.layout.suit.tile,
	awful.layout.suit.tile.left,
	awful.layout.suit.tile.bottom,
	awful.layout.suit.tile.top,
	awful.layout.suit.fair,
	awful.layout.suit.fair.horizontal,
	awful.layout.suit.spiral,
	awful.layout.suit.spiral.dwindle,
	awful.layout.suit.max,
	awful.layout.suit.max.fullscreen,
	awful.layout.suit.magnifier,
	awful.layout.suit.corner.nw,
	-- awful.layout.suit.corner.ne,
	-- awful.layout.suit.corner.sw,
	-- awful.layout.suit.corner.se,
}

local tag_names = { "a", "s", "d", "f", "g", "z", "x", "c", "v", "b" }

awful.screen.connect_for_each_screen(function(s)
	-- Each screen has its own tag table.
	awful.tag(tag_names, s, awful.layout.layouts[1])
end)

-- {{{ Mouse bindings
root.buttons(gears.table.join(
	awful.button({}, 3, function()
		mymainmenu:toggle()
	end),
	awful.button({}, 4, awful.tag.viewnext),
	awful.button({}, 5, awful.tag.viewprev)
))
-- }}}

-- {{{ Key bindings
globalkeys = gears.table.join(

	-- basics
	awful.key({ mod }, "t", awesome.restart),

	-- focus
	awful.key({ mod }, "h", awful.client.focus.bydirection("left")),
	awful.key({ mod }, "j", awful.client.focus.bydirection("down")),
	awful.key({ mod }, "k", awful.client.focus.bydirection("up")),
	awful.key({ mod }, "l", awful.client.focus.bydirection("right")),

	-- shift
	awful.key({ mod, "Shift" }, "h", awful.client.swap.bydirection("left")),
	awful.key({ mod, "Shift" }, "j", awful.client.swap.bydirection("down")),
	awful.key({ mod, "Shift" }, "k", awful.client.swap.bydirection("up")),
	awful.key({ mod, "Shift" }, "l", awful.client.swap.bydirection("right")),

	-- cycle
	awful.key({ mod }, "m", function()
		awful.screen.focus_relative(1)
	end),

	-- cycle layouts
	awful.key({ mod }, "space", function()
		awful.layout.inc(1)
	end),

	-- Standard program
	awful.key({ mod }, "Return", function()
		awful.spawn(terminal)
	end),

	awful.key({ mod }, "l", function()
		awful.tag.incmwfact(0.05)
	end),
	awful.key({ mod, "Control" }, "h", function()
		awful.tag.incmwfact(-0.05)
	end),
	awful.key({ mod, "Shift" }, "h", function()
		awful.tag.incnmaster(1, nil, true)
	end),
	awful.key({ mod, "Shift" }, "l", function()
		awful.tag.incnmaster(-1, nil, true)
	end),
	awful.key({ mod, "Control" }, "h", function()
		awful.tag.incncol(1, nil, true)
	end),
	awful.key({ mod, "Control" }, "l", function()
		awful.tag.incncol(-1, nil, true)
	end),
	awful.key({ mod, "Shift" }, "space", function()
		awful.layout.inc(-1)
	end),

	awful.key({ mod }, "1", function()
		awful.spawn("thunar")
	end),
	awful.key({ mod }, "2", function()
		awful.spawn("firefox")
	end),
	awful.key({ mod }, "3", function()
		awful.spawn("rofi -show calc")
	end),
	awful.key({ mod }, "4", function()
		awful.spawn("kitty -e ncmpcpp")
	end),
	awful.key({ mod }, "0", function()
		awful.spawn("i3lock -c 000000")
	end),

	awful.key({ mod, "Control" }, "n", function()
		local c = awful.client.restore()
		-- Focus restored client
		if c then
			c:emit_signal("request::activate", "key.unminimize", { raise = true })
		end
	end, { description = "restore minimized", group = "client" })
)

clientkeys = gears.table.join(
	-- basics
	awful.key({ mod }, "w", function(c)
		c:kill()
	end),

	-- window attributes
	awful.key({ mod, "Control" }, "e", function(c)
		c.fullscreen = not c.fullscreen
		c:raise()
	end),
	awful.key({ mod, "Control" }, "q", awful.client.floating.toggle),

	awful.key({ mod, "Control" }, "Return", function(c)
		c:swap(awful.client.getmaster())
	end, { description = "move to master", group = "client" }),
	awful.key({ mod }, "o", function(c)
		c:move_to_screen()
	end, { description = "move to screen", group = "client" }),
	awful.key({ mod }, "t", function(c)
		c.ontop = not c.ontop
	end, { description = "toggle keep on top", group = "client" }),
	awful.key({ mod }, "n", function(c)
		-- The client currently has the input focus, so it cannot be
		-- minimized, since minimized clients can't have the focus.
		c.minimized = true
	end, { description = "minimize", group = "client" }),
	awful.key({ mod }, "m", function(c)
		c.maximized = not c.maximized
		c:raise()
	end, { description = "(un)maximize", group = "client" }),
	awful.key({ mod, "Control" }, "m", function(c)
		c.maximized_vertical = not c.maximized_vertical
		c:raise()
	end, { description = "(un)maximize vertically", group = "client" }),
	awful.key({ mod, "Shift" }, "m", function(c)
		c.maximized_horizontal = not c.maximized_horizontal
		c:raise()
	end, { description = "(un)maximize horizontally", group = "client" })
)

-- Bind all key numbers to tags.
-- Be careful: we use keycodes to make it work on any keyboard layout.
-- This should map on the top row of your keyboard, usually 1 to 9.
-- for
local tags = { "a", "s", "d", "f", "g", "z", "x", "c", "v", "b" }
for i, tag_name in ipairs(tags) do
	globalkeys = gears.table.join(
		globalkeys,
		-- View tag only.
		awful.key({ mod }, tag_name, function()
			local screen = awful.screen.focused()
			local tag = screen.tags[i]
			if tag then
				tag:view_only()
			end
		end),
		-- Toggle tag display.
		awful.key({ mod, "Control" }, tag_name, function()
			local screen = awful.screen.focused()
			local tag = screen.tags[i]
			if tag then
				awful.tag.viewtoggle(tag)
			end
		end),
		-- Move client to tag.
		awful.key({ mod, "Shift" }, tag_name, function()
			if client.focus then
				local tag = client.focus.screen.tags[i]
				if tag then
					client.focus:move_to_tag(tag)
				end
			end
		end),
		-- Toggle tag on focused client.
		awful.key({ mod, "Control", "Shift" }, tag_name, function()
			if client.focus then
				local tag = client.focus.screen.tags[i]
				if tag then
					client.focus:toggle_tag(tag)
				end
			end
		end)
	)
end

clientbuttons = gears.table.join(
	awful.button({}, 1, function(c)
		c:emit_signal("request::activate", "mouse_click", { raise = true })
	end),
	awful.button({ mod }, 1, function(c)
		c:emit_signal("request::activate", "mouse_click", { raise = true })
		awful.mouse.client.move(c)
	end),
	awful.button({ mod }, 3, function(c)
		c:emit_signal("request::activate", "mouse_click", { raise = true })
		awful.mouse.client.resize(c)
	end)
)

-- Set keys
root.keys(globalkeys)
-- }}}

-- {{{ Rules
-- Rules to apply to new clients (through the "manage" signal).
awful.rules.rules = {
	-- All clients will match this rule.
	{
		rule = {},
		properties = {
			border_width = beautiful.border_width,
			border_color = beautiful.border_normal,
			focus = awful.client.focus.filter,
			raise = true,
			keys = clientkeys,
			buttons = clientbuttons,
			screen = awful.screen.preferred,
			placement = awful.placement.no_overlap + awful.placement.no_offscreen,
		},
	},

	-- Floating clients.
	{
		rule_any = {
			instance = {
				"DTA", -- Firefox addon DownThemAll.
				"copyq", -- Includes session name in class.
				"pinentry",
			},
			class = {
				"Arandr",
				"Blueman-manager",
				"Gpick",
				"Kruler",
				"MessageWin", -- kalarm.
				"Sxiv",
				"Tor Browser", -- Needs a fixed window size to avoid fingerprinting by screen size.
				"Wpa_gui",
				"veromix",
				"xtightvncviewer",
			},

			-- Note that the name property shown in xprop might be set slightly after creation of the client
			-- and the name shown there might not match defined rules here.
			name = {
				"Event Tester", -- xev.
			},
			role = {
				"AlarmWindow", -- Thunderbird's calendar.
				"ConfigManager", -- Thunderbird's about:config.
				"pop-up", -- e.g. Google Chrome's (detached) Developer Tools.
			},
		},
		properties = { floating = true },
	},

	-- Add titlebars to normal clients and dialogs
	{ rule_any = { type = { "normal", "dialog" } }, properties = { titlebars_enabled = true } },

	-- Set Firefox to always map on the tag named "2" on screen 1.
	-- { rule = { class = "Firefox" },
	--   properties = { screen = 1, tag = "2" } },
}
-- }}}

-- {{{ Signals
-- Signal function to execute when a new client appears.
client.connect_signal("manage", function(c)
	-- Set the windows at the slave,
	-- i.e. put it at the end of others instead of setting it master.
	-- if not awesome.startup then awful.client.setslave(c) end

	if awesome.startup and not c.size_hints.user_position and not c.size_hints.program_position then
		-- Prevent clients from being unreachable after screen count changes.
		awful.placement.no_offscreen(c)
	end
end)

-- Enable sloppy focus, so that focus follows mouse.
client.connect_signal("mouse::enter", function(c)
	c:emit_signal("request::activate", "mouse_enter", { raise = false })
end)

client.connect_signal("focus", function(c)
	c.border_color = beautiful.border_focus
end)
client.connect_signal("unfocus", function(c)
	c.border_color = beautiful.border_normal
end)
-- }}}