/* This file is part of the Zebra server.
   Copyright (C) Index Data

Zebra 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; either version 2, or (at your option) any later
version.

Zebra 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.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

*/

#if HAVE_CONFIG_H
#include <config.h>
#endif
#include <stdlib.h>
#include <assert.h>
#include <string.h>
#include <ctype.h>

#include <yaz/log.h>
#include <yaz/odr.h>
#include <yaz/proto.h>
#include <idzebra/data1.h>
#include <yaz/oid_db.h>

static Z_Variant *read_variant(int argc, char **argv, NMEM nmem,
                               const char *file, int lineno)
{
    Z_Variant *r = (Z_Variant *)nmem_malloc(nmem, sizeof(*r));
    int i;

    r->globalVariantSetId = odr_oiddup_nmem(nmem, yaz_oid_varset_variant_1);
    if (argc)
        r->triples = (Z_Triple **)nmem_malloc(nmem, sizeof(Z_Triple*) * argc);
    else
        r->triples = 0;
    r->num_triples = argc;
    for (i = 0; i < argc; i++)
    {
        int zclass, type;
        char value[512];
        Z_Triple *t;

        if (sscanf(argv[i], "(%d,%d,%511[^)])", &zclass, &type, value) < 3)
        {
            yaz_log(YLOG_WARN, "%s:%d: Syntax error in variant component '%s'",
                    file, lineno, argv[i]);
            return 0;
        }
        t = r->triples[i] = (Z_Triple *)nmem_malloc(nmem, sizeof(Z_Triple));
        t->variantSetId = 0;
        t->zclass = nmem_intdup(nmem, zclass);
        t->type = nmem_intdup(nmem, type);
        /*
         * This is wrong.. we gotta look up the correct type for the
         * variant, I guess... damn this stuff.
         */
        if (*value == '@')
        {
            t->which = Z_Triple_null;
            t->value.null = odr_nullval();
        }
        else if (d1_isdigit(*value))
        {
            t->which = Z_Triple_integer;
            t->value.integer = nmem_intdup(nmem, atoi(value));
        }
        else
        {
            t->which = Z_Triple_internationalString;
            t->value.internationalString = nmem_strdup(nmem, value);
        }
    }
    return r;
}

static Z_Occurrences *read_occurrences(char *occ, NMEM nmem,
                                       const char *file, int lineno)
{
    Z_Occurrences *op = (Z_Occurrences *)nmem_malloc(nmem, sizeof(*op));
    char *p;

    if (!occ)
    {
        op->which = Z_Occurrences_values;
        op->u.values = (Z_OccurValues *)
            nmem_malloc(nmem, sizeof(Z_OccurValues));
        op->u.values->start = nmem_intdup(nmem, 1);
        op->u.values->howMany = 0;
    }
    else if (!strcmp(occ, "all"))
    {
        op->which = Z_Occurrences_all;
        op->u.all = odr_nullval();
    }
    else if (!strcmp(occ, "last"))
    {
        op->which = Z_Occurrences_last;
        op->u.all = odr_nullval();
    }
    else
    {
        Z_OccurValues *ov = (Z_OccurValues *)nmem_malloc(nmem, sizeof(*ov));

        if (!d1_isdigit(*occ))
        {
            yaz_log(YLOG_WARN, "%s:%d: Bad occurrences-spec %s",
                    file, lineno, occ);
            return 0;
        }
        op->which = Z_Occurrences_values;
        op->u.values = ov;
        ov->start = nmem_intdup(nmem, atoi(occ));
        if ((p = strchr(occ, '+')))
            ov->howMany = nmem_intdup(nmem, atoi(p + 1));
        else
            ov->howMany = 0;
    }
    return op;
}


