core/loop.h

/*
 *  Copyright (c) 2004-2005    National Research Council
 *
 *  All rights reserved.
 *
 *  This material is confidential and proprietary information of
 *  National Research Council Canada ("Confidential Information").
 *  This Confidential Information may only be used and reproduced
 *  in accordance with the terms of the license agreement.
 *
 */

#ifndef __INCLUDED_SCOPIRA_CORE_LOOP_H__
#define __INCLUDED_SCOPIRA_CORE_LOOP_H__

#include <string>
#include <vector>
#include <map>

#include <scopira/tool/export.h>
#include <scopira/tool/output.h>
#include <scopira/tool/fileflow.h>
#include <scopira/tool/netflow.h>
#include <scopira/tool/objflowloader.h>

namespace scopira
{
  namespace tool
  {
    class dll; //fwd
  }
  namespace core
  {
    class basic_loop;

    class objecttype; //fwd
  }
}

class scopira::core::basic_loop : public virtual scopira::tool::object
{
  public:
    static basic_loop * instance(void) { return dm_instance; }

    SCOPIRA_EXPORT basic_loop(int &argc, char **&argv);

    SCOPIRA_EXPORT ~basic_loop();

    SCOPIRA_EXPORT void list_config(std::vector<std::string> &out) const;

    SCOPIRA_EXPORT bool has_config(const std::string &key) const;

    SCOPIRA_EXPORT const std::string & get_config(const std::string &key) const;

    SCOPIRA_EXPORT bool get_config(const std::string &key, std::string &out) const;

    SCOPIRA_EXPORT void set_config(const std::string &key, const std::string &val);

    SCOPIRA_EXPORT void set_config_default(const std::string &key, const std::string &val);

    SCOPIRA_EXPORT void set_config_save(const std::string &key, const std::string &val);


    SCOPIRA_EXPORT void commit_config(void);

    SCOPIRA_EXPORT static std::string config_expand(const std::string &s);

    SCOPIRA_EXPORT objecttype * get_root_objecttype(void);

    const std::string &get_context_name(void) const { return dm_appctx; }

    const std::string &get_config_dir(void) const { return dm_configdir; }

    SCOPIRA_EXPORT bool save_object(const std::string &name, scopira::tool::object *o);

    SCOPIRA_EXPORT bool load_object(const std::string &name, scopira::tool::count_ptr<scopira::tool::object> &out);

  private:
    SCOPIRA_EXPORT static basic_loop *dm_instance;

#ifndef NDEBUG
    // used to balance reference counts in debug mode only
    scopira::tool::objrefcounter dm_refcounter;
#endif
    scopira::tool::fileflow dm_default_out;
    scopira::tool::output_stack dm_outputstack;
    scopira::tool::count_ptr<scopira::tool::objflowloader> dm_objloader;

    struct value_t
    {
      std::string pm_val;                 // the current value
      bool pm_disk;                       // is thie value marked for saving
      int pm_creatorid;                  // the stack level that this value was created at (used in config stacks)

      value_t(void) : pm_disk(false), pm_creatorid(0) { }
      value_t(const std::string &val, int creatorid) : pm_val(val), pm_disk(false),
        pm_creatorid(creatorid) { }
    };

    typedef std::map<std::string, value_t> configmap_t;
    configmap_t dm_configmap;

    // the current creator id to assign to new values
    // its incremented between each "phase" or level or context of parsing
    struct parse_context_t {
      int pm_creatorid;
      bool pm_disk;       // mark new nodes for writing (sets the value_t->pm_disk)
    };
    
    parse_context_t dm_parse_context;

    std::string dm_emptrystring;
    std::string dm_appctx;
    std::string dm_configdir;

    scopira::tool::count_ptr<objecttype> dm_root_objecttype;

    std::vector<scopira::tool::count_ptr< scopira::tool::dll > > dm_dynlibs;
    
#ifdef PLATFORM_win32
    scopira::tool::net_loop dm_netlooper;
#endif

  private:
    void parse_config(int argc, char **argv);
    void parse_config_file(const std::string &filename, bool cryonerrors);
    void parse_config_string(const std::string &s, bool newlinesonly);
    void parse_config_pair(const std::string &s);
    void process_config(void);
    void process_lib(const std::string &libname);
    void process_libdir(const std::string &dirname);
};

#endif