#include <string.h>
#include "status.h"
#include "status.m"
#include <misc.h>
#include <tlmplib.h>
#include <private_msg.h>

static HELP_FILE help_stats ("status","diskstats");
static CONFIG_FILE fstats ("/proc/diskstats",help_stats,CONFIGF_OPTIONAL);

const int NBTITLE=4;

struct DISKSTATS{
	SSTRING dev;
	int vals[NBTITLE];
	DISKSTATS(){
		reset();
	}
	DISKSTATS(const DISKSTATS &v){
		dev = v.dev;
		memcpy (vals,v.vals,sizeof(vals));
	}
	void reset(){
		memset (vals,0,sizeof(vals));
	}
};

struct DISKSTATS_UI: public DISKSTATS{
	int diffs[20];
	int gauges[20];
	DISKSTATS_UI(){
		reset();
		memset (diffs,0,sizeof(diffs));
		memset (gauges,0,sizeof(gauges));
	}
};

<mod>
void diskstats_show()
{
	if (!fstats.exist()){
		xconf_error (MSG_U(E_DISKSTATS
			,"No disk statistic files available on this system.\n"
			 "Linuxconf was looking for file %s")
			,fstats.getpath());
		return;
	}
	glocal DISKSTATS stats;
	DIALOG dia;
	<obj COROUTINE co>();
	<f run>
		<call loadfile>(fstats,true);
		<f oneline>
			SSTRINGS ws;
			int n = str_splitline (line,' ',ws);
			if (n >= 7){
				const char *dev = ws.getitem(2)->get();
				glocal.stats.dev = dev;
				if (strncmp(dev,"ram",3)!=0
					&& strncmp (dev,"fd",2)!=0){
					glocal.stats.reset();
					int vals[n];
					for (int i=3; i<n; i++){
						vals[i-3] = ws.getitem(i)->getval();
					}
					glocal.stats.vals[0] = vals[0];
					if (n >= 9){
						glocal.stats.vals[1] = vals[2];
						glocal.stats.vals[2] = vals[3];
						glocal.stats.vals[3] = vals[5];
					}else{
						glocal.stats.vals[1] = vals[1];
						glocal.stats.vals[2] = vals[2];
						glocal.stats.vals[3] = vals[3];
					}
					glocal.COROUTINE.yield();
				}
			}
			return 0;
		</f>
		</call>
	</f>
	</obj>
	int nbdisk = 0;
	co.restart();
	while(co.next()) nbdisk++;
	DISKSTATS_UI tbstats[nbdisk];
	{
		co.restart();
		int no = 0;
		while(co.next()){
			tbstats[no].dev = glocal.stats.dev;
			memcpy (tbstats[no].vals,glocal.stats.vals,sizeof(glocal.stats.vals));
			no++;
		}
	}
	int maxdiff[NBTITLE] = {100,100,100,100};
	for (int i=0; i<NBTITLE; i++){
		static const char *tbsection[]={
			MSG_U(I_READS,"Reads"),
			MSG_U(I_READSECTS,"Read sectors"),
			MSG_U(I_WRITES,"Writes"),
			MSG_U(I_WRITESECTS,"Write sectors")
		};
		const char *section = tbsection[i];
		dia.newf_title (section,1,"",section);
		dia.newf_num (MSG_U(I_SCALE,"Scale"),maxdiff[i]);
		dia.set_lastreadonly();
		dia.newline();
		for (int j=0; j<nbdisk; j++){
			DISKSTATS_UI &v = tbstats[j];
			dia.newf_num (v.dev.get(),v.diffs[i]);
			dia.set_lastreadonly();
			dia.newf_gauge (NULL,v.gauges[i],100);
			dia.newline();
		}
	}
	int nof = 0;
	PRIVATE_MESSAGE msg;
	dialog_settimer (msg,1,true);
	dia.waitfor (msg);
	while (1){
		MENU_STATUS code = dia.edit (MSG_U(T_DISKSTATS,"Disk activity")
			,MSG_U(I_DISKSTATS
				,"Disk activity checked every second\n"
				 "The number of disk operations (reads/writes)\n"
				 "and the total disk sectors moved (read sectors/write sectors\n"
				 "are shown.\n"
				 "\n"
				 "The scale is computed.")
			,help_stats,nof,MENUBUT_CANCEL);
		if (code == MENU_CANCEL || code == MENU_ESCAPE){
			break;
		}else if (code == MENU_MESSAGE){
			co.restart();
			int no = 0;
			while(co.next()){
				DISKSTATS_UI &v = tbstats[no];
				for (int i=0; i<4; i++){
					v.diffs[i] = glocal.stats.vals[i] - v.vals[i];
					v.vals[i] = glocal.stats.vals[i];
				}
				no++;
			}
			// Auto-scale. Find the maximum
			for (int i=0; i<nbdisk; i++){
				DISKSTATS_UI &v = tbstats[i];
				for (int j=0; j<NBTITLE; j++){
					if (maxdiff[j] < v.diffs[j]) maxdiff[j] = v.diffs[j];
				}
			}
			for (int i=0; i<NBTITLE; i++){
				int maxd = maxdiff[i];
				if (maxd <= 100){
					maxd = 100;
				}else if (maxd <= 1000){
					maxd = 1000;
				}else if (maxd <= 10000){
					maxd = 10000;
				}else if (maxd <= 100000){
					maxd = 100000;
				}else if (maxd <= 1000000){
					maxd = 1000000;
				}
				maxdiff[i] = maxd;
			}
			// Compute the gauge value
			for (int i=0; i<nbdisk; i++){
				DISKSTATS_UI &v = tbstats[i];
				for (int j=0; j<NBTITLE; j++){
					v.gauges[j] = v.diffs[j] * 100 / maxdiff[j];
				}
			}
			dia.reload();
		}
	}
}
</mod>