aboutsummaryrefslogtreecommitdiff
path: root/backend/tools/DescriptionGenerator/DescriptionGeneratorMain.cpp
blob: 11890f6e425c5343c9339a1b0f08e268280b3580 (plain)
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
// SPDX-License-Identifier: GPL-3.0-or-later

#include "TargetDescriptionGenerator.h"
#include "rcc/Support/TextDiagnostic.h"

#include <filesystem>
#include <fstream>
#include <iostream>
#include <iterator>
#include <string>
#include <string_view>

namespace {

struct CommandLine final {
  std::string InputPath;
  std::string HeaderPath;
  std::string TablesPath;
};

[[nodiscard]] bool parseCommandLine(int ArgumentCount, char *Arguments[], CommandLine &Result) {
  for (int Index = 1; Index < ArgumentCount; ++Index) {
    const std::string_view Argument(Arguments[Index]);
    if (Argument == "--input" || Argument == "--header" || Argument == "--tables") {
      if (Index + 1 >= ArgumentCount) {
        std::cerr << "rcc-backend-descgen: missing value for " << Argument << '\n';
        return false;
      }
      std::string *Destination = nullptr;
      if (Argument == "--input")
        Destination = &Result.InputPath;
      else if (Argument == "--header")
        Destination = &Result.HeaderPath;
      else
        Destination = &Result.TablesPath;
      *Destination = Arguments[++Index];
      continue;
    }
    std::cerr << "rcc-backend-descgen: unknown argument '" << Argument << "'\n";
    return false;
  }
  if (!Result.InputPath.empty() && !Result.HeaderPath.empty() && !Result.TablesPath.empty())
    return true;
  std::cerr << "usage: rcc-backend-descgen --input <file> --header <file> --tables <file>\n";
  return false;
}

[[nodiscard]] bool writeFile(const std::filesystem::path &Path, std::string_view Contents) {
  std::error_code Error;
  std::filesystem::create_directories(Path.parent_path(), Error);
  if (Error) {
    std::cerr << "rcc-backend-descgen: cannot create output directory for '" << Path.string()
              << "': " << Error.message() << '\n';
    return false;
  }
  std::ofstream Output(Path, std::ios::binary | std::ios::trunc);
  if (!Output) {
    std::cerr << "rcc-backend-descgen: cannot open output '" << Path.string() << "'\n";
    return false;
  }
  Output.write(Contents.data(), static_cast<std::streamsize>(Contents.size()));
  if (Output) {
    Output.close();
    return true;
  }
  std::cerr << "rcc-backend-descgen: cannot write output '" << Path.string() << "'\n";
  return false;
}

} // namespace

int main(int ArgumentCount, char *Arguments[]) {
  CommandLine Command;
  if (!parseCommandLine(ArgumentCount, Arguments, Command))
    return 1;

  std::ifstream Input(Command.InputPath, std::ios::binary);
  if (!Input) {
    std::cerr << "rcc-backend-descgen: cannot open input '" << Command.InputPath << "'\n";
    return 1;
  }
  const std::string SourceText{std::istreambuf_iterator<char>(Input),
                               std::istreambuf_iterator<char>()};
  const rcc::targetgen::DescriptionGenerationResult Result =
      rcc::targetgen::generateDescription(Command.InputPath, SourceText);
  if (!Result.succeeded()) {
    const rcc::TextSource Source(Command.InputPath, SourceText);
    for (const rcc::TextDiagnostic &Diagnostic : Result.Diagnostics)
      std::cerr << rcc::formatTextDiagnostic(Diagnostic, Source) << '\n';
    return 1;
  }

  return writeFile(Command.HeaderPath, Result.Output->RegisterHeader) &&
                 writeFile(Command.TablesPath, Result.Output->TargetTables)
             ? 0
             : 1;
}