-- 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")
-- my dbus bindings
local ldbusw = require("ldbusw")

-- {{{ 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("/home/q3k/.config/awesome/theme.lua")

-- This is used later as the default terminal and editor to run.
terminal = "urxvt"
editor = os.getenv("EDITOR") or "nano"
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"

-- Table of layouts to cover with awful.layout.inc, order matters.
local 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.max,
}
-- }}}

-- {{{ Wallpaper
if beautiful.wallpaper then
    for s = 1, screen.count() do
        gears.wallpaper.maximized(beautiful.wallpaper, s, true)
    end
end
-- }}}

-- {{{ Tags
-- Define a tag table which hold all screen tags.
tags = {}
for s = 1, screen.count() do
    -- Each screen has its own tag table.
    tags[s] = awful.tag({ " 零 ", " 一 ", " 二 ", " 三 ", " 四 ", " 五 ", " www ", " xmpp ", " irc ", " mail " }, s, layouts[1])
end
-- }}}

-- {{{ Menu
-- Create a laucher widget and a main menu
myawesomemenu = {
   { "manual", terminal .. " -e man awesome" },
   { "edit config", editor_cmd .. " " .. awesome.conffile },
   { "restart", awesome.restart },
   { "quit", awesome.quit }
}

mymainmenu = awful.menu({ items = { { "awesome", myawesomemenu, beautiful.awesome_icon },
                                    { "open terminal", terminal }
                                  }
                        })
mylauncher = awful.widget.launcher({ image = beautiful.awesome_icon,
                                     menu = mymainmenu })

-- Menubar configuration
menubar.utils.terminal = terminal -- Set the terminal for applications that require it
-- }}}

