// ================================================================= //
//                                                                   //
//   File      : ptclean.cxx                                         //
//   Purpose   : prepare db for ptserver/ptpan                       //
//                                                                   //
//   Coded by Ralf Westram (coder@reallysoft.de) in September 2011   //
//   Institute of Microbiology (Technical University Munich)         //
//   http://www.arb-home.de/                                         //
//                                                                   //
// ================================================================= //

#include "ptclean.h"
#include <arbdbt.h>
#include <arb_progress.h>
#include <arb_diff.h>

#define pt_assert(cond) arb_assert(cond)

class EntryRemover : virtual Noncopyable {
    // deletes all entries from DB, which are of no use for PTSERVER

    GBDATA         *gb_main;
    GB_transaction  ta;
    char           *ali_name;

    mutable arb_progress progress;

    enum Need {
        NONE,
        ALL,
        SOME_OF_ROOT,
        SOME_OF_SPECIES_DATA,
        SOME_OF_EXTENDED_DATA,
        SOME_OF_SPECIES,
        SOME_OF_EXTENDED,
        SOME_OF_ALI_CONTAINER, // below species or SAI
    };

    Need data_needed(GBDATA *gbd, const char *keyname, Need from) const {
        switch (from) {
            case SOME_OF_ROOT:
                if (strcmp(keyname, GB_SYSTEM_FOLDER)== 0) return ALL;
                if (strcmp(keyname, "genom_db")      == 0) return ALL;
                if (strcmp(keyname, "gene_map")      == 0) return ALL;
                if (strcmp(keyname, "species_data")  == 0) return SOME_OF_SPECIES_DATA;
                if (strcmp(keyname, "extended_data") == 0) return SOME_OF_EXTENDED_DATA;
                if (strcmp(keyname, "presets")       == 0) return ALL;
                break;

            case SOME_OF_SPECIES_DATA:
                if (strcmp(keyname, "species") == 0) {
                    progress.inc();
                    return SOME_OF_SPECIES;
                }
                break;

            case SOME_OF_EXTENDED_DATA:
                if (strcmp(keyname, "extended") == 0) {
                    const char *sainame = GBT_get_name_or_description(gbd);
                    if (strcmp(sainame, "ECOLI") == 0) return SOME_OF_EXTENDED;
                    if (strcmp(sainame, "HELIX") == 0) return SOME_OF_EXTENDED;
                    if (strcmp(sainame, "HELIX_NR") == 0) return SOME_OF_EXTENDED;
                }
                break;

            case SOME_OF_SPECIES:
                if (strcmp(keyname, "abspos") == 0) return ALL;
                if (strcmp(keyname, "start")  == 0) return ALL;
                if (strcmp(keyname, "stop")   == 0) return ALL;
                FALLTHROUGH;

            case SOME_OF_EXTENDED:
                if (strcmp(keyname, "name")      == 0) return ALL;
                if (strcmp(keyname, "acc")       == 0) return ALL;
                if (strcmp(keyname, "full_name") == 0) return ALL;
                if (strcmp(keyname, ali_name)    == 0) return SOME_OF_ALI_CONTAINER;
                break;

            case SOME_OF_ALI_CONTAINER:
                if (strcmp(keyname, "data") == 0) return ALL;
                break;

            case NONE: pt_assert(0); break;
            case ALL:  pt_assert(0); break;
        }

        return NONE;
    }

    GB_ERROR del_child(GBDATA *gb_entry, const char *keyname, Need from) {
        GB_ERROR error = NULp;
        Need     need  = data_needed(gb_entry, keyname, from);
        switch (need) {
            case NONE: error = GB_delete(gb_entry); break;
            case ALL:  break;
            default:   error = del_subentries(gb_entry, need); break;
        }
        return error;
    }
    GB_ERROR del_subentries(GBDATA *gb_father, Need from) {
        GB_ERROR error = NULp;
        if (!GB_is_temporary(gb_father)) {
            GBDATA *gb_next_child = NULp;
            for (GBDATA *gb_child = GB_child(gb_father); gb_child && !error; gb_child = gb_next_child) {
                gb_next_child = GB_nextChild(gb_child);
                const char *key = GB_read_key_pntr(gb_child);
                error           = del_child(gb_child, key, from);
            }
        }
        return error;
    }

public:
    EntryRemover(GBDATA *gb_main_)
        : gb_main(gb_main_),
          ta(gb_main),
          ali_name(GBT_get_default_alignment(gb_main)),
          progress("Remove unused database entries", GBT_get_species_count(gb_main))
    {}
    ~EntryRemover() {
        free(ali_name);
        progress.done();
    }

    GB_ERROR del_unwanted_entries() {
        GB_ERROR error;
        if (!ali_name) error = GB_await_error(); // (should) handle error from GBT_get_default_alignment called in ctor
        else           error = del_subentries(gb_main, SOME_OF_ROOT);
        return error;
    }
};

inline GB_ERROR clean_ptserver_database(GBDATA *gb_main) {
    return EntryRemover(gb_main).del_unwanted_entries();
}

GB_ERROR cleanup_ptserver_database(GBDATA *gb_main) {
    GB_ERROR error = GB_request_undo_type(gb_main, GB_UNDO_NONE);
    if (!error) {
        GB_topSecurityLevel unsecured(gb_main);
        error = clean_ptserver_database(gb_main);
    }
    return error;
}
// --------------------------------------------------------------------------------

#ifdef UNIT_TESTS
#ifndef TEST_UNIT_H
#include <test_unit.h>
#endif
#include <arb_file.h>

void TEST_SLOW_ptclean() {
    GB_shell    shell;
    GBDATA     *gb_main = GB_open("TEST_pt_src.arb", "rw");
    const char *saveas  = "TEST_pt_cleaned.arb";

    TEST_REJECT_NULL(gb_main);
    TEST_EXPECT_NO_ERROR(cleanup_ptserver_database(gb_main));
    TEST_EXPECT_NO_ERROR(GB_save_as(gb_main, saveas, "a"));
    GB_close(gb_main);

// #define TEST_AUTO_UPDATE
#if defined(TEST_AUTO_UPDATE)
    TEST_COPY_FILE("TEST_pt_cleaned.arb", "TEST_pt_cleaned_expected.arb");
#else
    TEST_EXPECT_TEXTFILES_EQUAL("TEST_pt_cleaned.arb", "TEST_pt_cleaned_expected.arb");
#endif
    TEST_EXPECT_ZERO_OR_SHOW_ERRNO(GB_unlink(saveas));
}
TEST_PUBLISH(TEST_SLOW_ptclean);

#endif // UNIT_TESTS

// --------------------------------------------------------------------------------