/*
** Copyright 2003, Oliver Tappe, zooey@hirschkaefer.de. All rights reserved.
** Distributed under the terms of the MIT License.
*/

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

#include <Application.h>
#include <Catalog.h>
#include <DefaultCatalog.h>
#include <Entry.h>
#include <Locale.h>
#include <Path.h>
#include <Roster.h>

class CatalogTest {
        public:
                void Run();
                void Check();
};

#define B_TRANSLATION_CONTEXT "CatalogTest"


#define catSig "x-vnd.Be.locale.catalogTest"
#define catName catSig".catalog"

void
CatalogTest::Run()
{
        printf("app...");
        status_t res;
        BString s;
        s << "string" << "\x01" << B_TRANSLATION_CONTEXT << "\x01";
        size_t hashVal = CatKey::HashFun(s.String());
        assert(be_locale != NULL);
        system("mkdir -p ./locale/catalogs/"catSig);

        // create an empty catalog of default type...
        BPrivate::EditableCatalog cata("Default", catSig, "German");
        assert(cata.InitCheck() == B_OK);

        // ...and populate the catalog with some data:
        res = cata.SetString("string", "Schnur", B_TRANSLATION_CONTEXT);
        assert(res == B_OK);
        res = cata.SetString(hashVal, "Schnur_id");
                // add a second entry for the same hash-value, but with different
                // translation
        assert(res == B_OK);
        res = cata.SetString("string", "String", "programming");
        assert(res == B_OK);
        res = cata.SetString("string", "Textpuffer", "programming",
                "Deutsches Fachbuch");
        assert(res == B_OK);
        res = cata.SetString("string", "Leine", B_TRANSLATION_CONTEXT,
                "Deutsches Fachbuch");
        assert(res == B_OK);
        res = cata.WriteToFile("./locale/catalogs/"catSig"/german.catalog");
        assert(res == B_OK);

        // check if we are getting back the correct strings:
        s = cata.GetString(("string"), B_TRANSLATION_CONTEXT);
        assert(s == "Schnur");
        s = cata.GetString(hashVal);
        assert(s == "Schnur_id");
        s = cata.GetString("string", "programming");
        assert(s == "String");
        s = cata.GetString("string", "programming", "Deutsches Fachbuch");
        assert(s == "Textpuffer");
        s = cata.GetString("string", B_TRANSLATION_CONTEXT, "Deutsches Fachbuch");
        assert(s == "Leine");

        // now we create a new (base) catalog and embed this one into the app-file:
        BPrivate::EditableCatalog catb("Default", catSig, "English");
        assert(catb.InitCheck() == B_OK);
        // the following string is unique to the embedded catalog:
        res = catb.SetString("string", "string", "base");
        assert(res == B_OK);
        // the following id is unique to the embedded catalog:
        res = catb.SetString(32, "hashed string");
        assert(res == B_OK);
        // the following string will be hidden by the definition inside the
        // german catalog:
        res = catb.SetString("string", "hidden", B_TRANSLATION_CONTEXT);
        assert(res == B_OK);
        app_info appInfo;
        res = be_app->GetAppInfo(&appInfo);
        assert(res == B_OK);
        // embed created catalog into application file (catalogTest):
        res = catb.WriteToResource(&appInfo.ref);
        assert(res == B_OK);

        printf("ok.\n");
        Check();
}


void
CatalogTest::Check()
{
        status_t res;
        printf("app-check...");
        BString s;
        s << "string" << "\x01" << B_TRANSLATION_CONTEXT << "\x01";
        size_t hashVal = CatKey::HashFun(s.String());
        // ok, we now try to re-load the catalog that has just been written:
        //
        // actually, the following code can be seen as an example of what an
        // app needs in order to translate strings:
        BCatalog cat;
        res = be_locale->GetAppCatalog(&cat);
        assert(res == B_OK);
        // fetch basic data:
        uint32 fingerprint = 0;
        res = cat.GetFingerprint(&fingerprint);
        assert(res == B_OK);
        BString lang;
        res = cat.GetLanguage(&lang);
        assert(res == B_OK);
        BString sig;
        res = cat.GetSignature(&sig);
        assert(res == B_OK);

        // now check strings:
        s = B_TRANSLATE_ID(hashVal);
        assert(s == "Schnur_id");
        s = B_TRANSLATE_ALL("string", "programming", "");
        assert(s == "String");
        s = B_TRANSLATE_ALL("string", "programming", "Deutsches Fachbuch");
        assert(s == "Textpuffer");
        s = B_TRANSLATE_COMMENT("string", "Deutsches Fachbuch");
        assert(s == "Leine");
        // the following string should be found in the embedded catalog only:
        s = B_TRANSLATE_ALL("string", "base", NULL);
        assert(s == "string");
        // the following id should be found in the embedded catalog only:
        s = B_TRANSLATE_ID(32);
        assert(s == "hashed string");
        // the following id doesn't exist anywhere (hopefully):
        s = B_TRANSLATE_ID(-1);
        assert(s == "");
        // the following string exists twice, in the embedded as well as in the
        // external catalog. So we should get the external translation (as it should
        // override the embedded one):
        s = B_TRANSLATE("string");
        assert(s == "Schnur");

        // now check if trying to access same catalog by specifying its data works:
        BCatalog cat2(sig.String(), lang.String(), fingerprint);
        assert(cat2.InitCheck() == B_OK);
        // now check if trying to access same catalog with wrong fingerprint fails:
        BCatalog cat3(sig.String(), lang.String(), fingerprint*-1);
        assert(cat3.InitCheck() == B_NO_INIT);
        // translating through an invalid catalog should yield the native string:
        s = cat3.GetString("string");
        assert(s == "string");

        printf("ok.\n");
}


int
main(int argc, char **argv)
{
        BApplication* testApp
                = new BApplication("application/"catSig);

        // change to app-folder:
        app_info appInfo;
        be_app->GetAppInfo(&appInfo);
        BEntry appEntry(&appInfo.ref);
        BEntry appFolder;
        appEntry.GetParent( &appFolder);
        BPath appPath;
        appFolder.GetPath( &appPath);
        chdir( appPath.Path());

        CatalogTest catTest;
        catTest.Run();

        char cwd[B_FILE_NAME_LENGTH];
        getcwd(cwd, B_FILE_NAME_LENGTH);
        BString addonName(cwd);
        addonName << "/" "catalogTestAddOn";
        image_id image = load_add_on(addonName.String());
        assert(image >= B_OK);
        void (*runAddonFunc)() = 0;
        get_image_symbol(image, "run_test_add_on",
                B_SYMBOL_TYPE_TEXT, (void **)&runAddonFunc);
        assert(runAddonFunc);
        runAddonFunc();

        catTest.Check();

        delete testApp;
}