/home/paulfitz/cvs/coopy_scm/coopy/src/ssfossil/src/csv_hooks.cpp

Go to the documentation of this file.

#include <stdio.h>

extern "C" {
#include "config.h"
#include "blob.h"
}

extern "C" { 
#include "csv.h"
}

#include <coopy/SheetCompare.h>
#include <coopy/CsvRender.h>
#include <coopy/CsvTextBook.h>
#include <coopy/SheetPatcher.h>
#include <coopy/PolyBook.h>
#include <coopy/BookCompare.h>

#include "ssfossil.h"

using namespace coopy::store;
using namespace coopy::cmp;

extern "C" int csv_merge(Blob *pPivot, Blob *pV1, Blob *pV2, Blob *pOut);
extern "C" int csv_render(Blob *in, Blob *out);
extern "C" int csvs_diff(Blob *pV1, Blob *pV2, Blob *pOut);

extern "C" void cvs_merge_cb1 (void *s, size_t i, void *p, int quoted) {
  ((CsvSheet*)p)->addField((char *)s, i);
}

extern "C" void cvs_merge_cb2 (int c, void *p) {
  ((CsvSheet*)p)->addRecord();
}

int blob_to_csv(Blob *pIn, CsvSheet& csv) {
  SheetStyle style;
  if (pIn==NULL) return -1;
  struct csv_parser p;
  csv_init(&p,0); // CSV_APPEND_NULL does not seem reliable
  csv_set_delim(&p,style.getDelimiter()[0]);
  int result = csv_parse(&p,
                         blob_buffer(pIn),
                         blob_size(pIn),
                         cvs_merge_cb1,
                         cvs_merge_cb2,
                         (void*)(&csv));
  csv_fini(&p,cvs_merge_cb1,
           cvs_merge_cb2,
           (void*)(&csv));
  csv_free(&p);
  if (result!=(int)blob_size(pIn)) {
    return -1;
  }
  if (!csv.isValid()) {
    return -1;
  }
  return 0;
}

int blob_to_csvs(Blob *pIn, CsvTextBook& csvs) {
  if (pIn==NULL) return -1;
  if (!csvs.readCsvsData(blob_buffer(pIn),blob_size(pIn))) {
    return -1;
  }
  return 0;
}


void blob_show_csv(const DataSheet& csv, const SheetStyle& style, Blob *pOut) {
  blob_appendf(pOut,"%s",csv.encode(style).c_str());
}

void blob_show_csvs(CsvTextBook& csvs, Blob *pOut) {
  blob_appendf(pOut,"%s",csvs.writeCsvsData().c_str());
}

/*
  return 1 if a result has been set, otherwise 0; -1 aborts.
 */
int csv_merge(Blob *pPivot, Blob *pV1, Blob *pV2, Blob *pOut) {
  if (ssfossil_debug()) {
    printf("csv_merge\n");
    printf("BASE [%s]\n", blob_buffer(pPivot));
    printf("VERSION 1 [%s]\n", blob_buffer(pV1));
    printf("VERSION 2 [%s]\n", blob_buffer(pV2));
  }
  CsvTextBook csvs0(true);
  CsvTextBook csvs1(true);
  CsvTextBook csvs2(true);
  if (blob_to_csvs(pPivot,csvs0)==0 && 
      blob_to_csvs(pV1,csvs1)==0 && 
      blob_to_csvs(pV2,csvs2)==0) {
    BookCompare merger;
    Patcher *p = Patcher::createByName("merge");
    if (!p) {
      fprintf(stderr,"Out of memory.\n");
      exit(1);
    }
    p->attachOutputBook(csvs1);
    CompareFlags flags;
    int r = merger.compare(csvs0,csvs1,csvs2,*p,flags);
    delete p;
    p = NULL;
    if (r==0) {
      blob_zero(pOut);
      //blob_appendf(pOut,"Hello from %s:%d\n", __FILE__, __LINE__);
      //blob_appendf(pOut,"Conflict resolution is being modified.\n");
      blob_show_csvs(csvs1,pOut);
      return 1;
    }
  }
  return 0;
}



/*
  return 1 if a result has been set, otherwise 0; -1 aborts.
 */
int csv_render(Blob *in, Blob *out) {
  CsvSheet sheet;
  if (blob_to_csv(in,sheet)==0) {
    CsvRender render;
    render.setFull(false);
    render.setDecorate(true);
    std::string result = render.renderHtml(sheet);
    blob_appendf(out,"%s",result.c_str());
    return 1;
  }
  return 0;
}


extern "C" int csvs_diff(Blob *pV1, Blob *pV2, Blob *pOut) {
  CsvTextBook book1(true);
  CsvTextBook book2(true);
  if (blob_to_csvs(pV1,book1)!=0) {
    return 1;
  }
  if (blob_to_csvs(pV2,book2)!=0) {
    return 1;
  }
  PolyBook obook;
  //MergeOutputTdiff highlighter;
  SheetPatcher *highlighter = SheetPatcher::createForDescription();
  highlighter->attachOutputBook(obook);
  COOPY_ASSERT(highlighter);
  CompareFlags flags;
  BookCompare cmp;
  PolyBook tbook;
  tbook.take(new CsvTextBook(true));
  Property p;
  tbook.copy(book1,p);
  highlighter->attachBook(tbook);
  cmp.compare(book1,book1,book2,*highlighter,flags);
  delete highlighter;  highlighter = NULL;

  PolySheet result = tbook.readSheetByIndex(0);
  blob_zero(pOut);
  blob_show_csv(result,SheetStyle(),pOut);
  return 0;
}