#include <stdio.h>
#include <pwd.h>
#include <string.h>
#include <stdlib.h>
#include <ctype.h>
#include "printer.h"
#include "printer.m"
#include <tlmpdia.h>
#include <tlmplib.h>
#include <userconf.h>
#include <translat.h>

static HELP_FILE help_printjobs ("printer","printjobs");

#define _TLMP_walkprintjobs

struct _F_walkprintjobs {
	#define _F_walkprintjobs_onejob(name) void name onejob(const char *job, int uid, int size, const char *date)
	virtual _F_walkprintjobs_onejob( )=0;
};

<mod>
static int walkprintjobs(_F_walkprintjobs &c, const char *printer)
{
	<glocal>
		bool header;
		_F_walkprintjobs *c;
		int ret;
	</glocal>
	char cmd[strlen(printer)+10];
	snprintf (cmd,sizeof(cmd)-1,"lpq -P%s",printer);
	glocal.header = true;
	glocal.c = &c;
	glocal.ret = 0;
	<call walkpopen>(cmd,20);
	<f oneline>
		if (glocal.header){
			const char *pt = str_skip(line);
			if (strncmp(pt,"Rank",4)==0){
				glocal.header = false;
			}
		}else{
			glocal.ret++;
			const char *pt = str_skipword(line);
			pt = str_skip(pt);
			char owner[200],job[200];
			pt = str_copyword (owner,pt);
			pt = str_skip(pt);
			pt = str_skipword(pt);	// class
			pt = str_copyword (job,pt);
			pt = str_skip(pt);
			int uid = 0;
			struct passwd *p = getpwnam(owner);
			if (p != NULL) uid = p->pw_uid;
			pt = str_skipword (pt);	// Skip the file name
									// Does not work if there are spaces
			pt = str_skip(pt);
			int size = atoi(pt);
			const char *endpt = strstr (line,"bytes");
			if (endpt != NULL){
				// Trick with old lpr system
				pt += 5;
				while (isspace(endpt[-1])) endpt--;
				while (isdigit(endpt[-1])) endpt--;
				size = atoi(endpt);
			}
			pt = str_skip(pt);
			glocal.c->onejob (job,uid,size,pt);
		}
		return 0;
	</f>
	</call>
	return glocal.ret;
}
</mod>

/*
	Get the list of queue defined in /etc/printcap
*/
int queue_getlist(SSTRINGS &tb)
{
	FILE_CFG *fin = cf_printcapfile.fopen ("/etc/printcap","r");
	if (fin != NULL){
		char buf[2000];
		while (fgets_strip(buf,sizeof(buf)-1,fin,'\\','#',NULL)!=NULL){
			// First, we split the queue name from the end of the line
			char *pt = strchr(buf,':');
			if (pt != NULL) *pt = '\0';
			// Then we only keep the primary queue name
			pt = strchr(buf,'|');
			if (pt != NULL) *pt = '\0';
			pt = str_skip(buf);
			if (*pt != '\0'){
				tb.add (new SSTRING(pt));
			}
		}
		fclose (fin);
	}
	return tb.getnb();
}

static PRIVILEGES tb;

static void queue_setprivi()
{
	if (tb.getnb()==0){
		SSTRINGS tbq;
		int n = queue_getlist (tbq);
		for (int i=0; i<n; i++){
			const char *q = tbq.getitem(i)->get();
			char id[PATH_MAX];
			sprintf (id,"lpqueue_%s",q);
			tb.add (new PRIVILEGE (id,q
				,MSG_U(T_QUEUEMNG,"Printer queue management")));
		}
	}
}

/*
	Return the co-admin privilege associated with a print queue
*/
PRIVILEGE *queue_lookuppriv (const char *p)
{
	queue_setprivi();
	char id[PATH_MAX];
	sprintf (id,"lpqueue_%s",p);
	return privilege_lookup (id);
}

