/**
 * @file llxuiparser.h
 * @brief Utility functions for handling XUI structures in XML
 *
 * $LicenseInfo:firstyear=2003&license=viewerlgpl$
 * Second Life Viewer Source Code
 * Copyright (C) 2010, Linden Research, Inc.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation;
 * version 2.1 of the License only.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 *
 * Linden Research, Inc., 945 Battery Street, San Francisco, CA  94111  USA
 * $/LicenseInfo$
 */

#ifndef LLXUIPARSER_H
#define LLXUIPARSER_H

#include "llinitparam.h"
#include "llregistry.h"
#include "llxmlnode.h"

#include <boost/function.hpp>
#include <iosfwd>
#include <stack>
#include <set>

class LLView;

// lookup widget type by name
class LLWidgetTypeRegistry
:   public LLRegistrySingleton<std::string, const std::type_info*, LLWidgetTypeRegistry>
{
    LLSINGLETON_EMPTY_CTOR(LLWidgetTypeRegistry);
};


// global static instance for registering all widget types
typedef boost::function<LLView* (LLXMLNodePtr node, LLView *parent, LLXMLNodePtr output_node)> LLWidgetCreatorFunc;

typedef LLRegistry<std::string, LLWidgetCreatorFunc> widget_registry_t;

class LLChildRegistryRegistry
: public LLRegistrySingleton<const std::type_info*, widget_registry_t, LLChildRegistryRegistry>
{
    LLSINGLETON_EMPTY_CTOR(LLChildRegistryRegistry);
};

class LLXSDWriter : public LLInitParam::Parser
{
    LOG_CLASS(LLXSDWriter);
public:
    void writeXSD(const std::string& name, LLXMLNodePtr node, const LLInitParam::BaseBlock& block, const std::string& xml_namespace);

    /*virtual*/ std::string getCurrentElementName() { return LLStringUtil::null; }
    /*virtual*/ std::string getCurrentFileName() { return LLStringUtil::null; }
    LLXSDWriter();
    ~LLXSDWriter();

protected:
    void writeAttribute(const std::string& type, const Parser::name_stack_t&, S32 min_count, S32 max_count, const std::vector<std::string>* possible_values);
    void addAttributeToSchema(LLXMLNodePtr nodep, const std::string& attribute_name, const std::string& type, bool mandatory, const std::vector<std::string>* possible_values);
    LLXMLNodePtr mAttributeNode;
    LLXMLNodePtr mElementNode;
    LLXMLNodePtr mSchemaNode;

    typedef std::set<std::string> string_set_t;
    typedef std::map<LLXMLNodePtr, string_set_t> attributes_map_t;
    attributes_map_t    mAttributesWritten;
};



// NOTE: DOES NOT WORK YET
// should support child widgets for XUI
class LLXUIXSDWriter : public LLXSDWriter
{
public:
    void writeXSD(const std::string& name, const std::string& path, const LLInitParam::BaseBlock& block);
};


class LLXUIParserImpl;

class LLXUIParser : public LLInitParam::Parser
{
LOG_CLASS(LLXUIParser);

public:
    LLXUIParser();
    typedef LLInitParam::Parser::name_stack_t name_stack_t;

    /*virtual*/ std::string getCurrentElementName();
    /*virtual*/ std::string getCurrentFileName() { return mCurFileName; }
    /*virtual*/ void parserWarning(const std::string& message);
    /*virtual*/ void parserError(const std::string& message);

    void readXUI(LLXMLNodePtr node, LLInitParam::BaseBlock& block, const std::string& filename = LLStringUtil::null, bool silent=false);
    template<typename BLOCK>
    void writeXUI(LLXMLNodePtr node,
                const BLOCK& block,
                const LLInitParam::predicate_rule_t rules = LLInitParam::default_parse_rules(),
                const LLInitParam::BaseBlock* diff_block = NULL)
    {
        if (!diff_block
            && !rules.isAmbivalent(LLInitParam::HAS_DEFAULT_VALUE))
        {
            diff_block = &LLInitParam::defaultValue<BLOCK>();
        }
        writeXUIImpl(node, block, rules, diff_block);
    }

private:
    LLXUIParser(const LLXUIParser& other); // no-copy
    void writeXUIImpl(LLXMLNodePtr node,
        const LLInitParam::BaseBlock& block,
        const LLInitParam::predicate_rule_t rules,
        const LLInitParam::BaseBlock* diff_block);
    bool readXUIImpl(LLXMLNodePtr node, LLInitParam::BaseBlock& block);
    bool readAttributes(LLXMLNodePtr nodep, LLInitParam::BaseBlock& block);

    //reader helper functions
    static bool readFlag(Parser& parser, void* val_ptr);
    static bool readBoolValue(Parser& parser, void* val_ptr);
    static bool readStringValue(Parser& parser, void* val_ptr);
    static bool readU8Value(Parser& parser, void* val_ptr);
    static bool readS8Value(Parser& parser, void* val_ptr);
    static bool readU16Value(Parser& parser, void* val_ptr);
    static bool readS16Value(Parser& parser, void* val_ptr);
    static bool readU32Value(Parser& parser, void* val_ptr);
    static bool readS32Value(Parser& parser, void* val_ptr);
    static bool readF32Value(Parser& parser, void* val_ptr);
    static bool readF64Value(Parser& parser, void* val_ptr);
    static bool readVector3Value(Parser& parser, void* val_ptr);
    static bool readColor4Value(Parser& parser, void* val_ptr);
    static bool readUIColorValue(Parser& parser, void* val_ptr);
    static bool readUUIDValue(Parser& parser, void* val_ptr);
    static bool readSDValue(Parser& parser, void* val_ptr);

