mirror of https://github.com/arendst/Tasmota.git
142 lines
3.2 KiB
C++
Executable File
142 lines
3.2 KiB
C++
Executable File
/********************************************************************
|
|
** Copyright (c) 2018-2020 Guan Wenliang
|
|
** This file is part of the Berry default interpreter.
|
|
** skiars@qq.com, https://github.com/Skiars/berry
|
|
** See Copyright Notice in the LICENSE file or at
|
|
** https://github.com/Skiars/berry/blob/master/LICENSE
|
|
********************************************************************/
|
|
#include "main.h"
|
|
#include "block_builder.h"
|
|
#include "coc_parser.h"
|
|
#include "macro_table.h"
|
|
#include "str_build.h"
|
|
#include <fstream>
|
|
#include <sstream>
|
|
|
|
void builder::parse_all(const std::string &filename)
|
|
{
|
|
size_t pos = filename.find_last_of(".");
|
|
std::string ext = pos < filename.size() ? filename.substr(pos) : "";
|
|
if (ext == ".c" || ext == ".cc" || ext == ".cpp") {
|
|
std::string text = readfile(filename);
|
|
coc_parser parser(text);
|
|
push_strtab(parser.strtab());
|
|
for (auto object : parser.objects()) {
|
|
block_builder builder(&object, m_macro);
|
|
push_strtab(builder.strtab());
|
|
builder.dumpfile(m_output);
|
|
}
|
|
}
|
|
}
|
|
|
|
void builder::push_strtab(const std::vector<std::string> &list)
|
|
{
|
|
for (auto s : list)
|
|
m_strmap[s] = 0;
|
|
}
|
|
|
|
std::string builder::readfile(const std::string &filename)
|
|
{
|
|
std::ifstream in(filename);
|
|
std::ostringstream tmp;
|
|
tmp << in.rdbuf();
|
|
return tmp.str();
|
|
}
|
|
|
|
#ifndef _MSC_VER
|
|
#include <dirent.h>
|
|
#include <unistd.h>
|
|
#include <sys/stat.h>
|
|
#else
|
|
#include <windows.h>
|
|
#include <direct.h>
|
|
#endif
|
|
|
|
#ifndef _MSC_VER
|
|
void builder::scandir(const std::string &srcpath)
|
|
{
|
|
DIR *dp;
|
|
struct dirent *ep;
|
|
dp = opendir(srcpath.data());
|
|
if (dp != NULL) {
|
|
while ((ep = readdir(dp)) != NULL) {
|
|
std::string fname(ep->d_name);
|
|
parse_all(srcpath + "/" + fname);
|
|
}
|
|
closedir(dp);
|
|
}
|
|
}
|
|
#else
|
|
void builder::scandir(const std::string &srcpath)
|
|
{
|
|
HANDLE find;
|
|
WIN32_FIND_DATA data;
|
|
find = FindFirstFile((srcpath + "/*").data(), &data);
|
|
if (find != INVALID_HANDLE_VALUE) {
|
|
do {
|
|
std::string fname(data.cFileName);
|
|
parse_all(srcpath + "/" + fname);
|
|
} while (FindNextFile(find, &data) != 0);
|
|
FindClose(find);
|
|
}
|
|
}
|
|
#endif
|
|
|
|
void builder::build()
|
|
{
|
|
for (auto it : m_input) {
|
|
scandir(it);
|
|
}
|
|
str_build sb(m_strmap);
|
|
sb.build(m_output);
|
|
}
|
|
|
|
builder::builder(int argc, char **argv)
|
|
{
|
|
m_state = Input;
|
|
for (int i = 1; i < argc; ++i) {
|
|
add_arg(argv[i]);
|
|
}
|
|
m_macro = new macro_table();
|
|
for (auto it : m_config) {
|
|
m_macro->scan_file(it);
|
|
}
|
|
}
|
|
|
|
builder::~builder()
|
|
{
|
|
delete m_macro;
|
|
}
|
|
|
|
void builder::add_arg(const std::string &arg)
|
|
{
|
|
if (arg == "-i") {
|
|
m_state = Input;
|
|
} else if (arg == "-o") {
|
|
m_state = Output;
|
|
} else if (arg == "-c") {
|
|
m_state = Config;
|
|
} else {
|
|
switch (m_state) {
|
|
case Output:
|
|
m_output = arg;
|
|
break;
|
|
case Config:
|
|
m_config.push_back(arg);
|
|
break;
|
|
case Input:
|
|
default:
|
|
m_input.push_back(arg);
|
|
break;
|
|
}
|
|
m_state = Input;
|
|
}
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
builder arg(argc, argv);
|
|
arg.build();
|
|
return 0;
|
|
}
|