static Z_ETagUnit *read_tagunit(char *buf, NMEM nmem,
                                const char *file, int lineno)
{
    Z_ETagUnit *u = (Z_ETagUnit *)nmem_malloc(nmem, sizeof(*u));
    int terms;
    int type;
    char value[512], occ[512];

    if (*buf == '*')
    {
        u->which = Z_ETagUnit_wildPath;
        u->u.wildPath = odr_nullval();
    }
    else if (*buf == '?')
    {
        u->which = Z_ETagUnit_wildThing;
        if (buf[1] == ':')
            u->u.wildThing = read_occurrences(buf+2, nmem, file, lineno);
        else
            u->u.wildThing = read_occurrences(0, nmem, file, lineno);
    }
    else if ((terms = sscanf(buf, "(%d,%511[^)]):%511[a-zA-Z0-9+]",
                             &type, value, occ)) >= 2)
    {
        int numval;
        Z_SpecificTag *t;
        char *valp = value;
        int force_string = 0;

        if (*valp == '\'')
        {
            valp++;
            force_string = 1;
            if (*valp && valp[strlen(valp)-1] == '\'')
                *valp = '\0';
        }
        u->which = Z_ETagUnit_specificTag;
        u->u.specificTag = t = (Z_SpecificTag *)nmem_malloc(nmem, sizeof(*t));
        t->tagType = nmem_intdup(nmem, type);
        t->tagValue = (Z_StringOrNumeric *)
            nmem_malloc(nmem, sizeof(*t->tagValue));
        if (!force_string && isdigit(*(unsigned char *)valp))
        {
            numval = atoi(valp);
            t->tagValue->which = Z_StringOrNumeric_numeric;
            t->tagValue->u.numeric = nmem_intdup(nmem, numval);
        }
        else
        {
            t->tagValue->which = Z_StringOrNumeric_string;
            t->tagValue->u.string = nmem_strdup(nmem, valp);
        }
        if (terms > 2) /* an occurrences-spec exists */
            t->occurrences = read_occurrences(occ, nmem, file, lineno);
        else
            t->occurrences = 0;
    }
    else if ((terms = sscanf(buf, "%511[^)]", value)) >= 1)
    {
        Z_SpecificTag *t;
        char *valp = value;

        u->which = Z_ETagUnit_specificTag;
        u->u.specificTag = t = (Z_SpecificTag *)nmem_malloc(nmem, sizeof(*t));
        t->tagType = nmem_intdup(nmem, 3);
        t->tagValue = (Z_StringOrNumeric *)
            nmem_malloc(nmem, sizeof(*t->tagValue));
        t->tagValue->which = Z_StringOrNumeric_string;
        t->tagValue->u.string = nmem_strdup(nmem, valp);
        t->occurrences = read_occurrences("all", nmem, file, lineno);
    }
    else
    {
        return 0;
    }
    return u;
}

/*
 * Read an element-set specification from a file.
 * NOTE: If !o, memory is allocated directly from the heap by nmem_malloc().
 */
