#include <string.h>
#include <stdlib.h>
#include "wuftpd.h"
#include "wuftpd.m"
#include <private_msg.h>
#include <tlmplib.h>
#include <tlmpdia.h>

static HELP_FILE help_logs ("wuftpd","logs");
static HELP_FILE help_banner ("wuftpd","banner");

struct ftpstats{
	long download;
	long upload;
	int nbdownload;
	int nbupload;
	int anonymous;
};

/*
	Count the number of file transfer in the log file.
*/
<mod>
static int logs_count (
	const char *file,
	ftpstats &st)
{
	<glocal>
		ftpstats st;
	</glocal>
	glocal.st.download = 0;
	glocal.st.upload = 0;
	glocal.st.nbdownload = 0;
	glocal.st.nbupload = 0;
	glocal.st.anonymous = 0;
	<call loadfile>(file,false);
	<f missing>
	</f>
	<f oneline>
		char username[100];
		int size;
		char direction,access_mode;
		sscanf (line
			,"%*s %*s %*s %*s %*s"	// Skip the date
			 " %*s %*s"				// Transfert time, remote host
			 " %d %*s %*s %*s %c %c %s"
			,&size,&direction,&access_mode,username);
		if (access_mode == 'a') glocal.st.anonymous++;
		if (direction == 'o'){
			glocal.st.download += size;
			glocal.st.nbdownload++;
		}else{
			glocal.st.upload += size;
			glocal.st.nbupload++;
		}
		return 0;
	</f>
	</call>
	st = glocal.st;
	return glocal.st.nbdownload + glocal.st.nbupload;
}
</mod>

<mod>
static void logs_viewstats (const char *file, const char *host)
{
	<glocal>
		DIALOG_TEXTBOX dia;
	</glocal>
	glocal.dia.setcontext ("");
	SSTRING cmd;
	cmd.setfromf ("-r -g -a -f%s",file);
	{
		SSTRING tmp;
		tmp.setfromf (MSG_U(I_OUTPUTCMD,"Output of command xferstats %s")
			,cmd.get());
		glocal.dia.newf_text ("",tmp.get());
		glocal.dia.newf_text ("","");
	}
	<call walkpopen>("xferstats",cmd.get(),40);
	<f oneline>
		glocal.dia.newf_text ("",line);
		return 0;
	</f>
	<f oneerr>
		glocal.dia.newf_text ("",line);
		return 0;
	</f>
	</call>
	SSTRING title;
	title.setfromf (MSG_U(T_HOSTFTPLOGS,"FTP stats for %s"),host);
	int nof = 0;
	glocal.dia.edit (title.get(),"",help_logs,nof,MENUBUT_QUIT);
}
</mod>

<mod>
static void logs_view (const char *file, const char *host)
{
	<glocal>
		const char *file;
		const char *host;
		PRIVATE_MESSAGE timer;
		bool first;
	</glocal>
	glocal.first = true;
	glocal.host = host;
	glocal.file = file;
	dialog_settimer (glocal.timer,5	,true);
	<call editrecords>(MSG_U(T_LOGDETAILS,"Log details"),"",help_logs);
	<f head>
		newf_head (MSG_U(H_LOGDETAILS
			,"File name\tSize\tDirection\tAccess mode\tUser\tDate"));
		setbutinfo (MENU_USR1,MSG_U(B_VIEWSTATS,"View stats")
			,MSG_R(B_VIEWSTATS));
		sethdispo ("lrllll");
		sortable();
		sortpolicy ("anaaaa");
		waitfor (glocal.timer);
	</f>
	<f load>
		int nbline = <call loadfile>(glocal.file,false);
		<f missing>
		</f>
		<f oneline>
			char fname[PATH_MAX],username[100];
			char month[4],day[3],hour[9],year[5];
			int size;
			char direction,access_mode;
			sscanf (line
				,"%*s %3s %2s %8s %4s"	// date
				 " %*s %*s"				// Transfert time, remote host
				 " %d %s %*s %*s %c %c %s"
				,month,day,hour,year
				,&size,fname,&direction,&access_mode,username);
			static const char *tbmonth[]={
				"Jan","Feb","Mar","Apr","May","Jun","Jul","Aug",
				"Sep","Oct","Nov","Dec"
			};
			int nomonth = 0;
			for (int i=0; i<12; i++){
				if (strcasecmp(tbmonth[i],month)==0){
					nomonth = i;
					break;
				}
			}
			char date[10+1+8+1];
			sprintf (date,"%04d/%02d/%02d %8.8s"
				,atoi(year),nomonth+1,atoi(day),hour);
			glocal.editrecords.new_menuitemf (fname,"%d\t%s\t%s\t%s\t%s"
				,size
				,direction=='o'
					? MSG_U(I_DOWNLOAD,"Download")
					: MSG_U(I_UPLOAD,"Upload")
				,access_mode=='a'
					? MSG_U(I_ANONYMOUS,"Anonymous")
					: access_mode == 'g'
						? MSG_U(I_GUEST,"Guest")
						: MSG_U(I_USER,"User")
				,username
				,date);
			return 0;
		</f>
		</call>
		if (glocal.first && nbline > 0){
			setcursor (nbline-1);
			glocal.first = false;
		}
	</f>
	<f editone>
	</f>
	<f otherbuttons>
		logs_viewstats (glocal.file,glocal.host);
	</f>
	</call>
	dialog_deltimer (glocal.timer);
}
</mod>



