-- {{{ Includes
-- Standard awesome library
local gears = require('gears')
local awful = require('awful')
awful.rules = require('awful.rules')
require('awful.autofocus')
-- Widget and layout library
local wibox = require('wibox')
-- Theme handling library
local beautiful = require('beautiful')
-- Notification library
local naughty = require('naughty')
local menubar = require('menubar')
-- Treesome layout module: https://github.com/RobSis/treesome
local treesome = require('treesome')
-- }}}
-- {{{ Custom functions
-- test if file exists
function exists(name)
	local f=io.open(name,'r')
	if f ~= nil then
		io.close(f)
		return true
	else
		return false
	end
end

-- verify if element is part of the table
function enters(element, table)
	for key, value in pairs(table) do
		if value == element then
			return true
		end
	end
	return false
end
-- }}}
-- {{{ 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 = err
			})
			in_error = false
		end
	)
end
-- }}}
-- {{{ Variable definitions
-- Themes define colours, icons, and wallpapers
beautiful.init('/usr/share/awesome/themes/default/theme.lua')
theme.border_width = 1
theme.font = 'Terminus 9'
theme.taglist_squares_sel = nil
theme.taglist_squares_unsel = nil

-- amount of pixels to snap windows from
snap = 7

-- set wallpaper
local wallpaper = '/home/von/Pictures/wallpaper.png'
if exists(wallpaper) then
	theme.wallpaper = wallpaper
end

-- This is used later as the default terminal and editor to run.
terminal = 'termite'
editor = os.getenv('EDITOR') or 'vim'
editor_cmd = terminal .. ' -e ' .. editor

-- Default modkey.
-- Usually, Mod4 is the key with a logo between Control and Alt.
-- If you do not like this or do not have such a key,
-- I suggest you to remap Mod4 to another key using xmodmap or other tools.
-- However, you can use another modifier like Mod1, but it may interact with others.
modkey = 'Mod4'

-- Tables of layouts to cover with awful.layout.inc, order matters.
local layouts = {
	tiled = {
		awful.layout.suit.tile,
		awful.layout.suit.tile.bottom,
		awful.layout.suit.tile.left,
		awful.layout.suit.tile.top
	},
	max = {
		awful.layout.suit.max,
		awful.layout.suit.max.fullscreen
	},
	float = {
		awful.layout.suit.floating
	},
	spiral = {
		treesome,
		awful.layout.suit.spiral,
		awful.layout.suit.spiral.dwindle
	},

	-- Some weird shit I don't use, but it still exists:

	fair = {
		awful.layout.suit.fair,
		awful.layout.suit.fair.horizontal
	},
	magnifier = {
		awful.layout.suit.magnifier
	}
}
-- }}}
-- {{{ Wallpaper
if beautiful.wallpaper then
	for s = 1, screen.count() do
		gears.wallpaper.centered(beautiful.wallpaper, s)
	end
end
-- }}}
-- {{{ Tags
-- Provide tag names and layout settings if we wish to define them
tags = {}
saved_layouts = {}
for s = 1, screen.count() do
	tags[s] = { name = {}, layout = {} }
	saved_layouts[s] = { tiled = {} }
end
-- screen 1
--tags[1].name = {
--	[1] = 'example',
--	[9] = 'example2'
--}
tags[1].layout = {
	[4] = layouts.float[1]
}
-- screens 2+
if screen.count() >= 2 then
	tags[2].layout = {
		[4] = layouts.float[1]
	}
end
-- Fill the missing values with defaults
for s = 1, screen.count() do
	-- Set default tiled layout for all the screens
	-- if s == 1 and screen.count() ~= 1 then -- in case I ever want to have default for less than 2 screens
	if s == 1 then
		default_tiled = layouts.tiled[3]
	else
		default_tiled = layouts.tiled[1]
	end
	for tag = 1, 9 do
		local name = tags[s].name[tag] or tostring(tag)
		local layout = tags[s].layout[tag] or default_tiled
		tags[s].name[tag] = name
		tags[s].layout[tag] = layout
		-- Assign saved tiled layout so that we can use it right away
		if enters(layout, layouts.tiled) then
			saved_layouts[s].tiled[name] = layout
		else
			saved_layouts[s].tiled[name] = default_tiled
		end
	end
end
-- Set tags instances in wm
for s = 1, screen.count() do
	tags[s] = awful.tag(tags[s].name, s, tags[s].layout)
end
-- }}}
-- {{{ Menu
-- Create a laucher widget and a main menu
mymainmenu_restart = {
	{ 'restart', awesome.restart }
}
--mymainmenu_quit = {
--	{ 'quit', awesome.quit }
--}

mymainmenu = awful.menu({
	items = {
		{ 'restart',	mymainmenu_restart,	beautiful.awesome_icon }
		-- { 'quit',	mymainmenu_quit,	beautiful.awesome_icon }
	}
})

-- Menubar configuration
menubar.utils.terminal = terminal -- Set the terminal for applications that require it
-- }}}
-- {{{ Menu for layoutbox
mylbmenu = awful.menu({
	items = {
		{ 'Tiled',
			function ()
				local screen = mouse.screen
				local tag = awful.tag.selected(screen).name
				awful.layout.set(saved_layouts[screen].tiled[tag])
			end
		},
		{ 'Maximized',	function () awful.layout.set(layouts.max[1]) end },
		{ 'Floating',	function () awful.layout.set(layouts.float[1]) end },
		{ 'Spiral',	function () awful.layout.set(layouts.spiral[1]) end }
	}
})
-- }}}
-- {{{ Wibox
-- Create a textclock widget
mytextclock = awful.widget.textclock(' %a %d %H:%M ')
mytextclock:set_font('Terminus Bold 11')
-- Create a mail notification widget
mytextbox = wibox.widget.textbox()
mytextbox:set_text('')
mytextbox:set_font('Terminus 9')
-- Create keyboard layout indicator widget
--mykblayout = wibox.widget.textbox()
--mykblayout:set_text('US')
--mykblayout:set_font('Terminus Bold 11')

-- Create a wibox for each screen and add it
mywibox = {}
mypromptbox = {}
mylayoutbox = {}
mytaglist = {}
mytaglist.buttons = awful.util.table.join(
	awful.button({        }, 1, awful.tag.viewonly),
	awful.button({ modkey }, 1, awful.client.movetotag),
	awful.button({        }, 3, awful.tag.viewtoggle),
	awful.button({ modkey }, 3, awful.client.toggletag)
)
mytasklist = {}
mytasklist.buttons = awful.util.table.join(
	awful.button({ }, 1,
		function (c)
			if c ~= client.focus then
				-- Without this, the following
				-- :isvisible() makes no sense
				c.minimized = false
				if not c:isvisible() then
					awful.tag.viewonly(c:tags()[1])
				end
				-- This will also un-minimize
				-- the client, if needed
				client.focus = c
				c:raise()
			end
		end),
	awful.button({ }, 3,
		function ()
			if instance then
				instance:hide()
				instance = nil
			else
				instance = awful.menu.clients({ width=250 })
			end
		end),
	awful.button({ }, 4,
		function ()
			awful.client.focus.byidx(1)
			if client.focus then
				client.focus:raise()
			end
		end),
	awful.button({ }, 5,
		function ()
			awful.client.focus.byidx(-1)
			if client.focus then
				client.focus:raise()
			end
		end)
)

for s = 1, screen.count() do
	-- Create a promptbox for each screen
	mypromptbox[s] = awful.widget.prompt()
	-- Create an imagebox widget which will contains an icon indicating which layout we're using.
	-- We need one layoutbox per screen.
	mylayoutbox[s] = awful.widget.layoutbox(s)
	mylayoutbox[s]:buttons(
		awful.util.table.join(
			awful.button({ }, 1,
				function ()
					local screen = mouse.screen
					local current_layout = awful.layout.get(screen)
					if enters(current_layout, layouts.tiled) then
						local tag = awful.tag.selected(screen).name
						awful.layout.inc(layouts.tiled, 1)
						saved_layouts[screen].tiled[tag] = awful.layout.get(screen)
					end
				end),
			awful.button({ }, 3, function () mylbmenu:toggle() end)
		)
	)
	-- Create a taglist widget
	mytaglist[s] = awful.widget.taglist(s, awful.widget.taglist.filter.noempty, mytaglist.buttons)

	-- Create a tasklist widget
	mytasklist[s] = awful.widget.tasklist(s, awful.widget.tasklist.filter.currenttags, mytasklist.buttons)

	-- Create the wibox
	mywibox[s] = awful.wibox({ position = 'top', height = '18', screen = s })

	local left_layout = wibox.layout.fixed.horizontal()
	local right_layout = wibox.layout.fixed.horizontal()
	if s == 1 then
		left_layout:add(mytextbox)
		right_layout:add(mypromptbox[s])
		right_layout:add(mytaglist[s])
		if screen.count() == 1 then
			right_layout:add(wibox.widget.systray())
			right_layout:add(mytextclock)
		end
		right_layout:add(mylayoutbox[s])
	elseif s == 2 then
		left_layout:add(mylayoutbox[s])
		left_layout:add(mytextclock)
		left_layout:add(wibox.widget.systray())
		left_layout:add(mytaglist[s])
		left_layout:add(mypromptbox[s])
	else
		left_layout:add(mytaglist[s])
		left_layout:add(mypromptbox[s])
		right_layout:add(mylayoutbox[s])
	end

	-- Now bring it all together (with the tasklist in the middle)
	local layout = wibox.layout.align.horizontal()
	layout:set_left(left_layout)
	layout:set_middle(mytasklist[s])
	layout:set_right(right_layout)

	mywibox[s]:set_widget(layout)
end
-- }}}
-- {{{ Mouse bindings
root.buttons(
	awful.util.table.join(
		awful.button({ }, 3, function () mymainmenu:toggle() end)
	)
)
-- }}}
-- {{{ Key bindings
globalkeys = awful.util.table.join(
	awful.key({ modkey,           }, 'Up',     function () awful.screen.focus_relative( 1) end),
	awful.key({ modkey,           }, 'Down',   function () awful.screen.focus_relative(-1) end),
	awful.key({ modkey,           }, 'Left',   awful.tag.viewprev),
	awful.key({ modkey,           }, 'Right',  awful.tag.viewnext),
	awful.key({ modkey,           }, 'Escape', awful.tag.history.restore),

	-- Switch between windows
	awful.key({ modkey,           }, 'j', function () awful.client.focus.byidx( 1) client.focus:raise() end),
	awful.key({ modkey,           }, 'k', function () awful.client.focus.byidx(-1) client.focus:raise() end),
	awful.key({ modkey,           }, 'i', function () client.focus:raise() end),
	awful.key({ modkey,           }, 'w', function () mymainmenu:show() end),

	-- Switch between screens
	awful.key({ modkey,           }, 'o', function () awful.screen.focus_relative(1) end),

	-- Layout manipulation
	awful.key({ modkey, 'Shift'   }, 'j', function () awful.client.swap.byidx( 1) end),
	awful.key({ modkey, 'Shift'   }, 'k', function () awful.client.swap.byidx(-1) end),
	awful.key({ modkey,           }, 'u', awful.client.urgent.jumpto),

	-- Mod#+Tab hotkeys
	awful.key({ modkey,           }, 'Tab',
		function ()
			awful.client.focus.history.previous()
			if client.focus then
				client.focus:raise()
			end
		end),
	awful.key({ 'Mod1',           }, 'Tab',
		function ()
			awful.client.focus.byidx(-1)
			if client.focus then
				client.focus:raise()
			end
		end),
	awful.key({ 'Mod1', 'Shift'   }, 'Tab',
		function ()
			awful.client.focus.byidx(1)
			if client.focus then
				client.focus:raise()
			end
		end),

	awful.key({ modkey,           }, 'l',     function () awful.tag.incmwfact( 0.05)                      end),
	awful.key({ modkey,           }, 'h',     function () awful.tag.incmwfact(-0.05)                      end),
	awful.key({ modkey, 'Shift'   }, 'h',     function () awful.tag.incnmaster( 1)                        end),
	awful.key({ modkey, 'Shift'   }, 'l',     function () awful.tag.incnmaster(-1)                        end),
	awful.key({ modkey, 'Control' }, 'h',     function () awful.tag.incncol( 1)                           end),
	awful.key({ modkey, 'Control' }, 'l',     function () awful.tag.incncol(-1)                           end),
	awful.key({ modkey,           }, 'f',     function () awful.layout.set(layouts.float[1])    end),
	awful.key({ modkey,           }, 'm',
		function ()
			local screen = mouse.screen
			local current_layout = awful.layout.get(screen)
			if not enters(current_layout, layouts.max) then
				awful.layout.set(layouts.max[1])
			else
				awful.layout.inc(layouts.max, 1)
			end
		end),
	awful.key({ modkey,           }, 't',
		function ()
			local screen = mouse.screen
			local current_layout = awful.layout.get(screen)
			local tag = awful.tag.selected(screen).name
			if not enters(current_layout, layouts.tiled) then
				awful.layout.set(saved_layouts[screen].tiled[tag])
			else
				awful.layout.inc(layouts.tiled, 1)
				saved_layouts[screen].tiled[tag] = awful.layout.get(screen)
			end
		end),
	awful.key({ modkey,           }, 's',
		function ()
			local screen = mouse.screen
			local current_layout = awful.layout.get(screen)
			if not enters(current_layout, layouts.spiral) then
				awful.layout.set(layouts.spiral[1])
			else
				awful.layout.inc(layouts.spiral, 1)
			end
		end),

	awful.key({ modkey, 'Control' }, 'n',     awful.client.restore),

	-- Prompt
	awful.key({ modkey,           }, 'r',     function () mypromptbox[mouse.screen]:run() end),
	awful.key({ 'Mod1',           }, 'F2',    function () mypromptbox[mouse.screen]:run() end),

	awful.key({ modkey,           }, 'e',
		function ()
			awful.prompt.run(
				{ prompt = 'Run Lua code: ' },
				mypromptbox[mouse.screen].widget,
				awful.util.eval, nil,
				awful.util.getdir('cache') .. '/history_eval'
			)
		end),
	-- Menubar
	awful.key({ modkey,           }, 'p',     function () menubar.show() end),
	-- Glbal commands
	awful.key({ 'Mod1', 'Shift'   }, 'F12',   function () awful.util.spawn('/home/von/vscripts/compton_toggle', false) end),
	awful.key({ modkey,           }, 'x',     function () awful.util.spawn(terminal) end),
	awful.key({ modkey,           }, 'q',     function () awful.util.spawn('/home/von/.local/bin/ticket_watch', false) end),
	awful.key({ modkey,           }, 'z',     function () awful.util.spawn('bash -c "until i3lock -entc 661111 -i /home/von/Pictures/wallpaper.png; do :; done"') end),
	awful.key({ modkey,           }, 'F6',    function () awful.util.spawn('/home/von/touchpad_hotkey.sh', false) end),
	awful.key({                   }, 'Print', function () awful.util.spawn('xfce4-screenshooter -ws /home/von/screenshots') end),
	awful.key({ modkey,           }, 'Print', function () awful.util.spawn('xfce4-screenshooter -fs /home/von/screenshots') end)
)

clientkeys = awful.util.table.join(
	awful.key({ modkey, 'Shift'   }, 'c',      function (c) c:kill()                         end),
	awful.key({ modkey,           }, 'Return', function (c) c:swap(awful.client.getmaster()) end),
	awful.key({ modkey, 'Shift'   }, 'o',      awful.client.movetoscreen                        ),
	awful.key({ modkey,           }, 'n',      function (c) c.minimized = true               end),

	-- Window properties
	awful.key({ 'Mod1',           }, 'Return', function (c) c.fullscreen = not c.fullscreen  end),
	awful.key({ modkey,           }, 'b',
		function (c)
			if c.border_width ~= 0 then
				c.border_width = 0
			else
				c.border_width = beautiful.border_width
			end
		end),
	awful.key({ modkey, 'Shift'   }, 't',      function (c) c.ontop = not c.ontop            end),
	awful.key({ modkey, 'Shift'   }, 's',      function (c) c.sticky = not c.sticky          end),
	awful.key({ modkey,           }, '`',      awful.client.floating.toggle                     ),
	awful.key({ modkey, 'Shift'   }, 'm',
		function (c)
			c.maximized_horizontal = not c.maximized_horizontal
			c.maximized_vertical   = not c.maximized_vertical
			if c.maximized_horizontal == true and c.maximized_vertical == true then
				c.border_width = 0
				client.focus:raise()
			else
				c.border_width = beautiful.border_width
			end
		end)
)

-- Bind all key numbers to tags.
-- Be careful: we use keycodes to make it works on any keyboard layout.
-- This should map on the top row of your keyboard, usually 1 to 9.
for i = 1, 9 do
	globalkeys = awful.util.table.join(globalkeys,
		awful.key({ modkey }, '#' .. i + 9,
			function ()
				local screen = mouse.screen
				local tag = awful.tag.gettags(screen)[i]
				if tag then
					if awful.tag.selected(screen) == tag and awful.tag.selectedlist(screen)[2] == nil then
						awful.tag.history.restore()
					else
						awful.tag.viewonly(tag)
					end
				end
			end),
		awful.key({ modkey, 'Control' }, '#' .. i + 9,
			function ()
				local screen = mouse.screen
				local tag = awful.tag.gettags(screen)[i]
				if tag then
					awful.tag.viewtoggle(tag)
				end
			end),
		awful.key({ modkey, 'Shift' }, '#' .. i + 9,
			function ()
				if client.focus then
					local tag = awful.tag.gettags(client.focus.screen)[i]
					if tag then
						awful.client.movetotag(tag)
					end
				end
			end),
		awful.key({ modkey, 'Control', 'Shift' }, '#' .. i + 9,
			function ()
				if client.focus then
					local tag = awful.tag.gettags(client.focus.screen)[i]
					if tag then
						awful.client.toggletag(tag)
					end
				end
			end)
	)
end

clientbuttons = awful.util.table.join(
	awful.button({        }, 1, function (c) client.focus = c; c:raise() end),
	awful.button({ modkey }, 1, function (c) client.focus = c; awful.mouse.client.move(c, snap) end),
	awful.button({ modkey }, 3, awful.mouse.client.resize)
)

-- Set keys
root.keys(globalkeys)
-- }}}
-- {{{ Rules
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,
			keys = clientkeys,
			buttons = clientbuttons
		}
	},
	-- Floating only rules:
	{
		rule_any = {
			class = {
				'Deadbeef',
				'Google-musicmanager',
				'mpv',
				'Pavucontrol',
				'pinentry',
				'plugin-container',
				'Skype',
				'Vncviewer'
			},
			instance = {
				'sun-awt-X11-XFramePeer'
			}
		},
		properties = {
			floating = true
		}
	},
	-- Floating borderless, because winamp nostalgia:
	{
		rule_any = {
			class = {
				'Audacious',
				'Qmmp'
			}
		},
		properties = {
			border_width = 0,
			floating = true
		}
	},
	-- Per app rules
	-- firefox
	{
		rule = { class = 'Firefox' },
		except = { instance = 'Navigator' },
		properties = { floating = true }
	},
	-- ardour3
	{
		rule = { class = 'Ardour' },
		except = { instance = 'ardour_editor' },
		properties = { floating = true }
	},
	-- Remove gaps between terminal windows:
	{
		rule_any = {
			class = {
				'Roxterm'
			}
		},
		properties = { size_hints_honor = false }
	},
	-- Specific desktops rules: place windows only on specific tags by default
	-- tag 4: games = maximized
	{
		rule_any = {
			instance = {
				'BaldursGate',
				'BaldursGateII',
				'IcewindDale'
			}
		},
		properties = {
			border_width = 0,
			maximized = true,
			tag = tags[1][4]
		}
	},
	-- tag 4: steam and games
	{
		rule_any = {
			class = { 'Steam' },
			instance = { 'Steam.exe' }
		},
		properties = { tag = tags[1][4] }
	},
	{
		rule_any = {
			class = {
				'Awesomenauts.bin.x86',
				'Civ5XP',
				'CivBE',
				'Cities In Motion.bin',
				'ck2',
				'csgo_linux',
				'DefenseGrid2',
				'deponia_tcj',
				'dota_linux',
				'dota2',
				'game.x86_64',
				'hl2_linux',
				'Pandora',
				'Strife',
				'Symphony.bin.x86_64',
				'eu4',
				'witcher2'
			},
			instance = {
				'Civ4BeyondSword.exe',
				'GameApp.exe',
				'KB.exe',
				'nwn2main.exe'
			},
			name = {
				'Cities in Motion 2',
				'GunsOfIcarusOnline',
				'Hand of Fate',
				'Serious Sam 3 - Linux'
			}
		},
		properties = {
			border_width = 0,
			floating = true,
			tag = tags[1][4]
		}
	}
}
-- }}}
-- {{{ Signals
-- Signal function to execute when a new client appears.
client.connect_signal(
	'manage',
	function (c, startup)
		-- Enable sloppy focus
		c:connect_signal(
			'mouse::enter',
			function(c)
				if awful.layout.get(c.screen) ~= layouts.magnifier[1] and awful.client.focus.filter(c) then
					client.focus = c
				end
			end
		)

		if not startup then
			-- Set the windows at the slave,
			-- i.e. put it at the end of others instead of setting it master.
			awful.client.setslave(c)

			-- Put windows in a smart way, only if they do not set an initial position.
			if not c.size_hints.user_position and not c.size_hints.program_position then
				awful.placement.no_overlap(c)
				awful.placement.no_offscreen(c)
			end
		end
	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)

-- save floating layout parameters for windows created and managed during the current session
tag.connect_signal(
	'property::layout',
	function(t)
		for k, c in ipairs(t:clients()) do
			if awful.layout.get(mouse.screen) == awful.layout.suit.floating then
				c:geometry(awful.client.property.get(c, 'floating_geometry'))
			end
		end
	end
)

client.connect_signal(
	'manage',
	function(c)
		if awful.layout.get(mouse.screen) == awful.layout.suit.floating then
			awful.client.property.set(c, 'floating_geometry', c:geometry())
		end
	end
)

client.connect_signal(
	'property::geometry',
	function(c)
		if awful.layout.get(mouse.screen) == awful.layout.suit.floating then
			awful.client.property.set(c, 'floating_geometry', c:geometry())
		end
	end
)
-- }}}
-- {{{ Autostart
-- don't forget you sync this file
-- this shit runs every time you restart your wm, dumbass.
---- set keyboard layouts
awful.util.spawn_with_shell('setxkbmap -layout us,ru -variant altgr-intl,typewriter -option ctrl:nocaps,grp:win_space_toggle,grp_led:scroll,compose:menu')
---- populate xrdb with .Xresources config
if exists('/home/von/.Xresources') then
	awful.util.spawn_with_shell('xrdb /home/von/.Xresources')
end
---- fix lack of terminus font
if exists('/home/von/.fonts/terminus') then
	awful.util.spawn_with_shell('xset fp- /home/von/.fonts/terminus/PCF')
	awful.util.spawn_with_shell('xset fp+ /home/von/.fonts/terminus/PCF')
end
---- execute all the other shit, installation specific
if exists('/home/von/.autostart') then
	awful.util.spawn_with_shell('/home/von/.autostart')
end
-- }}}