<mod>
void queue_getstat (
	const char *p,		// Printer or queue name
	bool &accept,
	bool &print,
	int &nbjob)			// Number of jobs in the spool
{
	<glocal>
	bool *accept;
	bool *print;
	</glocal>
	print = accept = false;
	nbjob = 0;
	glocal.accept = &accept;
	glocal.print = &print;
	SSTRING arg;
	arg.setfromf ("status %s",p);
	<call walkpopen>("lpc",arg.get(),20);
	<f oneline>
		if (strstr(line,"printing is enabled")!=NULL) glocal.print[0] = true;
		if (strstr(line,"queuing is enabled")!=NULL) glocal.accept[0] = true;
		return 0;
	</f>
	</call>
	nbjob = <call walkprintjobs>(p);
	<f onejob>
	</f>
	</call>
}
</mod>

<mod>
/*
	Manage the jobs for a printer queue
*/
void queue_jobs(const char *p)
{
	<glocal>
		const char *printer;
		bool header;
		SSTRINGS jobs;
		_F_editrecords *c;
	</glocal>
	glocal.printer = p;
	<call editrecords>(MSG_U(T_PRINTJOBS,"Print jobs")
		,""
		,help_printjobs);
	<f head>
		newf_head (MSG_U(H_PRINTJOBS,"Job ID\tUser\tFull name\tSize\tDate"));
		setbutinfo (MENU_USR1,MSG_U(B_ERASEALL,"Erase all")
			,MSG_R(B_ERASEALL));
		sortable();
		sortpolicy ("naan");
		sethdispo ("rllr");
	</f>
	<f load>
		glocal.jobs.remove_all();
		glocal.c = this;
		<call walkprintjobs>(glocal.printer);
		<f onejob>
			struct passwd *pw = getpwuid (uid);
			char line[200];
			if (pw !=NULL){
				snprintf (line,sizeof(line)-1,"%s\t%s\t%d\t%s"
					,pw->pw_name
					,pw->pw_gecos
					,size
					,date);
			}else{
				snprintf (line,sizeof(line)-1,"%d\t\t%d\t%s"
					,uid
					,size
					,date);
			}
			glocal.c->new_menuitem (job,line);
			glocal.jobs.add (new SSTRING(job));
		</f>
		</call>
	</f>
	<f editone>
		const char *job = glocal.jobs.getitem(no)->get();
		SSTRING msg;
		msg.setfromf(MSG_U(Q_REMOVEJOB
			,"Do you want to remove the print job %s\n"
			 "on printer queue %s")
			,job,glocal.printer);
		if (xconf_yesno(MSG_U(T_CONFIRM,"Confirm"),msg.get(),help_nil)==MENU_YES){
			SSTRING arg;
			arg.setfromf ("-P%s %s",glocal.printer,job);
			netconf_system_if ("lprm",arg.get());
		}
	</f>
	<f otherbuttons>
		if (glocal.jobs.getnb()==0){
			xconf_error (MSG_U(E_QUEUEEMPTY,"Queue is already empty"));
		}else if (xconf_yesno(MSG_R(T_CONFIRM)
			,MSG_U(Q_ERASEALL,"Are you sure you want to erase\n"
				"all print jobs")
			,help_nil)==MENU_YES){
			SSTRING arg;
			arg.setfromf ("-P%s",glocal.printer);
			for (int i=0; i<glocal.jobs.getnb(); i++){
				arg.appendf (" %s",glocal.jobs.getitem(i)->get());
			}
			netconf_system_if ("lprm",arg.get());
		}
	</f>
	</call>
}
</mod>


static HELP_FILE help_privi ("printconf","queueprivi");
static REGISTER_PRIVI_HELP p (help_privi
	,P_MSG_U(T_QUEUEPRIVI,"Privileges: Printer queue management"));


static PRIVILEGE_DECLARATOR vdom_decl(queue_setprivi);