/home/paulfitz/cvs/coopy_scm/coopy/src/libcoopy_core/MergeOutputSqlDiff.cpp

Go to the documentation of this file.

#include <coopy/MergeOutputSqlDiff.h>

#include <stdio.h>
#include <stdlib.h>

#define WANT_MAP2STRING
#define WANT_VECTOR2STRING
#include <coopy/Stringer.h>

using namespace std;
using namespace coopy::cmp;

static void getSqlQuote(const CompareFlags *flags, char *key, char *val) {
  char k = '\"';
  char v = '\'';
  if (flags!=NULL) {
    if (flags->variant=="access") {
      k = '[';
      v = '\'';
    }
  }
  *key = k;
  *val = v;
}

bool MergeOutputSqlDiff::changeColumn(const OrderChange& change) {
  char k, v;
  getSqlQuote(&getFlags(),&k,&v);
  string name = quoteSql(sheet_name,k,false);
  switch (change.mode) {
  case ORDER_CHANGE_DELETE:
    {
      int idx = change.identityToIndex(change.subject);
      if (change.namesBefore.size()<=idx) {
        fprintf(stderr, "Could not find column to remove\n");
        exit(1);
      } else {
        string c = quoteSql(change.namesBefore[idx],k,false);
        fprintf(out,"ALTER TABLE %s DROP COLUMN %s;\n",
                name.c_str(),
                c.c_str());
      }
    }       
    break;
  case ORDER_CHANGE_INSERT:
    {
      int idx = change.identityToIndexAfter(change.subject);
      if (change.namesAfter.size()<=idx) {
        fprintf(stderr, "Could not find column to insert\n");
        exit(1);
      } else {
        string c = quoteSql(change.namesAfter[idx],k,false);
        fprintf(out,"ALTER TABLE %s ADD COLUMN %s;\n",
                name.c_str(),
                c.c_str());
      }
    }
    break;
  case ORDER_CHANGE_RENAME:
    {
      int idx = change.identityToIndexAfter(change.subject);
      if (change.namesAfter.size()<=idx) {
        fprintf(stderr, "Could not find column to rename\n");
        exit(1);
      } else {
        string c = quoteSql(change.namesAfter[idx],k,false);
        string c2 = quoteSql(change.namesBefore[idx],k,false);
        fprintf(out,"ALTER TABLE %s RENAME COLUMN %s TO %s;\n",
                name.c_str(),
                c.c_str(),
                c2.c_str());
      }
    }
    break;
  }
  return true;
}


SqlText MergeOutputSqlDiff::getText(const RowChange& change, const char *sheet_name, const CompareFlags *flags) {
  SqlText text;
  string& name = text.name;
  string& vals = text.vals;
  string& conds = text.conds;
  string& val_columns = text.val_columns;
  string& val_values = text.val_values;
  char del1 = '\"';
  char del2 = '\'';
  getSqlQuote(flags,&del1,&del2);

  name = quoteSql(sheet_name,del1,false);
  for (RowChange::txt2cell::const_iterator it=change.val.begin(); 
       it!=change.val.end(); 
       it++) {
    if (it!=change.val.begin()) {
      vals += ", ";
      val_columns += ", ";
      val_values += ", ";
    }
    // Should dig up good default quoting rules
    string c = quoteSql(it->first,del1,false);
    vals += c;
    val_columns += c;
    vals += "=";
    if (it->second.escaped) {
      vals += "NULL";
      val_values += "NULL";
    } else {
      string q = quoteSql(it->second.text,del2,true);
      vals += q;
      val_values += q;
    }
  }

  for (RowChange::txt2cell::const_iterator it=change.cond.begin(); 
       it!=change.cond.end(); 
       it++) {
    if (it!=change.cond.begin()) {
      conds += " AND ";
    }
    string c = quoteSql(it->first,del1,false);
    conds += c;
    conds += "=";
    if (it->second.escaped) {
      conds += "NULL";
    } else {
      string q = quoteSql(it->second.text,del2,true);
      conds += q;
    }
  }
  return text;
}

bool MergeOutputSqlDiff::changeRow(const RowChange& change) {
  SqlText text = getText(change,sheet_name.c_str(),
                         &getFlags());
  string& name = text.name;
  string& vals = text.vals;
  string& conds = text.conds;
  string& val_columns = text.val_columns;
  string& val_values = text.val_values;

  switch (change.mode) {
  case ROW_CHANGE_INSERT:
    fprintf(out,"INSERT INTO %s (%s) VALUES (%s);\n", 
            name.c_str(),
            val_columns.c_str(),
            val_values.c_str());
    break;
  case ROW_CHANGE_DELETE:
    fprintf(out,"DELETE FROM %s WHERE %s;\n",
            name.c_str(),
            conds.c_str());
    break;
  case ROW_CHANGE_UPDATE:
    fprintf(out,"UPDATE %s SET %s WHERE %s;\n", 
            name.c_str(),
            vals.c_str(),
            conds.c_str());
    break;
  default:
    // skip
    break;
  }
  return true;
}