#!/usr/bin/env python3
"""
  fritzbox_traffic - A munin plugin for Linux to monitor AVM Fritzbox WAN traffic
  Copyright (C) 2015 Christian Stade-Schuldt
  Author: Christian Stade-Schuldt

  Updated to fritzconnection library version 1.3.1
  Copyright (C) 2020 Oliver Edelamnn
  Author: Oliver Edelmann

  Like Munin, this plugin is licensed under the GNU GPL v2 license
  http://www.opensource.org/licenses/GPL-2.0
  Like Munin, this plugin is licensed under the GNU GPL v2 license
  http://www.opensource.org/licenses/GPL-2.0
  This plugin requires the fritzconnection plugin. To install it using pip:
  pip install fritzconnection
  This plugin supports the following munin configuration parameters:
  #%# family=auto contrib
  #%# capabilities=autoconf
"""

import os
import sys

from fritzconnection.lib.fritzstatus import FritzStatus

hostname = os.path.basename(__file__).split('_')[1]

def print_values():
    try:
        conn = FritzStatus(address=hostname, password=os.environ['fritzbox_password'])
    except Exception as e:
        print(e)
        sys.exit("Couldn't get WAN traffic")

    traffic =  conn.transmission_rate
    up = traffic[0]*8
    down = traffic[1]*8
    print('down.value %d' % down)

    print('up.value %d' % up)

    if not os.environ.get('traffic_remove_max') or "false" in os.environ.get('traffic_remove_max'):
        max_traffic = conn.max_bit_rate
        print('maxdown.value %d' % max_traffic[1])

        print('maxup.value %d' % max_traffic[0])


def print_config():
    try:
        conn = FritzStatus(address=hostname, password=os.environ['fritzbox_password'])
    except Exception as e:
        print(e)
        sys.exit("Couldn't get WAN traffic")

    max_traffic = conn.max_bit_rate

    print("host_name %s" % hostname)
    print("graph_title AVM Fritz!Box WAN traffic (received)")
    print("graph_args --base 1000")
    print("graph_vlabel bit down per ${graph_period}")
    print("graph_category network")
    print("graph_order down maxdown")
    print("down.label bps")
    print("down.type GAUGE")
    print("down.draw AREA")
    print("down.min 0")
    print("down.max %d" % max_traffic[1])
    print("down.warning %.0f" % (max_traffic[1]*0.6))
    print("down.critical %.0f" % (max_traffic[1]*0.8))
    print("down.info Traffic of the WAN interface.")
    if not os.environ.get('traffic_remove_max') or "false" in os.environ.get('traffic_remove_max'):
        print("maxdown.label MAX")
        print("maxdown.type GAUGE")
        print("maxdown.graph LINE1")
        print("maxdown.info Maximum down speed of the WAN interface.")
    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")  # Some docs say it'll be called with fetch, some say no arg at all
    elif len(sys.argv) == 1 or (len(sys.argv) == 2 and sys.argv[1] == 'fetch'):
        try:
            print_values()
        except:
            sys.exit("Couldn't retrieve fritzbox traffic")