-- Standard awesome library
require("awful")
require("awful.autofocus")
require("awful.rules")
-- Theme handling library
require("beautiful")
-- Notification library
require("naughty")

require("wicked")
require("obvious.battery")

-- Load Debian menu entries
require("debian.menu")

-- {{{ Variable definitions
-- Themes define colours, icons, and wallpapers
-- theme_path = "/usr/share/awesome/themes/zenburn/theme.lua"
theme_path = os.getenv("HOME") .. "/.awesome/theme/theme.lua"

-- Actually load theme
beautiful.init(theme_path)

-- This is used later as the default terminal and editor to run.
terminal = "x-terminal-emulator"
lock = 'xscreensaver-command -lock'

-- 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.
layouts =
{
    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.max,
    -- awful.layout.suit.max.fullscreen,
    awful.layout.suit.magnifier,
    awful.layout.suit.floating
}

-- {{{ Markup helper functions
-- Inline markup is a tad ugly, so use these functions
-- to dynamically create markup.
function bg(color, text)
    if color == Nil then
      print "bg: Color is Nill"
    end
    return '<bg color="'..color..'" />'..text
end

function fg(color, text)
    if color == Nil then
      print "fg: Color is Nill"
    end
    return '<span color="'..color..'">'..text..'</span>'
end

function font(font, text)
    return '<span font_desc="'..font..'">'..text..'</span>'
end

function title()
    return '<title />'
end

function title_normal()
    return bg(beautiful.bg_normal, fg(beautiful.fg_normal, title()))
end

function title_focus()
    return bg(beautiful.bg_focus, fg(beautiful.fg_focus, title()))
end

function title_urgent()
    return bg(beautiful.bg_urgent, fg(beautufil.fg_urgent, title()))
end

function heading(text)
    return fg(beautiful.fg_focus, text)
end
-- }}}

-- {{{ Tags
-- Define tags table.
tags = {}
tagnames = { "WWW", "Mail", "3", "4", "5", "6", "7", "8", "9" }
for s = 1, screen.count() do
    tags[s] = awful.tag(tagnames, s, awful.layout.suit.tile)

    -- Set different default layouts per tag
    awful.layout.set(awful.layout.suit.max, tags[s][1])
    awful.layout.set(awful.layout.suit.fair, tags[s][2])
end
-- }}}

-- {{{ Wibox

-- Create a laucher widget and a main menu
myawesomemenu = {
   { "manual", terminal .. " -e man awesome" },
   { "edit config", "vim " .. awful.util.getdir("config") .. "/rc.lua" },
   { "restart", awesome.restart },
   { "quit", awesome.quit }
}

mymainmenu = awful.menu({ items = { { "awesome", myawesomemenu, beautiful.awesome_icon },
                                    { "Debian", debian.menu.Debian_menu.Debian },
                                    { "open terminal", terminal }
                                  }
                        })

mylauncher = awful.widget.launcher({ image = image(beautiful.awesome_icon),
                                     menu = mymainmenu })
-- }}}

-- {{{ Wibox
-- Create a textclock widget
mytextclock = awful.widget.textclock({ align = "right" }, " %d-%m-%Y %H:%M:%S ", 1)

 -- Create a systray
mysystray = widget({ type = "systray" })

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

mytasklist = {}
mytasklist.buttons = awful.util.table.join(
                     awful.button({ }, 1, function (c)
                                              if not c:isvisible() then
                                                  awful.tag.viewonly(c:tags()[1])
                                              end
                                              client.focus = c
                                              c:raise()
                                          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))
-- {{{ spacerwidget

spacerwidget = widget({ type = 'textbox', name = 'spacerwidget', align = 'right' })
spacerwidget.text = " | "

-- }}}

-- {{{ CPU Usage Widget

cputextwidget = widget({
    type = 'textbox',
    name = 'cputextwidget',
    align = 'right'
})

cputextwidget.text = heading('CPU')..': 00% '
wicked.register(cputextwidget, 'cpu', function (widget, args)
    local r = tonumber(args[1])
    local percent = args[1]..'%'
    if r < 10 then
        percent = '0'..percent
    end
    if r < 25 then
        percent = fg('green', percent)
    elseif r < 50 then
        percent = fg('yellow', percent)
    elseif r < 75 then
        percent = fg('orange', percent)
    else
        percent = fg('red', percent)
    end
    return heading('CPU')..': '..percent..' '
end, 2)

-- }}}
-- {{{ CPU Graph Widget

cpugraphwidget = widget({
    type = 'graph',
    name = 'cpugraphwidget',
    align = 'right'
})

cpugraphwidget.height = 0.85
cpugraphwidget.width = 45
cpugraphwidget.bg = '#333333'
cpugraphwidget.border_color = '#0a0a0a'
cpugraphwidget.grow = 'left'

cpugraphwidget:plot_properties_set('cpu', {
    fg = '#AEC6D8',
    fg_center = '#285577',
    fg_end = '#285577',
    vertical_gradient = false
})

wicked.register(cpugraphwidget, 'cpu', '$1', 1, 'cpu')

-- }}}
-- {{{ Memory Usage Widget

memtextwidget = widget({
  type = 'textbox',
  name = 'memtextwidget',
  align = 'right'
})

memtextwidget.text = heading('MEM')..': '
wicked.register(memtextwidget, 'mem', function (widget, args)
  -- Add extra preceding zeroes when needed
  local r = tonumber(args[1])
  local percent = args[1]..'%'
  if r < 10 then
    percent = '0'..percent
  end
  if r < 50 then
    percent = fg('green', percent)
  elseif r < 80 then
    percent = fg('orange', percent)
  else
    percent = fg('red', percent)
  end
  return heading('MEM')..': '..percent..' '..args[2]..'M'
end, 2)

--- }}}

for s = 1, screen.count() do
    -- Create a promptbox for each screen
    mypromptbox[s] = awful.widget.prompt({ layout = awful.widget.layout.horizontal.leftright })
    -- 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.label.all, mytaglist.buttons)

    -- Create a tasklist widget
    mytasklist[s] = awful.widget.tasklist(function(c)
                                              return awful.widget.tasklist.label.currenttags(c, s)
                                          end, mytasklist.buttons)

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

    -- Add widgets to the wibox - order matters
    mywibox[s].widgets = {
      {
        mytaglist[s],
        mypromptbox[s],
        mylayoutbox[s],
        layout = awful.widget.layout.horizontal.leftright
      },
      mysystray,
      mytextclock,
      spacerwidget,
      memtextwidget,
      spacerwidget,
      cpugraphwidget,
      cputextwidget,
      spacerwidget,
      obvious.battery(),
      mytasklist[s],
      layout = awful.widget.layout.horizontal.rightleft
    }
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)
))
-- }}}

