/*
 * AOPPA -- A program for extracting the AnarchyOnline resource database
 * Auno.org PostgreSQL nano table generator.
 *
 * Copyright (C) 2003  Oskari Saarenmaa <auno@auno.org>
 *
 * $Id: aunoorg-nanos.cc,v 1.4 2003/10/21 09:12:37 pickett Exp $
 *
 * 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 <stdarg.h>
#include <stdio.h>
#include <libpq-fe.h>

#include <list>
#include <map>

using namespace std;


static PGresult *
execf(PGconn *, const char *, ...);

int
main(int argc, char **argv)
{
  PGconn *dbconn;
  char buf[0x200];
  const char *dbname, *dbuser;
  
  if(argc != 3)
  {
    fprintf(stderr, "Usage: %s <dbname> <dbuser>\n", argv[0]);
    exit(1);
  }
  
  dbuser = (const char *)(argv[2]);
  dbname = (const char *)(argv[1]);

  snprintf(buf, sizeof buf, "dbname=%s user=%s", dbname, dbuser);
  dbconn = PQconnectdb(buf);
  if(dbconn == NULL)
  {
    fprintf(stderr, "Database structure allocation failed, out of memory?\n");
    exit(1);
  }
  else if(PQstatus(dbconn) != CONNECTION_OK)
  {
    fprintf(stderr, "Database login failed.\n");
    exit(1);
  }
  
  PGresult *q = PQexec(dbconn, "select aodb.xid,aodb.aoid,aodb_ext.icon,aodb_nano.school,aodb_strain.strain,aodb.ql,aodb_nano.ncu,aodb_ext.name,aodb_ext.info from aodb, aodb_nano, aodb_ext, aodb_strain where aodb.aoid = aodb_strain.aoid and aodb.xid = aodb_nano.xid and aodb.xid = aodb_ext.xid and aodb.current is true and aodb.metatype = 'n'");
  int i, rows = PQntuples(q);
  if(rows <= 0)
  {
    printf("No nanos found!\n");
  }
  else
  {
    PQclear(PQexec(dbconn, "drop table nanos"));
    PQclear(PQexec(dbconn, "create table nanos (aoid int4, icon int4, school int2, strain int2, es int2, ql int2, ncu int2, stack int2, prof int2, lvl int2, mm int2, bm int2, pm int2, mc int2, ts int2, si int2, name text, info text)"));

    for(i=0; i<rows; i++)
    {
      int xid    = atoi(PQgetvalue(q, i, 0)),
          aoid   = atoi(PQgetvalue(q, i, 1)),
          icon   = atoi(PQgetvalue(q, i, 2)),
          school = atoi(PQgetvalue(q, i, 3)),
          strain = atoi(PQgetvalue(q, i, 4)),
          ql     = atoi(PQgetvalue(q, i, 5)),
          ncu    = atoi(PQgetvalue(q, i, 6)),
          prof=0, lvl=0, mm=0, bm=0, pm=0, mc=0, ts=0, si=0, skip=0, es=0, stack=0;
      const char *name = PQgetvalue(q, i, 7),
                 *info = PQgetvalue(q, i, 8);
      char xname[0x4000], xinfo[0x4000];
      PGresult *xq = execf(dbconn, "select attribute,value,type from aodb_req where xid = %d and hook = 3", xid);
      int xi, xrows = PQntuples(xq);
      for(xi=0; xi<xrows; xi++)
      {
        int attr  = atoi(PQgetvalue(xq, xi, 0)),
            type  = atoi(PQgetvalue(xq, xi, 2)),
            value = atoi(PQgetvalue(xq, xi, 1)) + (type == 2);
        switch(attr)
        {
          case 0x3c : case 0x170 : prof = value; break;
          case 0x36 : lvl = value; break;
          case 0x7a : si = value; break;
          case 0x7f : mm = value; break;
          case 0x80 : bm = value; break;
          case 0x81 : pm = value; break;
          case 0x82 : mc = value; break;
          case 0x83 : ts = value; break;
          case 0x1c7 : skip++; break; // NPC
          case 0x185 : es = value; break;
        }
      }
      PQclear(xq);
      xq = execf(dbconn, "select value from aodb_other where xid = %d and attribute = %d", xid, 0x227);
      if(PQntuples(xq))
        stack = atoi(PQgetvalue(xq, 0, 0));
      else
        stack = ncu;
      PQclear(xq);
      
      PQescapeString(xname, name, strlen(name));
      PQescapeString(xinfo, info, strlen(info));
      PQclear(execf(dbconn, "insert into nanos values (%d, %d, %d, %d, %d, %d, %d, %d, %d, %d, %d, %d, %d, %d, %d, %d, '%s', '%s')",
        aoid, icon, school, strain, es, ql, ncu, stack, prof, lvl, mm, bm, pm, mc, ts, si, xname, xinfo));
      printf("%06d :: %s\n", aoid, name);
    }
    PQclear(PQexec(dbconn, "update nanos set lvl = NULL where lvl = 0"));
    PQclear(PQexec(dbconn, "update nanos set mm = NULL where mm = 0"));
    PQclear(PQexec(dbconn, "update nanos set bm = NULL where bm = 0"));
    PQclear(PQexec(dbconn, "update nanos set pm = NULL where pm = 0"));
    PQclear(PQexec(dbconn, "update nanos set mc = NULL where mc = 0"));
    PQclear(PQexec(dbconn, "update nanos set ts = NULL where ts = 0"));
    PQclear(PQexec(dbconn, "update nanos set si = NULL where si = 0"));
    PQclear(PQexec(dbconn, "vacuum full analyze nanos"));
  }

  PQclear(q);
  PQfinish(dbconn);
  return 0;
}

static PGresult *
execf(PGconn *db, const char *fmt, ...)
{
  va_list ap;
  char *str;
  
  va_start(ap, fmt);
  vasprintf(&str, fmt, ap);
  va_end(ap);
  
  PGresult *q = PQexec(db, str);
  free(str);
  return q;
}