/* gamatronic.c
 *
 * SEC UPS Driver ported to the new NUT API for Gamatronic UPS Usage.
 *
 * TODO: Replace lots of printf() by upslogx() or upsdebugx() below!
 *
 * Copyright (C)
 *   2001 John Marley <John.Marley@alcatel.com.au>
 *   2002 Jules Taplin <jules@netsitepro.co.uk>
 *   2002 Eric Lawson <elawson@inficad.com>
 *   2005 Arnaud Quette <arnaud.quette@gmail.com>
 *   2005 Nadav Moskovitch <blutz@walla.com / http://www.gamatronic.com>
 *
 * 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 2 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, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 *
 */

#include "main.h"
#include "serial.h"
#include "gamatronic.h"
#include "nut_stdint.h"

#define DRIVER_NAME	"Gamatronic UPS driver"
#define DRIVER_VERSION	"0.02"

/* driver description structure */
upsdrv_info_t upsdrv_info = {
	DRIVER_NAME,
	DRIVER_VERSION,
	"John Marley <John.Marley@alcatel.com.au>\n" \
	"Jules Taplin <jules@netsitepro.co.uk>\n" \
	"Eric Lawson <elawson@inficad.com>\n" \
	"Arnaud Quette <arnaud.quette@gmail.com>\n" \
	"Nadav Moskovitch <blutz@walla.com / http://www.gamatronic.com>",
	DRV_STABLE,
	{ NULL }
};

#define ENDCHAR	'\r'
#define IGNCHARS	""
#define SER_WAIT_SEC	1	/* allow 3.0 sec for ser_get calls */
#define SER_WAIT_USEC	0

static int sec_upsrecv (char *buf)
{
	char lenbuf[4];
	int ret;

	ser_get_line(upsfd, buf, 140, ENDCHAR, IGNCHARS,SER_WAIT_SEC, SER_WAIT_USEC);
	if (buf[0] ==  SEC_MSG_STARTCHAR) {
		switch (buf[1]) {
			case SEC_NAK:
				return(-1);
			case SEC_ACK:
				return(0);
			case SEC_DATAMSG:
				strncpy(lenbuf, buf+2, 3);
				lenbuf[3] = '\0';
				ret = atoi(lenbuf);
				if (ret > 0) {
					strcpy(buf,buf+5);
					return(ret);
				}
				else return (-2);
			default:
				return(-2);
		}
	}
	else
		return (-2);
}

static ssize_t sec_cmd(const char mode, const char *command, char *msgbuf, ssize_t *buflen)
{
	char msg[140];
	ssize_t ret;

	memset(msg, 0, sizeof(msg));

	/* create the message string */
	if (*buflen > 0) {
		snprintf(msg, sizeof(msg), "%c%c%03zd%s%s", SEC_MSG_STARTCHAR,
			mode, (*buflen)+3, command, msgbuf);
	}
	else {
		snprintf(msg, sizeof(msg), "%c%c003%s", SEC_MSG_STARTCHAR,
			mode, command);
	}
	upsdebugx(1, "PC-->UPS: \"%s\"",msg);
	ret = ser_send(upsfd, "%s", msg);

	upsdebugx(1, " send returned: %zd",ret);

	if (ret == -1) return -1;

	ret = sec_upsrecv(msg);

	if (ret < 0) return -1;

	strncpy(msgbuf, msg, (size_t)ret);
	upsdebugx(1, "UPS<--PC: \"%s\"",msg);

/*
	*(msgbuf+ret) = '\0';
*/

	*buflen = ret;
	return ret;
}

static void addquery(const char *cmd, int field, int varnum, int pollflag)
{
	int q;

	for (q=0; q<SEC_QUERYLIST_LEN; q++) {
		if (sec_querylist[q].command == NULL) {
			/* command has not been recorded yet */
			sec_querylist[q].command = cmd;
			sec_querylist[q].pollflag = pollflag;
			upsdebugx(1, " Query %d is %s",q,cmd);
		}
		if (sec_querylist[q].command == cmd) {
			sec_querylist[q].varnum[field-1] = varnum;
			upsdebugx(1, " Querying varnum %d",varnum);
			break;
		}
	}
}

