view/common/FileToArgs.cpp

/*!
  \file
  \brief ファイルから引数情報を取得する

  \author Satofumi KAMIMURA

  $Id$

  \todo xpressive に置き換える
*/

#if (!HAVE_CONFIG_H || defined(HAVE_BOOST_REGEX)) &&    \
  !defined(FORCE_NO_BOOST_REGEX)
#define USE_BOOST_REGEX
#endif

#include "FileToArgs.h"
#include <string>
#include <fstream>
#include <vector>
#include <map>
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#ifdef USE_BOOST_REGEX
#include <boost/regex.hpp>
#endif

using namespace beego;


struct FileToArgs::pImpl {

  std::vector<std::string> args;
  std::vector<char*> argv_ptr;

  typedef std::map<std::string,size_t> TagMap;
  TagMap tag_map;

  bool parseLine(std::string& line, const char* programName) {

    if (line.empty()) {
      return false;
    }

    if (line[0] == '#') {
      return false;
    }

    // !!! strtok() は置き換えるべきかな？
    // !!! Boost の何かを使う？
    const char* token = strtok(const_cast<char*>(line.c_str()), " \t\r\n");
    if (token == NULL) {
      return false;
    }
    const char* next_token = strtok(NULL, " \t\r\n");
    if (next_token == NULL) {
      line = token;
      return true;
    }

    // トークンが２つ以上あった場合の処理
#ifdef USE_BOOST_REGEX
    boost::regex pattern(token);
    if (! boost::regex_match(programName, pattern)) {
      return false;
    }
#else
    if (strcmp(token, programName)) {
      return false;
    }
#endif
    line = next_token;
    return true;
  }
};


FileToArgs::FileToArgs(void) : pimpl(new pImpl), argc(0) {
}


FileToArgs::~FileToArgs(void) {
}


bool FileToArgs::load(const char* fname, const char* programName) {
  pimpl->args.clear();
  argc = 0;

  // 最初にプログラム名を格納
  pimpl->args.push_back(programName);

  // ファイルを開く
  std::ifstream fin(fname);
  if (! fin.is_open()) {
    return false;
  }

  while (! fin.eof()) {

    // 行毎の情報を引数として扱う
    std::string line;
    getline(fin, line);

    if (! pimpl->parseLine(line, programName)) {
      continue;
    }

    // '=' までのタグが重複したら、後からの内容で上書きする
    size_t tag_size = line.find('=');
    if (tag_size != std::string::npos) {
      std::string tag = line.substr(0, tag_size);
      pImpl::TagMap::iterator p = pimpl->tag_map.find(tag);
      if (p != pimpl->tag_map.end()) {
        size_t find_index = p->second;
        pimpl->args[find_index] = line;
        continue;
      } else {
        size_t insert_index = pimpl->args.size();
        pimpl->tag_map.
          insert(std::pair<std::string,size_t>(tag, insert_index));
      }
    }

    pimpl->args.push_back(line);
  }

  // argc, argv の生成
  for (std::vector<std::string>::iterator it = pimpl->args.begin();
       it != pimpl->args.end(); ++it) {
    pimpl->argv_ptr.push_back(&(*it)[0]);
  }
  argc = static_cast<int>(pimpl->args.size());
  argv = &(pimpl->argv_ptr)[0];

  return true;
}