#include <stdio.h>

#include <stdio.h>
#include <unistd.h>
#include <getopt.h>
#include <stdbool.h>
#include <string.h>
#include <libxml/tree.h>
#include <libxml/debugXML.h>
#include <libxslt/xslt.h>
#include <libxslt/transform.h>
#include <libxslt/xsltutils.h>
#include "s1kd_tools.h"
#include "dmrl.h"

#define PROG_NAME "s1kd-dmrl"
#define VERSION "1.12.0"

static void showHelp(void)
{
puts("Usage: " PROG_NAME " [options] <DML>...");
puts("");
puts("Options:");
puts(" -$, --issue <iss> Which issue of the spec to use.");
puts(" -@, --out <dir> Output to specified directory.");
puts(" -%, --templates <dir> Custom XML template directory.");
puts(" -D, --dmtypes <path> Specify .dmtypes file name.");
puts(" -d, --defaults <path> Specify .defaults file name.");
#ifndef _WIN32
puts(" -F, --fail Fail on first error from s1kd-new* commands.");
#endif
puts(" -f, --overwrite Overwrite existing CSDB objects.");
puts(" -h, -?, --help Show usage message.");
puts(" -m, --use-remarks Use the remarks for entries in the objects.");
puts(" -N, --omit-issue Omit issue/inwork numbers.");
puts(" -q, --quiet Don't report errors if objects exist.");
puts(" -s, --commands Output s1kd-new* commands only.");
puts(" -v, --verbose Print the names of newly created objects.");
puts(" --version Show version information.");
LIBXML2_PARSE_LONGOPT_HELP
}

static void show_version(void)
{
printf("%s (s1kd-tools) %s\n", PROG_NAME, VERSION);
printf("Using libxml %s and libxslt %s\n", xmlParserVersion, xsltEngineVersion);
}

int main(int argc, char **argv)
{
int i;
xmlDocPtr dmrl;
xsltStylesheetPtr dmrlStylesheet;
int err = 0;
bool execute = true;
bool noIssue = false;
#ifndef _WIN32
bool failOnFirstErr = false;
#endif
bool overwrite = false;
bool noOverwriteError = false;
bool verbose = false;
char *specIssue = NULL;
char *templateDir = NULL;
char *outDir = NULL;
char *defaultsFname = NULL;
char *dmtypesFname = NULL;
bool use_remarks = false;

const char *sopts = "D:d:smNfFq$:%:@:vh?";
struct option lopts[] = {
{"version" , no_argument , 0, 0},
{"help" , no_argument , 0, 'h'},
{"defaults" , required_argument, 0, 'd'},
{"dmtypes" , required_argument, 0, 'D'},
{"commands" , no_argument , 0, 's'},
{"use-remarks", no_argument , 0, 'm'},
{"omit-issue" , no_argument , 0, 'N'},
{"overwrite" , no_argument , 0, 'f'},
{"fail" , no_argument , 0, 'F'},
{"quiet" , no_argument , 0, 'q'},
{"issue" , required_argument, 0, '$'},
{"verbose" , no_argument , 0, 'v'},
{"templates" , required_argument, 0, '%'},
{"out" , required_argument, 0, '@'},
LIBXML2_PARSE_LONGOPT_DEFS
{0, 0, 0, 0}
};
int loptind = 0;

dmrl = read_xml_mem((const char *) dmrl_xsl, dmrl_xsl_len);
dmrlStylesheet = xsltParseStylesheetDoc(dmrl);

while ((i = getopt_long(argc, argv, sopts, lopts, &loptind)) != -1) {
switch (i) {
case 0:
if (strcmp(lopts[loptind].name, "version") == 0) {
show_version();
return 0;
}
LIBXML2_PARSE_LONGOPT_HANDLE(lopts, loptind, optarg)
break;
case 'd':
defaultsFname = strdup(optarg);
break;
case 'D':
dmtypesFname = strdup(optarg);
break;
case 's':
execute = false;
break;
case 'm':
use_remarks = true;
break;
case 'N':
noIssue = true;
break;
case 'f':
overwrite = true;
break;
#ifndef _WIN32
case 'F':
failOnFirstErr = true;
break;
#endif
case 'q':
noOverwriteError = true;
break;
case '$':
specIssue = strdup(optarg);
break;
case 'v':
verbose = true;
break;
case '%':
templateDir = strdup(optarg);
break;
case '@':
outDir = strdup(optarg);
break;
case 'h':
case '?':
showHelp();
return 0;
}
}

for (i = optind; i < argc; ++i) {
xmlDocPtr in, out;
xmlChar *content;
const char *params[21];
char iss[8];
char *templs = NULL;
char *outd = NULL;
char *defname = NULL;
char *dmtname = NULL;

if (!(in = read_xml_doc(argv[i]))) {
continue;
}

params[0] = "no-issue";
params[1] = noIssue ? "true()" : "false()";

params[2] = "overwrite";
params[3] = overwrite ? "true()" : "false()";

params[4] = "no-overwrite-error";
params[5] = noOverwriteError ? "true()" : "false()";

params[6] = "spec-issue";
if (specIssue) {
snprintf(iss, 8, "\"%s\"", specIssue);
params[7] = iss;
} else {
params[7] = "false()";
}

params[8] = "verbose";
params[9] = verbose ? "true()" : "false()";

params[10] = "templates";
if (templateDir) {
templs = malloc(strlen(templateDir) + 3);
sprintf(templs, "\"%s\"", templateDir);
params[11] = templs;
} else {
params[11] = "false()";
}

params[12] = "outdir";
if (outDir) {
outd = malloc(strlen(outDir) + 3);
sprintf(outd, "\"%s\"", outDir);
params[13] = outd;
} else {
params[13] = "false()";
}

params[14] = "defaults";
if (defaultsFname) {
defname = malloc(strlen(defaultsFname) + 3);
sprintf(defname, "\"%s\"", defaultsFname);
params[15] = defname;
} else {
params[15] = "false()";
}

params[16] = "dmtypes";
if (dmtypesFname) {
dmtname = malloc(strlen(dmtypesFname) + 3);
sprintf(dmtname, "\"%s\"", dmtypesFname);
params[17] = dmtname;
} else {
params[17] = "false()";
}

params[18] = "use-remarks";
params[19] = use_remarks ? "true()" : "false()";

params[20] = NULL;

out = xsltApplyStylesheet(dmrlStylesheet, in, params);

free(templs);
free(outd);
free(defname);
free(dmtname);

xmlFreeDoc(in);

content = xmlNodeGetContent(out->children);

if (execute) {
#ifdef _WIN32
/* FIXME: Implement alternative to fmemopen and
* WEXITSTATUS in order to use the -F ("fail on first
* error") option on a Windows system.
*/
char *line = NULL;
while ((line = strtok(line ? NULL : (char *) content, "\n"))) {
system(line);
}
#else
FILE *lines;
char line[LINE_MAX];

lines = fmemopen(content, xmlStrlen(content), "r");

while (fgets(line, LINE_MAX, lines))
if ((err += WEXITSTATUS(system(line))) != 0 && failOnFirstErr) break;

fclose(lines);
#endif
} else {
fputs((char *) content, stdout);
}

xmlFree(content);

xmlFreeDoc(out);
}

xsltFreeStylesheet(dmrlStylesheet);

free(specIssue);
free(templateDir);
free(outDir);
free(defaultsFname);
free(dmtypesFname);

xsltCleanupGlobals();
xmlCleanupParser();

return err;
}