static void sec_setinfo(int varnum, char *value)
{
	if (*sec_varlist[varnum].setcmd)
	{ /*Not empty*/
		if (sec_varlist[varnum].flags == FLAG_STRING) {
			dstate_setinfo(sec_varlist[varnum].setcmd,"%s", value);
	 	}
		else if (sec_varlist[varnum].unit == 1) {
			dstate_setinfo(sec_varlist[varnum].setcmd,"%s", value);
		}
		else if (sec_varlist[varnum].flags == FLAG_MULTI) {
			if (atoi(value) < 0) {
				dstate_setinfo(sec_varlist[varnum].setcmd,"0");
			}
			else {
				dstate_setinfo(sec_varlist[varnum].setcmd,"%d", atoi(value) * sec_varlist[varnum].unit);
			}
		}
		else {
			dstate_setinfo(sec_varlist[varnum].setcmd,"%.1f", atof(value) / sec_varlist[varnum].unit);
		}
	}
}

static void update_pseudovars( void )
{
	status_init();

	if(strncmp(sec_varlist[9].value, "1", 1)== 0) {
		status_set("OFF");
	}
	if(strncmp(sec_varlist[76].value, "0", 1)== 0) {
		status_set("OL");
	}
	if(strncmp(sec_varlist[76].value, "1", 1)== 0) {
		status_set("OB");
	}
	if(strncmp(sec_varlist[76].value, "2", 1)== 0) {
		status_set("BYPASS");
	}
	if(strncmp(sec_varlist[76].value, "3", 1)== 0) {
		status_set("TRIM");
	}
	if(strncmp(sec_varlist[76].value, "4", 1)== 0) {
		status_set("BOOST");
	}
	if(strncmp(sec_varlist[10].value, "1", 1)== 0) {
		status_set("OVER");
	}
	if(strncmp(sec_varlist[22].value, "1", 1)== 0) {
		status_set("LB");
	}
	if(strncmp(sec_varlist[19].value, "2", 1)== 0) {
		status_set("RB");
	}

	status_commit();
}

static void sec_poll ( int pollflag ) {
	ssize_t msglen;
	int f, q;
	char retbuf[140], *n, *r;

	for (q=0; q<SEC_QUERYLIST_LEN; q++) {
		if (sec_querylist[q].command == NULL) break;
		if (sec_querylist[q].pollflag != pollflag) continue;

		msglen = 0;
		sec_cmd(SEC_POLLCMD, sec_querylist[q].command, retbuf, &msglen);
		r = retbuf;
		*(r+msglen) = '\0';
		for (f=0; f<SEC_MAXFIELDS; f++) {
			n = strchr(r, ',');
			if (n != NULL) *n = '\0';

			if (sqv(q,f) > 0) {
				if (strcmp(sec_varlist[sqv(q,f)].value, r) != 0) {
#ifdef HAVE_PRAGMAS_FOR_GCC_DIAGNOSTIC_IGNORED_FORMAT_TRUNCATION
#pragma GCC diagnostic push
#endif
#ifdef HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_FORMAT_TRUNCATION
#pragma GCC diagnostic ignored "-Wformat-truncation"
#endif
					/* NOTE: We intentionally limit the amount
					 * of characters picked from "r" buffer
					 * into respectively sized "*.value"
					 */
					int len = snprintf(sec_varlist[sqv(q,f)].value,
						sizeof(sec_varlist[sqv(q,f)].value), "%s", r);

					if (len < 0) {
						upsdebugx(1, "%s: got an error while extracting value", __func__);
					}

					if ((intmax_t)len > (intmax_t)sizeof(sec_varlist[sqv(q,f)].value)
					||  (intmax_t)strnlen(r, sizeof(retbuf)) > (intmax_t)sizeof(sec_varlist[sqv(q,f)].value)
					) {
						upsdebugx(1, "%s: value was truncated", __func__);
					}
#ifdef HAVE_PRAGMAS_FOR_GCC_DIAGNOSTIC_IGNORED_FORMAT_TRUNCATION
#pragma GCC diagnostic pop
#endif
					sec_setinfo(sqv(q,f), r);
				}

				/* If SEC VAR is alarm and it's on, add it to the alarm property */
				if (sec_varlist[sqv(q,f)].flags & FLAG_ALARM && strncmp(r, "1", 1)== 0) {
					alarm_set(sec_varlist[sqv(q,f)].name);
				}
			}

			if (n == NULL) break;
			r = n+1;
		}
	}

}