-- {{{ Wibox
-- Create a textclock widget
mytextclock = awful.widget.textclock()

-- 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),
                    awful.button({ }, 4, function(t) awful.tag.viewnext(awful.tag.getscreen(t)) end),
                    awful.button({ }, 5, function(t) awful.tag.viewprev(awful.tag.getscreen(t)) end)
                    )
mytasklist = {}
mytasklist.buttons = awful.util.table.join(
                     awful.button({ }, 1, function (c)
                                              if c == client.focus then
                                                  c.minimized = true
                                              else
                                                  -- 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))

-- WIDGETS
function string:split(sep)
        local sep, fields = sep or ":", {}
        local pattern = string.format("([^%s]+)", sep)
        self:gsub(pattern, function(c) fields[#fields+1] = c end)
        return fields
end

q3k = {}

-- hack!
function q3k.GetProcessOutput(process)
    local File = io.open("/tmp/awq3kres", "r")
    while File ~= nil do
        File:close()
        awful.util.spawn_with_shell("rm /tmp/awq3kres")
        File = io.open("/tmp/awq3kres", "r")
    end

    awful.util.spawn_with_shell('bash -c "' .. process ..'" > /tmp/awq3kres')

    File = io.open("/tmp/awq3kres", "r")
    -- spinlock!
    while File == nil do
        File = io.open("/tmp/awq3kres", "r")
    end
    File:close()
    File = io.open("/tmp/awq3kres", "r")
    local Data = File:read("*all")
    File:close()

    return Data
end

function q3k.MemoryStats()
    local File = io.open("/proc/meminfo", "r")
    local Data = File:read("*all")
    File:close()

    local MemInfo = {}
    Data = Data:split("\n")
    for _, V in pairs(Data) do
        local Label = V:split(":")[1]
        local Columns = V:split(" ")
        local Value = Columns[#Columns - 1]

        MemInfo[Label] = Value
    end

    return tonumber(MemInfo["MemTotal"]), tonumber(MemInfo["MemFree"]) + tonumber(MemInfo["Cached"])
end

function q3k.ReadBatteryField(name)
	local File = io.open("/sys/class/power_supply/BAT0/"..name, "r")
    if not File then
        return None
    end
	local Data = File:read("*all")
	File:close()
	return Data
end

function q3k.ReadBatteryNumber(name)
    local Value = q3k.ReadBatteryField(name)
    if not Value then
        return nil
    end
    return tonumber(Value)/1000
end

-- I copied this logic from the acpi utility source.
-- There is no fucking way I will ever have any silightest clue about why
-- and how this works.
function q3k.PowerStats()
    local Status = q3k.ReadBatteryField("status")
    local EnergyNow = q3k.ReadBatteryNumber("energy_now")
    if not EnergyNow then
        EnergyNow = q3k.ReadBatteryNumber("charge_now")
    end
    local EnergyFull = q3k.ReadBatteryNumber("energy_full")
    if not EnergyFull then
        EnergyFull = q3k.ReadBatteryNumber("charge_full")
    end
    local Percent = string.format("%i%%", (EnergyNow / EnergyFull * 100))

    local FormatTimeLeft = function(s)
        local m = s / 60
        local h = m / 60
        return string.format("%02i:%02i", h, m%60)
    end

    local Voltage = q3k.ReadBatteryNumber("voltage_now")
    local Rate = q3k.ReadBatteryNumber("power_now")
    if not Rate then
        Rate = q3k.ReadBatteryNumber("current_now")
    end
    local RemainingCapacity = q3k.ReadBatteryNumber("charge_now")
    if not RemainingCapacity then
        RemainingCapacity = EnergyNow
    else
        RemainingCapacity = EnergyNow * 1000 / Voltage
        Rate = Rate * 1000 / Voltage
    end
    if Status:sub(1,3) == "Dis" then
        local Seconds = 3600 * RemainingCapacity / Rate
        local Left = "↓" .. FormatTimeLeft(Seconds)
        return Percent, Left
    elseif Status:sub(1,3) == "Cha" then
        local Seconds = 3600 * (EnergyFull - RemainingCapacity) / Rate
        local Left = "↑" .. FormatTimeLeft(Seconds)
        return Percent, Left
    end

    return Percent, "?"
end

function q3k.AtHackerspace()
    local Response = {}
    http.request({ url="http://at.hackerspace.pl/api", method="GET", sink=ltn12.sink.table(Response) })

    if (#Response ~= 1) then
        return "?"
    end

    local Data = json.decode(Response[1])
    local Users = {}
    for _, User in pairs(Data.users) do
        Users[#Users + 1] = User.login
    end

    if (#Users == 0) then
        return "nobody"
    else
        return table.concat(Users, ", ")
    end
end

local SystemBus = ldbusw.Bus:Connect("system")
local SessionBus = ldbusw.Bus:Connect("session")
local Wireless = SystemBus:GetObject("org.wicd.daemon", "/org/wicd/daemon/wireless")
local Wired = SystemBus:GetObject("org.wicd.daemon", "/org/wicd/daemon/wired")

function q3k.NetworkStatus()
	if Wired.IsWiredUp() and Wired.GetWiredIP() then
		local IP =  Wired.GetWiredIP()
		if not IP then
			IP = "?"
		end
		return string.format("🔗 %s", IP)
	elseif Wireless.GetCurrentDBMStrength() ~= 0 then
		local Name, Error = Wireless.GetCurrentNetwork()
		if not Name then
			return "📶 ?"
		end
		local Strength, Error = Wireless.GetCurrentDBMStrength()

		return string.format("📶 %s %sdBm %s", Name, Strength, Wireless.GetWirelessIP())
	else
		return "⛔"
	end
end

-- pulse's dbus api sucks balls

function q3k:PulseStatus()
	local function OpenPulseBus()
		local Lookup = SessionBus:GetObject("org.PulseAudio1", "/org/pulseaudio/server_lookup1")
		local Socket = Lookup.Get("org.PulseAudio.ServerLookup1", "Address")
		if not Socket then
			return nil, nil
		end
		local PulseBus = ldbusw.Bus:Open(Socket)
		return Socket, PulseBus
	end
	if not self.PulseBus then
		self.PulseSocket, self.PulseBus = OpenPulseBus()
		if not self.PulseBus then
			return nil, "n/s"
		end
	end

	local PulseDevice = self.PulseBus:GetObject("org.PulseAudio", "/org/pulseaudio/core1")
	if not PulseDevice then
		self.PulseBus = nil
		return nil, "n/b"
	end
	local Sinks = PulseDevice.Get("org.PulseAudio.Core1", "Sinks")
        if not Sinks then
		return nil, "n/o"
        end
	if #Sinks < 1 then
		return nil, "n/d"
	end
	local Sink = self.PulseBus:GetObject("org.PulseAudio", Sinks[1])
	local Volume = Sink.Get("org.PulseAudio.Core1.Device", "Volume")
	local Muted = Sink.Get("org.PulseAudio.Core1.Device", "Mute")
	-- average the volume
	local AverageVolume = 0
	for k, v in pairs(Volume) do
		AverageVolume = AverageVolume + v
	end
	AverageVolume = AverageVolume / #Volume
	-- return in percent
	return (AverageVolume * 100 ) / 65536, Muted
end

function q3k.UniBlock(Percent)
    local i = Percent * 8 / 100
    if i < 1 then
        return " "
    end
    return string.char(0xe2) .. string.char(0x96) .. string.char(0x81 + (i - 1))
end

sysusage = wibox.widget.textbox()
sysusage:set_text("...")

sysusage_timer = timer({ timeout = 3 })
sysusage_timer:connect_signal("timeout", function()
    local File = io.open("/proc/loadavg", "r")
    local Data = File:read("*all")
    File:close()

    local Load = Data:split(" ")
    local LoadText = "⚙ " ..  Load[1]

    local MemoryTotal, MemoryFree = q3k.MemoryStats()
    local MemoryText = string.format("⛁ %im", MemoryFree/1000)

    local PowerPercent, PowerLeft = q3k.PowerStats()
    local PowerText = string.format("⚡ %s (%s)", PowerPercent, PowerLeft)

    local NetworkText = q3k.NetworkStatus()
    local PulseVolume, PulseMuted = q3k:PulseStatus()
    local PulseText = ""
    if PulseVolume == nil then
        PulseText = "🔊 " .. PulseMuted
    else
        if PulseMuted then
            PulseText = "🔊 ░"
        else
        	PulseText = "🔊 " .. q3k.UniBlock(PulseVolume)
        end
    end

    --local HSStatus = q3k.AtHackerspace()
    --local HSText = string.format("hs: %s", HSStatus)

    sysusage:set_text(" " .. PulseText .. " | " .. NetworkText .. " | " .. LoadText .. "  | " .. PowerText .. " |")
    --sysusage.text = " " .. LoadText .. " | " .. MemoryText .. " | " .. PowerText .. " | " .. HSText
end)
sysusage_timer:start()

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 () awful.layout.inc(layouts, 1) end),
                           awful.button({ }, 3, function () awful.layout.inc(layouts, -1) end),
                           awful.button({ }, 4, function () awful.layout.inc(layouts, 1) end),
                           awful.button({ }, 5, function () awful.layout.inc(layouts, -1) end)))
    -- Create a taglist widget
    mytaglist[s] = awful.widget.taglist(s, awful.widget.taglist.filter.all, 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", screen = s })

    -- Widgets that are aligned to the left
    local left_layout = wibox.layout.fixed.horizontal()
    left_layout:add(mylauncher)
    left_layout:add(mytaglist[s])
    left_layout:add(mypromptbox[s])

    -- Widgets that are aligned to the right
    local right_layout = wibox.layout.fixed.horizontal()
    if s == 1 then right_layout:add(wibox.widget.systray()) end
    right_layout:add(sysusage)
    right_layout:add(mytextclock)
    right_layout:add(mylayoutbox[s])

    -- 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),
    awful.button({ }, 4, awful.tag.viewnext),
    awful.button({ }, 5, awful.tag.viewprev)
))
-- }}}

