genbookutil.cpp

Go to the documentation of this file.

/******************************************************************************
 *
 *  genbookutil.cpp -   
 *
 * $Id: genbookutil.cpp 3063 2014-03-04 13:04:11Z chrislit $
 *
 * Copyright 2002-2013 CrossWire Bible Society (http://www.crosswire.org)
 *  CrossWire Bible Society
 *  P. O. Box 2528
 *  Tempe, AZ  85280-2528
 *
 * 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 version 2.
 *
 * 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.
 *
 */

#ifdef _MSC_VER
    #pragma warning( disable: 4251 )
#endif

#include <entriesblk.h>
#include <iostream>
#include <stdio.h>
#include <treekeyidx.h>
#include <rawgenbook.h>

#ifndef NO_SWORD_NAMESPACE
using namespace sword;
#endif

void printTree(TreeKeyIdx treeKey, TreeKeyIdx *target = 0, int level = 1) {
    if (!target)
        target = &treeKey;
    
    unsigned long currentOffset = target->getOffset();
    std::cout << ((currentOffset == treeKey.getOffset()) ? "==>" : "");
    for (int i = 0; i < level; i++) std::cout << "\t";
    std::cout << treeKey.getLocalName() << "/\n";
    if (treeKey.firstChild()) {
        printTree(treeKey, target, level+1);
        treeKey.parent();
    }
    if (treeKey.nextSibling())
        printTree(treeKey, target, level);

}


void printLocalName(TreeKeyIdx *treeKey) {
    std::cout << "locaName: " << treeKey->getLocalName() << "\n";
}


void setLocalName(TreeKeyIdx *treeKey) {
    char buf[1023];
    std::cout << "Enter New Node Name: ";
    fgets(buf, 1000, stdin);
    SWBuf name = buf;
    treeKey->setLocalName(name.trim());
    treeKey->save();
}


void gotoPath(TreeKeyIdx *treeKey) {
    char buf[1023];
    std::cout << "Enter Path: ";
    fgets(buf, 1000, stdin);
    SWBuf path = buf;
    (*treeKey) = path.trim();
}


void assurePath(TreeKeyIdx *treeKey) {
    char buf[1023];
    std::cout << "Enter Path: ";
    fgets(buf, 1000, stdin);
    SWBuf path = buf;
    treeKey->assureKeyPath(path.trim());
}


void viewEntryText(RawGenBook *book) {
    std::cout << "\n";
    std::cout << book->renderText();
    std::cout << "\n";
}


void setEntryText(RawGenBook *book) {
    SWBuf body;
    TreeKeyIdx *treeKey = (TreeKeyIdx *)(SWKey *)(*book);
    if (treeKey->getOffset()) {
        char buf[1023];
        std::cout << "Enter New Entry Text ('.' on a line by itself to end): \n";
        do {
            fgets(buf, 1000, stdin);
            SWBuf text = buf;
            text.trim();
            if ((text[0] == '.') && (text[1] == 0))
                break;
            body += text;
            body += "\n";
        } while (true);

        (*book) << body.c_str();
    }
    else    std::cout << "Can't add entry text to root node\n";
}


void appendSibbling(TreeKeyIdx *treeKey) {
    if (treeKey->getOffset()) {
        char buf[1023];
        std::cout << "Enter New Sibbling Name: ";
        fgets(buf, 1000, stdin);
        SWBuf name = buf;
        treeKey->append();
        treeKey->setLocalName(name.trim());
        treeKey->save();
    }
    else    std::cout << "Can't add sibling to root node\n";
}


void appendChild(TreeKeyIdx *treeKey) {
    char buf[1023];
    std::cout << "Enter New Child Name: ";
    fgets(buf, 1000, stdin);
    SWBuf name = buf;
    treeKey->appendChild();
    treeKey->setLocalName(name.trim());
    treeKey->save();
}


void deleteNode(TreeKeyIdx *treeKey) {
    std::cout << "Removing entry [" << treeKey->getText() << "]\n";
    treeKey->remove();
}


void removeEntry(EntriesBlock *eb, int index) {
    if (index < eb->getCount()) {
        std::cout << "Removing entry [" << index << "]\n";
        eb->removeEntry(index);
    }
    else std::cout << "Invalid entry number\n\n";
}


int main(int argc, char **argv) {

    if (argc != 2) {
        fprintf(stderr, "usage: %s <tree/key/data/path>\n", *argv);
        exit(-1);
    }

    TreeKeyIdx *treeKey = new TreeKeyIdx(argv[1]);

    if (treeKey->popError()) {
        RawGenBook::createModule(argv[1]);
    }
    delete treeKey;

    RawGenBook *book = new RawGenBook(argv[1]);
    TreeKeyIdx root = *((TreeKeyIdx *)((SWKey *)(*book)));
    treeKey = (TreeKeyIdx *)(SWKey *)(*book);

    SWBuf input;
    char line[1024];

    do {
        std::cout << "[" << treeKey->getText() << "] > ";
        fgets(line, 1000, stdin);
        input = line;
        input.trim();
        if (input.length() > 0) {
            switch (input[0]) {
                case 'n': printLocalName(treeKey); break;
                case 's': setLocalName(treeKey); break;
                case 'g': gotoPath(treeKey); break;
                case 'G': assurePath(treeKey); break;
                case 'p':   root.root(); printTree(root, treeKey); break;
                case 'a':   appendSibbling(treeKey); break;
                case 'c':   appendChild(treeKey); break;
                case 'd':   deleteNode(treeKey); break;
                case 'j':   treeKey->nextSibling(); break;
                case 'k':   treeKey->previousSibling(); break;
                case 'h':   treeKey->parent(); break;
                case 'l':   treeKey->firstChild(); break;
                case 'r':   treeKey->root(); break;
                case 't':   setEntryText(book); break;
                case 'v':   viewEntryText(book); break;
                case 'q': break;
                case '?':
                default:
                    std::cout << "\n p - print tree\n";
                    std::cout << " n - get local name\n";
                    std::cout << " s - set local name\n";
                    std::cout << " j - next sibbling\n";
                    std::cout << " k - previous sibbling\n";
                    std::cout << " h - parent\n";
                    std::cout << " l - first child\n";
                    std::cout << " r - root\n";
                    std::cout << " g - goto path\n";
                    std::cout << " G   goto path; create if it doesn't exist\n";
                    std::cout << " a - append sibbling\n";
                    std::cout << " c - append child\n";
                    std::cout << " d - delete node\n";
                    std::cout << " v - view entry text\n";
                    std::cout << " t - set entry text\n";
                    std::cout << " q - quit\n\n";
                    break;
            }
        }
    }
    while (input.compare("q"));

    delete treeKey;

    return 0;
}