    //writer helper functions
    static bool writeFlag(Parser& parser, const void* val_ptr, name_stack_t&);
    static bool writeBoolValue(Parser& parser, const void* val_ptr, name_stack_t&);
    static bool writeStringValue(Parser& parser, const void* val_ptr, name_stack_t&);
    static bool writeU8Value(Parser& parser, const void* val_ptr, name_stack_t&);
    static bool writeS8Value(Parser& parser, const void* val_ptr, name_stack_t&);
    static bool writeU16Value(Parser& parser, const void* val_ptr, name_stack_t&);
    static bool writeS16Value(Parser& parser, const void* val_ptr, name_stack_t&);
    static bool writeU32Value(Parser& parser, const void* val_ptr, name_stack_t&);
    static bool writeS32Value(Parser& parser, const void* val_ptr, name_stack_t&);
    static bool writeF32Value(Parser& parser, const void* val_ptr, name_stack_t&);
    static bool writeF64Value(Parser& parser, const void* val_ptr, name_stack_t&);
    static bool writeVector3Value(Parser& parser, const void* val_ptr, name_stack_t&);
    static bool writeColor4Value(Parser& parser, const void* val_ptr, name_stack_t&);
    static bool writeUIColorValue(Parser& parser, const void* val_ptr, name_stack_t&);
    static bool writeUUIDValue(Parser& parser, const void* val_ptr, name_stack_t&);
    static bool writeSDValue(Parser& parser, const void* val_ptr, name_stack_t&);

    LLXMLNodePtr getNode(name_stack_t& stack);

private:
    Parser::name_stack_t            mNameStack;
    LLXMLNodePtr                    mCurReadNode;
    // Root of the widget XML sub-tree, for example, "line_editor"
    LLXMLNodePtr                    mWriteRootNode;

    typedef std::map<std::string, LLXMLNodePtr> out_nodes_t;
    out_nodes_t                     mOutNodes;
    LLXMLNodePtr                    mLastWrittenChild;
    S32                             mCurReadDepth;
    std::string                     mCurFileName;
    std::string                     mRootNodeName;
};

// LLSimpleXUIParser is a streamlined SAX-based XUI parser that does not support localization
// or parsing of a tree of independent param blocks, such as child widgets.
// Use this for reading non-localized files that only need a single param block as a result.
//
// NOTE: In order to support nested block parsing, we need callbacks for start element that
// push new blocks contexts on the mScope stack.
// NOTE: To support localization without building a DOM, we need to enforce consistent
// ordering of child elements from base file to localized diff file.  Then we can use a pair
// of coroutines to perform matching of xml nodes during parsing.  Not sure if the overhead
// of coroutines would offset the gain from SAX parsing
class LLSimpleXUIParserImpl;

class LLSimpleXUIParser : public LLInitParam::Parser
{
LOG_CLASS(LLSimpleXUIParser);
public:
    typedef LLInitParam::Parser::name_stack_t name_stack_t;
    typedef LLInitParam::BaseBlock* (*element_start_callback_t)(LLSimpleXUIParser&, const char* block_name);

    LLSimpleXUIParser(element_start_callback_t element_cb = NULL);
    virtual ~LLSimpleXUIParser();

    /*virtual*/ std::string getCurrentElementName();
    /*virtual*/ std::string getCurrentFileName() { return mCurFileName; }
    /*virtual*/ void parserWarning(const std::string& message);
    /*virtual*/ void parserError(const std::string& message);

    bool readXUI(const std::string& filename, LLInitParam::BaseBlock& block, bool silent=false);


private:
    //reader helper functions
    static bool readFlag(Parser&, void* val_ptr);
    static bool readBoolValue(Parser&, void* val_ptr);
    static bool readStringValue(Parser&, void* val_ptr);
    static bool readU8Value(Parser&, void* val_ptr);
    static bool readS8Value(Parser&, void* val_ptr);
    static bool readU16Value(Parser&, void* val_ptr);
    static bool readS16Value(Parser&, void* val_ptr);
    static bool readU32Value(Parser&, void* val_ptr);
    static bool readS32Value(Parser&, void* val_ptr);
    static bool readF32Value(Parser&, void* val_ptr);
    static bool readF64Value(Parser&, void* val_ptr);
    static bool readColor4Value(Parser&, void* val_ptr);
    static bool readUIColorValue(Parser&, void* val_ptr);
    static bool readUUIDValue(Parser&, void* val_ptr);
    static bool readSDValue(Parser&, void* val_ptr);

private:
    static void startElementHandler(void *userData, const char *name, const char **atts);
    static void endElementHandler(void *userData, const char *name);
    static void characterDataHandler(void *userData, const char *s, int len);

    void startElement(const char *name, const char **atts);
    void endElement(const char *name);
    void characterData(const char *s, int len);
    bool readAttributes(const char **atts);
    bool processText();

    Parser::name_stack_t            mNameStack;
    struct XML_ParserStruct*        mParser;
    LLXMLNodePtr                    mLastWrittenChild;
    S32                             mCurReadDepth;
    std::string                     mCurFileName;
    std::string                     mTextContents;
    const char*                     mCurAttributeValueBegin;
    std::vector<S32>                mTokenSizeStack;
    std::vector<std::string>        mScope;
    std::vector<bool>               mEmptyLeafNode;
    element_start_callback_t        mElementCB;

    std::vector<std::pair<LLInitParam::BaseBlock*, S32> > mOutputStack;
};


#endif //LLXUIPARSER_H