-- {{{ Key bindings
globalkeys = awful.util.table.join(
    awful.key({ modkey,           }, "Left",   awful.tag.viewprev       ),
    awful.key({ modkey,           }, "Right",  awful.tag.viewnext       ),
    awful.key({ modkey,           }, "Escape", awful.tag.history.restore),

    awful.key({ modkey,           }, "j",
        function ()
            awful.client.focus.byidx( 1)
            if client.focus then client.focus:raise() end
        end),
    awful.key({ modkey,           }, "k",
        function ()
            awful.client.focus.byidx(-1)
            if client.focus then client.focus:raise() end
        end),
    awful.key({ modkey,           }, "w", function () mymainmenu:show() 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, "Control" }, "j", function () awful.screen.focus_relative( 1) end),
    awful.key({ modkey, "Control" }, "k", function () awful.screen.focus_relative(-1) end),
    awful.key({ modkey,           }, "u", awful.client.urgent.jumpto),
    awful.key({ modkey,           }, "Tab",
        function ()
            awful.client.focus.history.previous()
            if client.focus then
                client.focus:raise()
            end
        end),

    -- Standard program
    awful.key({ modkey,           }, "Return", function () awful.util.spawn(terminal) end),
    awful.key({ modkey, "Control" }, "r", awesome.restart),
    awful.key({ modkey, "Shift"   }, "q", awesome.quit),

    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,           }, "space", function () awful.layout.inc(layouts,  1) end),
    awful.key({ modkey, "Shift"   }, "space", function () awful.layout.inc(layouts, -1) end),

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

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

    awful.key({ modkey }, "x",
              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)
)

