#!/usr/bin/env python
"""
  fritzbox_power_consumption - A munin plugin for Linux to monitor AVM Fritzbox
  Copyright (C) 2015 Christian Stade-Schuldt
  Author: Christian Stade-Schuldt
  Like Munin, this plugin is licensed under the GNU GPL v2 license
  http://www.opensource.org/licenses/GPL-2.0
  Add the following section to your munin-node's plugin configuration:

  [fritzbox_*]
  env.fritzbox_ip [ip address of the fritzbox]
  env.fritzbox_password [fritzbox password]
  
  This plugin supports the following munin configuration parameters:
  #%# family=auto contrib
  #%# capabilities=autoconf
"""

import os
import re
import sys
import fritzbox_helper as fh

PAGE = '/system/energy.lua'
pattern = re.compile('<td>(.+?)"bar\s(act|fillonly)"(.+?)\s(\d{1,3})\s%')
DEVICES = ['system', 'cpu', 'wifi', 'dsl', 'ab', 'usb']


def get_power_consumption():
    """get the current power consumption usage"""

    server = os.environ['fritzbox_ip']
    password = os.environ['fritzbox_password']

    sid = fh.get_sid(server, password)
    data = fh.get_page(server, sid, PAGE)
    matches = re.finditer(pattern, data)
    if matches:
        data = zip(DEVICES, [m.group(4) for m in matches])
        for d in data:
            print'%s.value %s' % (d[0], d[1])


def print_config():
    print "graph_title AVM Fritz!Box Power Consumption"
    print "graph_vlabel %"
    print "graph_category network"
    print "graph_order system cpu wifi dsl ab usb"
    print "system.label system"
    print "system.type GAUGE"
    print "system.graph LINE12"
    print "system.min 0"
    print "system.max 100"
    print "system.info Fritzbox overall power consumption"
    print "cpu.label cpu"
    print "cpu.type GAUGE"
    print "cpu.graph LINE1"
    print "cpu.min 0"
    print "cpu.max 100"
    print "cpu.info Fritzbox central processor power consumption"
    print "wifi.label wifi"
    print "wifi.type GAUGE"
    print "wifi.graph LINE1"
    print "wifi.min 0"
    print "wifi.max 100"
    print "wifi.info Fritzbox wifi power consumption"
    print "dsl.label dsl"
    print "dsl.type GAUGE"
    print "dsl.graph LINE1"
    print "dsl.min 0"
    print "dsl.max 100"
    print "dsl.info Fritzbox dsl power consumption"
    print "ab.label ab"
    print "ab.type GAUGE"
    print "ab.graph LINE1"
    print "ab.min 0"
    print "ab.max 100"
    print "ab.info Fritzbox analog phone ports power consumption"
    print "usb.label usb"
    print "usb.type GAUGE"
    print "usb.graph LINE1"
    print "usb.min 0"
    print "usb.max 100"
    print "usb.info Fritzbox usb devices power consumption"
    if os.environ.get('host_name'):
        print "host_name " + os.environ['host_name']


if __name__ == '__main__':
    if len(sys.argv) == 2 and sys.argv[1] == 'config':
        print_config()
    elif len(sys.argv) == 2 and sys.argv[1] == 'autoconf':
        print 'yes'
    elif len(sys.argv) == 1 or len(sys.argv) == 2 and sys.argv[1] == 'fetch':
        # Some docs say it'll be called with fetch, some say no arg at all
        try:
            get_power_consumption()
        except:
            sys.exit("Couldn't retrieve fritzbox power consumption")