Z_Espec1 *data1_read_espec1 (data1_handle dh, const char *file)
{
    FILE *f;
    NMEM nmem = data1_nmem_get (dh);
    int lineno = 0;
    int argc, size_esn = 0;
    char *argv[50], line[512];
    Z_Espec1 *res = (Z_Espec1 *)nmem_malloc(nmem, sizeof(*res));

    if (!(f = data1_path_fopen(dh, file, "r")))
        return 0;

    res->num_elementSetNames = 0;
    res->elementSetNames = 0;
    res->defaultVariantSetId = 0;
    res->defaultVariantRequest = 0;
    res->defaultTagType = 0;
    res->num_elements = 0;
    res->elements = 0;

    while ((argc = readconf_line(f, &lineno, line, 512, argv, 50)))
        if (!strcmp(argv[0], "elementsetnames"))
        {
            int nnames = argc-1, i;

            if (!nnames)
            {
                yaz_log(YLOG_WARN, "%s:%d: Empty elementsetnames directive",
                        file, lineno);
                continue;
            }

            res->elementSetNames =
                (char **)nmem_malloc(nmem, sizeof(char**)*nnames);
            for (i = 0; i < nnames; i++)
            {
                res->elementSetNames[i] = nmem_strdup(nmem, argv[i+1]);
            }
            res->num_elementSetNames = nnames;
        }
        else if (!strcmp(argv[0], "defaultvariantsetid"))
        {
            if (argc != 2)
            {
                yaz_log(YLOG_WARN, "%s:%d: Bad # of args for %s",
                        file, lineno, argv[0]);
                continue;
            }
            if (!(res->defaultVariantSetId =
                  odr_getoidbystr_nmem(nmem, argv[1])))
            {
                yaz_log(YLOG_WARN, "%s:%d: Bad defaultvariantsetid",
                        file, lineno);
                continue;
            }
        }
        else if (!strcmp(argv[0], "defaulttagtype"))
        {
            if (argc != 2)
            {
                yaz_log(YLOG_WARN, "%s:%d: Bad # of args for %s",
                        file, lineno, argv[0]);
                continue;
            }
            res->defaultTagType = nmem_intdup(nmem, atoi(argv[1]));
        }
        else if (!strcmp(argv[0], "defaultvariantrequest"))
        {
            if (!(res->defaultVariantRequest =
                  read_variant(argc-1, argv+1, nmem, file, lineno)))
            {
                yaz_log(YLOG_WARN, "%s:%d: Bad defaultvariantrequest",
                        file, lineno);
                continue;
            }
        }
        else if (!strcmp(argv[0], "simpleelement"))
        {
            Z_ElementRequest *er;
            Z_SimpleElement *se;
            Z_ETagPath *tp;
            char *path = argv[1];
            char *ep;
            int num, i = 0;

            if (!res->elements)
                res->elements = (Z_ElementRequest **)
                    nmem_malloc(nmem, size_esn = 24*sizeof(er));
            else if (res->num_elements >= (int) (size_esn/sizeof(er)))
            {
                Z_ElementRequest **oe = res->elements;
                size_esn *= 2;
                res->elements = (Z_ElementRequest **)
                    nmem_malloc (nmem, size_esn*sizeof(er));
                memcpy (res->elements, oe, size_esn/2);
            }
            if (argc < 2)
            {
                yaz_log(YLOG_WARN, "%s:%d: Bad # of args for %s",
                        file, lineno, argv[0]);
                continue;
            }

            res->elements[res->num_elements++] = er =
                (Z_ElementRequest *)nmem_malloc(nmem, sizeof(*er));
            er->which = Z_ERequest_simpleElement;
            er->u.simpleElement = se = (Z_SimpleElement *)
                nmem_malloc(nmem, sizeof(*se));
            se->variantRequest = 0;
            se->path = tp = (Z_ETagPath *)nmem_malloc(nmem, sizeof(*tp));
            tp->num_tags = 0;
            /*
             * Parse the element selector.
             */
            for (num = 1, ep = path; (ep = strchr(ep, '/')); num++, ep++)
                ;
            tp->tags = (Z_ETagUnit **)
                nmem_malloc(nmem, sizeof(Z_ETagUnit*)*num);

            for ((ep = strchr(path, '/')) ; path ;
                 (void)((path = ep) && (ep = strchr(path, '/'))))
            {
                Z_ETagUnit *tagunit;
                if (ep)
                    ep++;

                assert(i<num);
                tagunit = read_tagunit(path, nmem, file, lineno);
                if (!tagunit)
                {
                    yaz_log (YLOG_WARN, "%s%d: Bad tag unit at %s",
                             file, lineno, path);
                    break;
                }
                tp->tags[tp->num_tags++] = tagunit;
            }

            if (argc > 2 && !strcmp(argv[2], "variant"))
                se->variantRequest=
                    read_variant(argc-3, argv+3, nmem, file, lineno);
        }
        else
            yaz_log(YLOG_WARN, "%s:%d: Unknown directive '%s'",
                    file, lineno, argv[0]);
    fclose (f);
    return res;
}
/*
 * Local variables:
 * c-basic-offset: 4
 * c-file-style: "Stroustrup"
 * indent-tabs-mode: nil
 * End:
 * vim: shiftwidth=4 tabstop=8 expandtab
 */