clientkeys = awful.util.table.join(
    awful.key({ modkey,           }, "f",      function (c) c.fullscreen = not c.fullscreen  end),
    awful.key({ modkey, "Shift"   }, "c",      function (c) c:kill()                         end),
    awful.key({ modkey, "Control" }, "space",  awful.client.floating.toggle                     ),
    awful.key({ modkey, "Control" }, "Return", function (c) c:swap(awful.client.getmaster()) end),
    awful.key({ modkey,           }, "o",      awful.client.movetoscreen                        ),
    awful.key({ modkey,           }, "t",      function (c) c.ontop = not c.ontop            end),
    awful.key({ modkey,           }, "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),
    awful.key({ modkey,           }, "m",
        function (c)
            c.maximized_horizontal = not c.maximized_horizontal
            c.maximized_vertical   = not c.maximized_vertical
        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
                           awful.tag.viewonly(tag)
                        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 ()
                      local tag = awful.tag.gettags(client.focus.screen)[i]
                      if client.focus and tag then
                          awful.client.movetotag(tag)
                     end
                  end),
        awful.key({ modkey, "Control", "Shift" }, "#" .. i + 9,
                  function ()
                      local tag = awful.tag.gettags(client.focus.screen)[i]
                      if client.focus and tag then
                          awful.client.toggletag(tag)
                      end
                  end))
end

clientbuttons = awful.util.table.join(
    awful.button({ }, 1, function (c) client.focus = c; c:raise() end),
    awful.button({ modkey }, 1, awful.mouse.client.move),
    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 } },
    { rule = { class = "MPlayer" },
      properties = { floating = true } },
    { rule = { class = "pinentry" },
      properties = { floating = true } },
    { rule = { class = "gimp" },
      properties = { floating = true } },
    { rule = { class = "URxvt" },
      properties = { size_hints_honor = false } },
    -- Set Firefox to always map on tags number 2 of screen 1.
    -- { rule = { class = "Firefox" },
    --   properties = { tag = tags[1][2] } },
}
-- }}}

-- {{{ 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) ~= awful.layout.suit.magnifier
            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 does 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

    local titlebars_enabled = false
    if titlebars_enabled and (c.type == "normal" or c.type == "dialog") then
        -- buttons for the titlebar
        local buttons = awful.util.table.join(
                awful.button({ }, 1, function()
                    client.focus = c
                    c:raise()
                    awful.mouse.client.move(c)
                end),
                awful.button({ }, 3, function()
                    client.focus = c
                    c:raise()
                    awful.mouse.client.resize(c)
                end)
                )

        -- Widgets that are aligned to the left
        local left_layout = wibox.layout.fixed.horizontal()
        left_layout:add(awful.titlebar.widget.iconwidget(c))
        left_layout:buttons(buttons)

        -- Widgets that are aligned to the right
        local right_layout = wibox.layout.fixed.horizontal()
        right_layout:add(awful.titlebar.widget.floatingbutton(c))
        right_layout:add(awful.titlebar.widget.maximizedbutton(c))
        right_layout:add(awful.titlebar.widget.stickybutton(c))
        right_layout:add(awful.titlebar.widget.ontopbutton(c))
        right_layout:add(awful.titlebar.widget.closebutton(c))

        -- The title goes in the middle
        local middle_layout = wibox.layout.flex.horizontal()
        local title = awful.titlebar.widget.titlewidget(c)
        title:set_align("center")
        middle_layout:add(title)
        middle_layout:buttons(buttons)

        -- Now bring it all together
        local layout = wibox.layout.align.horizontal()
        layout:set_left(left_layout)
        layout:set_right(right_layout)
        layout:set_middle(middle_layout)

        awful.titlebar(c):set_widget(layout)
    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)
-- }}}