merge.i

%module merge

%include "../common.i"

%include <stl.i>
%include "../common.i"
%feature("autodoc", "3");

%import "kdb.i"

%{
#include "kdbmerge.h"
%}

ckdb::KeySet * elektraMerge (ckdb::KeySet * our, ckdb::Key * ourRoot, ckdb::KeySet * their, ckdb::Key * theirRoot, ckdb::KeySet * base, ckdb::Key * baseRoot, ckdb::Key * resultKey,
		      ckdb::MergeStrategy strategy, ckdb::Key * informationKey);
int elektraMergeGetConflicts (ckdb::Key * informationKey);
ckdb::KeySet * elektraMergeGetConflictingKeys (ckdb::Key * informationKey, ckdb::Key * root);

%inline %{
 	// we wrap the elektraMerge function, because I haven't found a way to correctly wrap the enum ...
	ckdb::KeySet * elektraMergeWrap (ckdb::KeySet * our, ckdb::Key * ourRoot, ckdb::KeySet * their, ckdb::Key * theirRoot, ckdb::KeySet * base, ckdb::Key * baseRoot, ckdb::Key * resultKey,
		     int strategy, ckdb::Key * informationKey) {
	       return elektraMerge (our, ourRoot, their, theirRoot, base, baseRoot, resultKey, (ckdb::MergeStrategy) strategy, informationKey);
       }
%}

%pythoncode {
from enum import Enum

class ConflictStrategy(Enum):
  ABORT = 1
  OUR = 3
  THEIR = 4

class MergeKeys:
  def __init__(self, keys, parentKey):
    self.keys = keys
    self.root = parentKey

class MergeResult:
  def __init__(self, mergedKeys, informationKey):
    if mergedKeys is None:
       self.mergedKeys = None
    else:
       self.mergedKeys = kdb.KeySet(mergedKeys)

    self.mergeInformation = informationKey

  def hasConflicts(self):
    return self.mergedKeys is None or elektraMergeGetConflicts(self.mergeInformation.getKey()) > 0

  def getConflictingKeys(self, rootKey = kdb.Key("/")):
    ckeys = elektraMergeGetConflictingKeys(self.mergeInformation.getKey(), rootKey.getKey())
    return kdb.KeySet(ckeys)

class Merger:
  def merge(self, base, ours, theirs, root, conflictStrategy):
    informationKey = kdb.Key()
    res = elektraMergeWrap(ours.keys.getKeySet(), ours.root.getKey(), theirs.keys.getKeySet(), theirs.root.getKey(), base.keys.getKeySet(), base.root.getKey(), root.getKey(), conflictStrategy.value, informationKey.getKey())
    return MergeResult(res, informationKey)
};