#!/usr/bin/python3
###############################################################################
#                                                                             #
# collecty - A system statistics collection daemon for IPFire                 #
# Copyright (C) 2012 IPFire development team                                  #
#                                                                             #
# This program is free software: you can redistribute it and/or modify        #
# it under the terms of the GNU General Public License as published by        #
# the Free Software Foundation, either version 3 of the License, or           #
# (at your option) any later version.                                         #
#                                                                             #
# This program is distributed in the hope that it will be useful,             #
# but WITHOUT ANY WARRANTY; without even the implied warranty of              #
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the               #
# GNU General Public License for more details.                                #
#                                                                             #
# You should have received a copy of the GNU General Public License           #
# along with this program.  If not, see <http://www.gnu.org/licenses/>.       #
#                                                                             #
###############################################################################

import os

from . import base

from ..colours import *
from ..constants import *
from ..i18n import _

class GraphTemplateLoadAvg(base.GraphTemplate):
	name = "loadavg"

	@property
	def rrd_graph(self):
		rrd_graph = [
			"LINE2:load15%s:%s" % (
				YELLOW, LABEL % _("15 Minutes"),
			),
			"GPRINT:load15_cur:%s" % FLOAT,
			"GPRINT:load15_avg:%s" % FLOAT,
			"GPRINT:load15_min:%s" % FLOAT,
			"GPRINT:load15_max:%s\\j" % FLOAT,

			"LINE2:load5%s:%s" % (
				ORANGE, LABEL % _("5 Minutes"),
			),
			"GPRINT:load5_cur:%s" % FLOAT,
			"GPRINT:load5_avg:%s" % FLOAT,
			"GPRINT:load5_min:%s" % FLOAT,
			"GPRINT:load5_max:%s\\j" % FLOAT,

			"LINE2:load1%s:%s" % (
				RED, LABEL % _("1 Minute"),
			),
			"GPRINT:load1_cur:%s" % FLOAT,
			"GPRINT:load1_avg:%s" % FLOAT,
			"GPRINT:load1_min:%s" % FLOAT,
			"GPRINT:load1_max:%s\\j" % FLOAT,

			# Headline
			"COMMENT:%s" % EMPTY_LABEL,
			"COMMENT:%s" % (COLUMN % _("Current")),
			"COMMENT:%s" % (COLUMN % _("Average")),
			"COMMENT:%s" % (COLUMN % _("Minimum")),
			"COMMENT:%s\\j" % (COLUMN % _("Maximum")),
		]

		return rrd_graph

	lower_limit = 0

	@property
	def graph_title(self):
		return _("Load Average")

	@property
	def graph_vertical_label(self):
		return _("Load")

	@property
	def rrd_graph_args(self):
		return [
			"--legend-direction=bottomup",
		]


class LoadAvgObject(base.Object):
	rrd_schema = [
		"DS:load1:GAUGE:0:U",
		"DS:load5:GAUGE:0:U",
		"DS:load15:GAUGE:0:U",
	]

	@property
	def id(self):
		return "default"

	def collect(self):
		return os.getloadavg()


class LoadAvgPlugin(base.Plugin):
	name = "loadavg"
	description = "Load Average Plugin"

	templates = [GraphTemplateLoadAvg]

	@property
	def objects(self):
		return [LoadAvgObject(self)]