aboutsummaryrefslogtreecommitdiff
path: root/src/headless/main.cpp
blob: c3ca8c2bb0ce2321c3d7188d150290fbd0615544 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
#include <algorithm>
#include <cctype>
#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <iterator>
#include <string>
#include <string_view>
#include <vector>

#include <openvic-dataloader/ParseWarning.hpp>
#include <openvic-dataloader/csv/LineObject.hpp>
#include <openvic-dataloader/csv/Parser.hpp>
#include <openvic-dataloader/v2script/AbstractSyntaxTree.hpp>
#include <openvic-dataloader/v2script/Parser.hpp>

std::string_view trim(std::string_view str) {
   std::string_view::iterator begin = str.begin();
   std::string_view::iterator end = str.end();
   for (;; begin++) {
      if (begin == end) return std::string_view();
      if (!std::isspace(*begin)) break;
   }
   end--;
   for (;; end--) {
      if (end == begin) return std::string_view();
      if (!std::isspace(*end)) break;
   }
   return std::string_view(&*begin, std::distance(begin, end));
}

bool insenitive_trim_eq(std::string_view lhs, std::string_view rhs) {
   lhs = trim(lhs);
   rhs = trim(rhs);
   return std::equal(
      lhs.begin(), lhs.end(),
      rhs.begin(), rhs.end(),
      [](char a, char b) { return std::tolower(a) == std::tolower(b); });
}

template<ovdl::csv::EncodingType Encoding>
int print_csv(const std::string_view path) {
   auto parser = ovdl::csv::Parser<Encoding>::from_file(path);
   if (parser.has_error()) {
      return 1;
   }

   parser.parse_csv();
   if (parser.has_error()) {
      return 2;
   }

   if (parser.has_warning()) {
      for (auto& warning : parser.get_warnings()) {
         std::cerr << "Warning: " << warning.message << std::endl;
      }
   }

   std::cout << "lines:\t\t" << parser.get_lines().size() << std::endl;
   for (const auto& line : parser.get_lines()) {
      std::cout << "line size:\t" << line.value_count() << std::endl;
      std::cout << "values:\t\t" << line << std::endl;
   }
   return EXIT_SUCCESS;
}

int print_v2script_simple(const std::string_view path) {
   auto parser = ovdl::v2script::Parser::from_file(path);
   if (parser.has_error()) {
      return 1;
   }

   parser.simple_parse();
   if (parser.has_error()) {
      return 2;
   }

   if (parser.has_warning()) {
      for (auto& warning : parser.get_warnings()) {
         std::cerr << "Warning: " << warning.message << std::endl;
      }
   }

   std::cout << parser.get_file_node() << std::endl;
   return EXIT_SUCCESS;
}

int main(int argc, char** argv) {
   switch (argc) {
      case 2:
         return print_v2script_simple(argv[1]);
      case 4:
         if (insenitive_trim_eq(argv[1], "csv") && insenitive_trim_eq(argv[2], "utf"))
            return print_csv<ovdl::csv::EncodingType::Utf8>(argv[3]);
         goto default_jump;
      case 3:
         if (insenitive_trim_eq(argv[1], "csv"))
            return print_csv<ovdl::csv::EncodingType::Windows1252>(argv[2]);
         [[fallthrough]];
      default:
      default_jump:
         std::fprintf(stderr, "usage: %s <filename>\n", argv[0]);
         std::fprintf(stderr, "usage: %s csv <filename>\n", argv[0]);
         std::fprintf(stderr, "usage: %s csv utf <filename>", argv[0]);
         return EXIT_FAILURE;
   }

   return 0;
}