<mod>
PUBLIC void WUFTPD::viewlogs()
{
	<glocal>
		VIRTUAL_FTPS sorted;
		PRIVATE_MESSAGE timer;
	</glocal>
	glocal.sorted.neverdelete();
	for (int i=0; i<virtuals.getnb(); i++){
		glocal.sorted.add (virtuals.getitem(i));
	}
	<call sort>(glocal.sorted);
	<f cmp>
		VIRTUAL_FTP *f1 = (VIRTUAL_FTP*)o1;
		VIRTUAL_FTP *f2 = (VIRTUAL_FTP*)o2;
		return f1->addr.cmp(f2->addr);
	</f>
	</call>
	dialog_settimer (glocal.timer,3,true);	
	<call editrecords>(MSG_U(T_FTPLOGS,"FTP transfer logs")
		,MSG_U(I_FTPLOGS,"Select one line to get a complete report")
		,help_logs);
	<f head>
		newf_head (MSG_U(H_FTPLOGS,"Host\tTransfers\tDownloads\tSize\tUploads\tSize\tAnonymous"));
		waitfor (glocal.timer);	// Just to wake up and refresh the list
		sethdispo ("lrrrrrr");
		sortable();
		sortpolicy ("annnnnn");
	</f>
	<f load>
		ftpstats st;
		int nb = logs_count (configf_lookuppath("/var/log/xferlog"),st);
		new_menuitemf (MSG_U(I_MAINSERVER,"Main server"),"%d\t%d\t%ld\t%d\t%ld\t%d"
			,nb,st.nbdownload,st.download,st.nbupload,st.upload
			,st.anonymous);
		for (int i=0; i<glocal.sorted.getnb(); i++){
			VIRTUAL_FTP *ftp = glocal.sorted.getitem(i);
			if (ftp->logfile.is_empty()){
				new_menuitemf (ftp->addr.get(),"-\t-\t-\t-\t-\t-");
			}else{
				nb = logs_count (ftp->logfile.get(),st);
				new_menuitemf (ftp->addr.get(),"%d\t%d\t%ld\t%d\t%ld\t%d"
					,nb,st.nbdownload,st.download,st.nbupload,st.upload
					,st.anonymous);
			}
		}
	</f>
	<f editone>
		if (no == 0){
			logs_view (configf_lookuppath("/var/log/xferlog"),MSG_R(I_MAINSERVER));
		}else{
			VIRTUAL_FTP *ftp = glocal.sorted.getitem(no-1);
			if (ftp->logfile.is_empty()){
				xconf_notice (MSG_U(N_NOLOGFORHOST
					,"There is no log file for this virtual host.\n"
					 "Consult the main server log"));
			}else{
				logs_view (ftp->logfile.get(),ftp->addr.get());
			}
		}
	</f>
	</call>
	dialog_deltimer (glocal.timer);	
}
</mod>

<mod>
void wuftpd_editbanner (const char *fname)
{
	<glocal>
		SSTRING text;
	</glocal>
	DIALOG dia;
	<call loadfile>(fname,false);
	<f missing>
	</f>
	<f oneline>
		glocal.text.append (line);
		return 0;
	</f>
	</call>
	dia.newf_textarea ("",glocal.text,70,15);
	int nof = 0;
	SSTRING title;
	title.setfromf (MSG_U(T_BANNERFILE,"Banner file %s"),fname);
	while (1){
		MENU_STATUS code = dia.edit (title.get(),"",help_banner,nof);
		if (code == MENU_CANCEL || code == MENU_ESCAPE){
			break;
		}else{
			int ok = <call savefile>(fname,false);
			<f dowrite>
				fputs (glocal.text.get(),fout);
				return 0;
			</f>
			</call>
			if (ok != -1) break;
		}
	}
}
</mod>