function myssh(host)
    awful.util.spawn(terminal .. " -e ssh " .. host)
end

function mysshcompletion(input, cur_pos, ncomp)
    -- Read ~/.ssh/known_hosts and try to complete the host.
    local f = io.open(os.getenv("HOME") .. "/.ssh/known_hosts")

    if not f
    then
        print("Failed to open known_hosts")
    end

    local output = {}
    while true do
        local match_count = 0
        local line = f:read("*line")
        if not line then break end

        match = string.match(line, "^(" .. input .. ".-)[, ]")
        if match
        then
            table.insert(output, match)
        end
    end

    f:close()

    -- If nothing was found stop
    if #output == 0
    then
        return input, cur_pos
    end

    -- cycle
    while ncomp > #output do
        ncomp = ncomp - #output
    end

    return output[ncomp], #output[ncomp] + 1
end

-- {{{ 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(true)        end),

    awful.key({ modkey,           }, "Tab",
        function ()
            awful.client.focus.byidx( 1)
            if client.focus then client.focus:raise() end
        end),
    awful.key({ modkey, "Shift"   }, "Tab",
        function ()
            awful.client.focus.byidx(-1)
            if client.focus then client.focus:raise() end
        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),

    -- 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),

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

    awful.key({ modkey }, "F2",
        function ()
          awful.prompt.run({ prompt = "Host: " },
          mypromptbox[mouse.screen].widget,
          myssh,
          mysshcompletion)
        end),

    awful.key({ modkey }, "F3",
        function ()
          awful.util.spawn(terminal .. " -e ssh sah")
        end),

    awful.key({ modkey }, "F4",
        function ()
            awful.prompt.run({ prompt = "Run Lua code: " },
            mypromptbox[mouse.screen].widget,
            awful.util.eval, awful.prompt.bash,
            awful.util.getdir("cache") .. "/history_eval")
        end),

    awful.key({ modkey }, "F12",
        function ()
          awful.util.spawn(lock)
        end),
    awful.key({ modkey }, "c",
      function()
        awful.util.spawn("mpc toggle")
      end),
    awful.key({ modkey }, "v",
      function()
        awful.util.spawn("mpc next")
      end),
    awful.key({ modkey }, "x",
      function()
        awful.util.spawn("mpc prev")
    end)
)

