#!/usr/bin/python3
# vim: set fileencoding=utf-8 :
#
# Munin plugin to monitor the switches in the pellematic
#
# Copyright 2013 Guido Guenther <agx@sigxcpu.org>
#
# Licesnse: GPLv3
#
# depends: python-requests
#
# FIXME: This currently doesn't probe the connected systems
#
#%# capabilities=autoconf
#%# family=contrib

import requests
import os
import re
import sys
import json

username = os.getenv('username', 'oekofen')
password = os.getenv('password', username)
debug = os.getenv('debug')
url = os.getenv('url')

# The switches we want to plot (on/off 0/1 only):
items = { "CAPPL:FA[0].L_kap_sensor_raumentnahme":
          'Capacitive Sensor RA',
          "CAPPL:FA[0].L_kap_sensor_zwischenbehaelter":
          'Capacitive Sensor Hopper',
          "CAPPL:FA[0].L_br1":
          'Burner Require',
          # Brandschutzklappe
          "CAPPL:FA[0].L_bsk_status":
          'BSK Status',
          # Motor suction turbine
          "CAPPL:FA[0].ausgang_motor[0]":
          'Motor Suction Turbine',
          #"CAPPL:FA[0].ausgang_motor[1]":
          #'',
          #"CAPPL:FA[0].ausgang_motor[2]":
          #'',
          #"CAPPL:FA[0].ausgang_motor[3]":
          #'',
          #"CAPPL:FA[0].ausgang_motor[4]":
          #'',
          #"CAPPL:FA[0].ausgang_motor[5]":
          #'',
          #"CAPPL:FA[0].ausgang_motor[8]":
          #'',
          # DHW (domestic hot water) pump switch?
          #"CAPPL:FA[0].ausgang_motor[9]":
          # Pellet transport inside boiler
          "CAPPL:FA[0].ausgang_motor[11]":
          'Motor Delivery',
          "CAPPL:FA[0].ausgang_stoermelderelais":
          'Relay Malfunction',
         }

def canon(name):
    return re.sub(r"[^a-zA-Z0-9_]", "_", name)

def get_name(item):
    if item['name'] in list(items.keys()):
        if item['shortText'] != '???':
            return item['shortText']
        else:
            return items[item['name']]
    else:
        return None

def print_config(url):
    print("""graph_title Pellematic Switches 
graph_vlabel On (1) or Off (0)
graph_category Heating
graph_info This graph shows the values for different on/off switches
""")
    names = []
    draw = "AREA"
    out = fetch_raw(url)
    for item in out:
        name = get_name(item)
        if name:
            names.append(name)

    for name in sorted(names):
        print("%s.label %s" % (canon(name), name))
        print("%s.type GAUGE" % canon(name))
        print("%s.draw %s" % (canon(name), draw))
        if draw == "AREA":
            draw = "STACK"
        if 'Malfunction' in name:
            print("%s.warning 0.1" % canon(name))
            print("%s.critical 1" % canon(name))

def fetch_raw(url):
    # Perform authentication and get the cookie
    auth_formdata = {'language': 'en',
                     'username': username,
                     'password': password,
                     'submit': 'Login',
                     }
    r = requests.post(url, data=auth_formdata)

    pksession = r.cookies['pksession']
    cookies = { 'pksession': pksession,
                'language': 'en' }
    params = { 'action': 'get',
               'attr': '1',
             }

    # These are the headers used by the supplied application
    # It mostly doesn't matter what we pass but the backend
    # breaks with a lua error without a Accept-Language header:
    headers = {'Accept': 'application/json',
               'Content-Type': 'application/x-www-form-urlencoded; charset=UTF-8',
               'Accept-Language': 'en',
              }

    # The items to fetch are passed as simple json string
    payload = json.dumps(list(items.keys()))
    r = requests.post(url,
                      data=payload,
                      params=params,
                      cookies=cookies,
                      headers=headers)
    ret = r.json()
    if debug:
        print(ret, file=sys.stderr)

    return ret


def fetch_values(url):
    out = fetch_raw(url)
    for item in out:
        name = get_name(item)
        if name:
            print("%s.value %s" % (canon(name), item['value']))


def main(args):
    if not url:
        print("No url configured", file=sys.stderr)
        return 1
    if len(args) > 1:
        if args[1] in [ 'autoconf', 'detect' ]:
            try:
                fetch_raw(url)
                print("yes")
                return 0
            except:
                print("no")
                return 1
        elif args[1] == 'config':
            try:
                print_config(url)
            except Exception as e:
                raise
                print("Failed to fetch config: '%s'" % e, file=sys.stderr)
                return 1
            return 0
    try:
        fetch_values(url)
    except Exception as e:
        print("Failed to fetch values: '%s'" % e, file=sys.stderr)
        raise
        return 1
    return 0

if __name__ == "__main__":
    sys.exit(main(sys.argv))