void upsdrv_initinfo(void)
{
	ssize_t msglen;
	int v;
	char *a, *p, avail_list[300];

	/* find out which variables/commands this UPS supports */
	msglen = 0;
	sec_cmd(SEC_POLLCMD, SEC_AVAILP1, avail_list, &msglen);
	p = avail_list + msglen;
	if (p != avail_list) *p++ = ',';
	msglen = 0;
	sec_cmd(SEC_POLLCMD, SEC_AVAILP2, p, &msglen);
	*(p+msglen) = '\0';

	if (strlen(avail_list) == 0) {
		fatalx(EXIT_FAILURE, "No available variables found!");
	}
	a = avail_list;
	while ((p = strtok(a, ",")) != NULL) {
		a = NULL;
		v = atoi(p);
		/* don't bother adding a write-only variable */
		if (sec_varlist[v].flags == FLAG_WONLY) continue;
		addquery(sec_varlist[v].cmd, sec_varlist[v].field, v, sec_varlist[v].poll);
	}

	/* poll one time values */
	sec_poll(FLAG_POLLONCE);

	printf("UPS: %s %s\n", dstate_getinfo("ups.mfr"), dstate_getinfo("ups.model"));
}

void upsdrv_updateinfo(void)
{
	alarm_init();
	/* poll status values values */
	sec_poll(FLAG_POLL);
	alarm_commit();
	update_pseudovars();
	dstate_dataok();
}

void upsdrv_shutdown(void)
{
	ssize_t msglen;
	char msgbuf[SMALLBUF];

	msglen = snprintf(msgbuf, sizeof(msgbuf), "-1");
	sec_cmd(SEC_SETCMD, SEC_SHUTDOWN, msgbuf, &msglen);

	msglen = snprintf(msgbuf, sizeof(msgbuf), "1");
	sec_cmd(SEC_SETCMD, SEC_AUTORESTART, msgbuf, &msglen);

	msglen = snprintf(msgbuf, sizeof(msgbuf), "2");
	sec_cmd(SEC_SETCMD, SEC_SHUTTYPE,msgbuf, &msglen);

	msglen = snprintf(msgbuf, sizeof(msgbuf), "5");
	sec_cmd(SEC_SETCMD, SEC_SHUTDOWN, msgbuf, &msglen);
}

/*
int instcmd(const char *cmdname, const char *extra)
{
	if (!strcasecmp(cmdname, "test.battery.stop")) {
		ser_send_buf(upsfd, ...);
		return STAT_INSTCMD_HANDLED;
	}

	upslogx(LOG_NOTICE, "instcmd: unknown command [%s]", cmdname);
	return STAT_INSTCMD_UNKNOWN;
}
*/

void upsdrv_help(void)
{
}

/* list flags and values that you want to receive via -x */
void upsdrv_makevartable(void)
{
	/* allow '-x xyzzy' */
	/* addvar(VAR_FLAG, "xyzzy", "Enable xyzzy mode"); */

	/* allow '-x foo=<some value>' */
	/* addvar(VAR_VALUE, "foo", "Override foo setting"); */
}

static void setup_serial(const char *port)
{
	char temp[140];
	int i;
	ssize_t ret;

	/* Detect the ups baudrate  */
	for (i=0; i<5; i++) {
		ser_set_speed(upsfd, device_path, baud_rates[i].rate);
		ret = ser_send(upsfd, "^P003MAN");
		ret = sec_upsrecv(temp);
		if (ret >= -1) break;
	}

	if (i == 5) {
		printf("Can't talk to UPS on port %s!\n",port);
		printf("Check the cabling and portname and try again\n");
		printf("Please note that this driver only support UPS Models with SEC Protocol\n");
		ser_close(upsfd, device_path);
		exit (1);
	}
	else
		printf("Connected to UPS on %s baudrate: %zu\n",
			port, baud_rates[i].name);
}

void upsdrv_initups(void)
{
	upsfd = ser_open(device_path);
	setup_serial(device_path);
	/* upsfd = ser_open(device_path); */
	/* ser_set_speed(upsfd, device_path, B1200); */

	/* probe ups type */

	/* to get variables and flags from the command line, use this:
	 *
	 * first populate with upsdrv_buildvartable above, then...
	 *
	 *                   set flag foo : /bin/driver -x foo
	 * set variable 'cable' to '1234' : /bin/driver -x cable=1234
	 *
	 * to test flag foo in your code:
	 *
	 * 	if (testvar("foo"))
	 * 		do_something();
	 *
	 * to show the value of cable:
	 *
	 *      if ((cable == getval("cable")))
	 *		printf("cable is set to %s\n", cable);
	 *	else
	 *		printf("cable is not set!\n");
	 *
	 * don't use NULL pointers - test the return result first!
	 */

	/* the upsh handlers can't be done here, as they get initialized
	 * shortly after upsdrv_initups returns to main.
	 */
}

void upsdrv_cleanup(void)
{
	/* free(dynamic_mem); */
	ser_close(upsfd, device_path);
}