-- Client awful tagging: this is useful to tag some clients and then do stuff like move to tag on them
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, "Shift"   }, "r",      function (c) c:redraw()                       end),
    awful.key({ modkey,           }, "n",      function (c) c.minimized = not c.minimized    end),
    awful.key({ modkey,           }, "t",      function (c) c.ontop = not c.ontop            end),
    awful.key({ modkey,           }, "m",
        function (c)
            c.maximized_horizontal = not c.maximized_horizontal
            c.maximized_vertical   = not c.maximized_vertical
        end)
)

-- Compute the maximum number of digit we need, limited to 9
keynumber = 0
for s = 1, screen.count() do
   keynumber = math.min(9, math.max(#tags[s], keynumber));
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, keynumber do
    globalkeys = awful.util.table.join(globalkeys,
        awful.key({ modkey }, "#" .. i + 9,
                   function ()
                         local screen = mouse.screen
                         if tags[screen][i] then
                             awful.tag.viewonly(tags[screen][i])
                         end
                  end),
        awful.key({ modkey, "Control" }, "#" .. i + 9,
                   function ()
                       local screen = mouse.screen
                       if tags[screen][i] then
                          awful.tag.viewtoggle(tags[screen][i])
                       end
                  end),
        awful.key({ modkey, "Shift" }, "#" .. i + 9,
                   function ()
                       if client.focus and tags[client.focus.screen][i] then
                           awful.client.movetotag(tags[client.focus.screen][i])
                       end
                  end),
        awful.key({ modkey, "Control", "Shift" }, "#" .. i + 9,
                   function ()
                       if client.focus and tags[client.focus.screen][i] then
                           awful.client.toggletag(tags[client.focus.screen][i])
                       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 = true,
                     keys = clientkeys,
                     buttons = clientbuttons } },

  -- Table of clients that should be set floating. The index may be either
  -- the application class or instance. The instance is useful when running
  -- a console app in a terminal like (Music on Console)
  --    x-terminal-emulator -name mocp -e mocp
  { rule = { class = "MPlayer" },
  properties = { floating = true, ontop = true } },
  { rule = { class = "pinentry" },
  properties = { floating = true } },
  { rule = { class = "gimp" },
  properties = { floating = true } },
  { rule = { class = "vlc" },
  properties = { floating = true } },
  { rule = { class = "QEMU" },
  properties = { floating = true, ontop = true } },
  { rule = { class = "speedcrunch" },
  properties = { floating = true } },
  { rule = { name = "Meld" },
  properties = { maximized_vertical = true, maximized_horizontal = true } },

  -- Applications to be moved to a pre-defined tag by class or instance.
  -- Use the screen and tags indices.
  { rule = { instance = "opera" },
  properties = { tag = tags[1][1] } },
  { rule = { class = "Firefox" },
  properties = { tag = tags[1][1] } },
  { rule = { class = "Iceweasel" },
  properties = { tag = tags[1][1] } },
  { rule = { class = "Liferea" },
  properties = { tag = tags[1][1] } },
  { rule = { instance = "akregator" },
  properties = { tag = tags[1][1] } },
  { rule = { instance = "irssi" },
  properties = { tag = tags[1][2] } },
  { rule = { class = "vmware" },
  properties = { tag = tags[1][9] } },
}
-- }}}

-- {{{ Signals
-- Signal function to execute when a new client appears.
client.add_signal("manage", function (c, startup)
  -- Add a titlebar
  -- awful.titlebar.add(c, { modkey = modkey })

  -- Enable sloppy focus
  -- c:add_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
end)

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