strgen.cpp

Go to the documentation of this file.

/*
 * This file is part of OpenTTD.
 * OpenTTD is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, version 2.
 * OpenTTD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with OpenTTD. If not, see <http://www.gnu.org/licenses/>.
 */
 
#include "../stdafx.h"
#include "../core/endian_func.hpp"
#include "../core/mem_func.hpp"
#include "../error_func.h"
#include "../string_func.h"
#include "../strings_type.h"
#include "../misc/getoptdata.h"
#include "../table/control_codes.h"
#include "../3rdparty/fmt/std.h"
 
#include "strgen.h"
 
#include <filesystem>
#include <fstream>
 
#include "../table/strgen_tables.h"
 
#include "../safeguards.h"
 
 
#ifdef _MSC_VER
# define LINE_NUM_FMT(s) "{} ({}): warning: {} (" s ")\n"
#else
# define LINE_NUM_FMT(s) "{}:{}: " s ": {}\n"
#endif
 
void StrgenWarningI(const std::string &msg)
{
  if (_show_todo > 0) {
    fmt::print(stderr, LINE_NUM_FMT("warning"), _file, _cur_line, msg);
  } else {
    fmt::print(stderr, LINE_NUM_FMT("info"), _file, _cur_line, msg);
  }
  _warnings++;
}
 
void StrgenErrorI(const std::string &msg)
{
  fmt::print(stderr, LINE_NUM_FMT("error"), _file, _cur_line, msg);
  _errors++;
}
 
[[noreturn]] void StrgenFatalI(const std::string &msg)
{
  fmt::print(stderr, LINE_NUM_FMT("FATAL"), _file, _cur_line, msg);
#ifdef _MSC_VER
  fmt::print(stderr, LINE_NUM_FMT("warning"), _file, _cur_line, "language is not compiled");
#endif
  throw std::exception();
}
 
[[noreturn]] void FatalErrorI(const std::string &msg)
{
  fmt::print(stderr, LINE_NUM_FMT("FATAL"), _file, _cur_line, msg);
#ifdef _MSC_VER
  fmt::print(stderr, LINE_NUM_FMT("warning"), _file, _cur_line, "language is not compiled");
#endif
  exit(2);
}
 
struct FileStringReader : StringReader {
  std::ifstream input_stream;
 
  FileStringReader(StringData &data, const std::filesystem::path &file, bool master, bool translation) :
      StringReader(data, file.generic_string(), master, translation)
  {
    this->input_stream.open(file, std::ifstream::binary);
  }
 
  std::optional<std::string> ReadLine() override
  {
    std::string result;
    if (!std::getline(this->input_stream, result)) return std::nullopt;
    return result;
  }
 
  void HandlePragma(char *str) override;
 
  void ParseFile() override
  {
    this->StringReader::ParseFile();
 
    if (StrEmpty(_lang.name) || StrEmpty(_lang.own_name) || StrEmpty(_lang.isocode)) {
      FatalError("Language must include ##name, ##ownname and ##isocode");
    }
  }
};
 
void FileStringReader::HandlePragma(char *str)
{
  if (!memcmp(str, "id ", 3)) {
    this->data.next_string_id = std::strtoul(str + 3, nullptr, 0);
  } else if (!memcmp(str, "name ", 5)) {
    strecpy(_lang.name, str + 5, lastof(_lang.name));
  } else if (!memcmp(str, "ownname ", 8)) {
    strecpy(_lang.own_name, str + 8, lastof(_lang.own_name));
  } else if (!memcmp(str, "isocode ", 8)) {
    strecpy(_lang.isocode, str + 8, lastof(_lang.isocode));
  } else if (!memcmp(str, "textdir ", 8)) {
    if (!memcmp(str + 8, "ltr", 3)) {
      _lang.text_dir = TD_LTR;
    } else if (!memcmp(str + 8, "rtl", 3)) {
      _lang.text_dir = TD_RTL;
    } else {
      FatalError("Invalid textdir {}", str + 8);
    }
  } else if (!memcmp(str, "digitsep ", 9)) {
    str += 9;
    strecpy(_lang.digit_group_separator, strcmp(str, "{NBSP}") == 0 ? NBSP : str, lastof(_lang.digit_group_separator));
  } else if (!memcmp(str, "digitsepcur ", 12)) {
    str += 12;
    strecpy(_lang.digit_group_separator_currency, strcmp(str, "{NBSP}") == 0 ? NBSP : str, lastof(_lang.digit_group_separator_currency));
  } else if (!memcmp(str, "decimalsep ", 11)) {
    str += 11;
    strecpy(_lang.digit_decimal_separator, strcmp(str, "{NBSP}") == 0 ? NBSP : str, lastof(_lang.digit_decimal_separator));
  } else if (!memcmp(str, "winlangid ", 10)) {
    const char *buf = str + 10;
    long langid = std::strtol(buf, nullptr, 16);
    if (langid > (long)UINT16_MAX || langid < 0) {
      FatalError("Invalid winlangid {}", buf);
    }
    _lang.winlangid = (uint16_t)langid;
  } else if (!memcmp(str, "grflangid ", 10)) {
    const char *buf = str + 10;
    long langid = std::strtol(buf, nullptr, 16);
    if (langid >= 0x7F || langid < 0) {
      FatalError("Invalid grflangid {}", buf);
    }
    _lang.newgrflangid = (uint8_t)langid;
  } else if (!memcmp(str, "gender ", 7)) {
    if (this->master) FatalError("Genders are not allowed in the base translation.");
    char *buf = str + 7;
 
    for (;;) {
      const char *s = ParseWord(&buf);
 
      if (s == nullptr) break;
      if (_lang.num_genders >= MAX_NUM_GENDERS) FatalError("Too many genders, max {}", MAX_NUM_GENDERS);
      strecpy(_lang.genders[_lang.num_genders], s, lastof(_lang.genders[_lang.num_genders]));
      _lang.num_genders++;
    }
  } else if (!memcmp(str, "case ", 5)) {
    if (this->master) FatalError("Cases are not allowed in the base translation.");
    char *buf = str + 5;
 
    for (;;) {
      const char *s = ParseWord(&buf);
 
      if (s == nullptr) break;
      if (_lang.num_cases >= MAX_NUM_CASES) FatalError("Too many cases, max {}", MAX_NUM_CASES);
      strecpy(_lang.cases[_lang.num_cases], s, lastof(_lang.cases[_lang.num_cases]));
      _lang.num_cases++;
    }
  } else {
    StringReader::HandlePragma(str);
  }
}
 
bool CompareFiles(const std::filesystem::path &path1, const std::filesystem::path &path2)
{
  /* Check for equal size, but ignore the error code for cases when a file does not exist. */
  std::error_code error_code;
  if (std::filesystem::file_size(path1, error_code) != std::filesystem::file_size(path2, error_code)) return false;
 
  std::ifstream stream1(path1, std::ifstream::binary);
  std::ifstream stream2(path2, std::ifstream::binary);
 
  return std::equal(std::istreambuf_iterator<char>(stream1.rdbuf()),
      std::istreambuf_iterator<char>(),
      std::istreambuf_iterator<char>(stream2.rdbuf()));
}
 
struct FileWriter {
  std::ofstream output_stream; 
  const std::filesystem::path path; 
 
  FileWriter(const std::filesystem::path &path, std::ios_base::openmode openmode) : path(path)
  {
    this->output_stream.open(path, openmode);
  }
 
  void Finalise()
  {
    this->output_stream.close();
  }
 
  virtual ~FileWriter()
  {
    /* If we weren't closed an exception was thrown, so remove the temporary file. */
    if (this->output_stream.is_open()) {
      this->output_stream.close();
      std::filesystem::remove(this->path);
    }
  }
};
 
struct HeaderFileWriter : HeaderWriter, FileWriter {
  const std::filesystem::path real_path;
  int prev;
  uint total_strings;
 
  HeaderFileWriter(const std::filesystem::path &path) : FileWriter("tmp.xxx", std::ofstream::out),
    real_path(path), prev(0), total_strings(0)
  {
    this->output_stream << "/* This file is automatically generated. Do not modify */\n\n";
    this->output_stream << "#ifndef TABLE_STRINGS_H\n";
    this->output_stream << "#define TABLE_STRINGS_H\n";
  }
 
  void WriteStringID(const std::string &name, int stringid) override
  {
    if (prev + 1 != stringid) this->output_stream << "\n";
    fmt::print(this->output_stream, "static const StringID {} = 0x{:X};\n", name, stringid);
    prev = stringid;
    total_strings++;
  }
 
  void Finalise(const StringData &data) override
  {
    /* Find the plural form with the most amount of cases. */
    int max_plural_forms = 0;
    for (uint i = 0; i < lengthof(_plural_forms); i++) {
      max_plural_forms = std::max(max_plural_forms, _plural_forms[i].plural_count);
    }
 
    fmt::print(this->output_stream,
      "\n"
      "static const uint LANGUAGE_PACK_VERSION     = 0x{:X};\n"
      "static const uint LANGUAGE_MAX_PLURAL       = {};\n"
      "static const uint LANGUAGE_MAX_PLURAL_FORMS = {};\n"
      "static const uint LANGUAGE_TOTAL_STRINGS    = {};\n"
      "\n",
      data.Version(), lengthof(_plural_forms), max_plural_forms, total_strings
    );
 
    this->output_stream << "#endif /* TABLE_STRINGS_H */\n";
 
    this->FileWriter::Finalise();
 
    std::error_code error_code;
    if (CompareFiles(this->path, this->real_path)) {
      /* files are equal. tmp.xxx is not needed */
      std::filesystem::remove(this->path, error_code); // Just ignore the error
    } else {
      /* else rename tmp.xxx into filename */
# if defined(_WIN32)
      std::filesystem::remove(this->real_path, error_code); // Just ignore the error, file probably doesn't exist
# endif
      std::filesystem::rename(this->path, this->real_path, error_code);
      if (error_code) FatalError("rename({}, {}) failed: {}", this->path, this->real_path, error_code.message());
    }
  }
};
 
struct LanguageFileWriter : LanguageWriter, FileWriter {
  LanguageFileWriter(const std::filesystem::path &path) : FileWriter(path, std::ofstream::binary | std::ofstream::out)
  {
  }
 
  void WriteHeader(const LanguagePackHeader *header) override
  {
    this->Write((const byte *)header, sizeof(*header));
  }
 
  void Finalise() override
  {
    this->output_stream.put(0);
    this->FileWriter::Finalise();
  }
 
  void Write(const byte *buffer, size_t length) override
  {
    this->output_stream.write((const char *)buffer, length);
  }
};
 
static const OptionData _opts[] = {
  GETOPT_GENERAL('C', '\0', "-export-commands", ODF_NO_VALUE),
  GETOPT_GENERAL('L', '\0', "-export-plurals",  ODF_NO_VALUE),
  GETOPT_GENERAL('P', '\0', "-export-pragmas",  ODF_NO_VALUE),
    GETOPT_NOVAL(     't',  "--todo"),
    GETOPT_NOVAL(     'w',  "--warning"),
    GETOPT_NOVAL(     'h',  "--help"),
  GETOPT_GENERAL('h', '?',  nullptr,            ODF_NO_VALUE),
    GETOPT_VALUE(     's',  "--source_dir"),
    GETOPT_VALUE(     'd',  "--dest_dir"),
  GETOPT_END(),
};
 
int CDECL main(int argc, char *argv[])
{
  std::filesystem::path src_dir(".");
  std::filesystem::path dest_dir;
 
  GetOptData mgo(argc - 1, argv + 1, _opts);
  for (;;) {
    int i = mgo.GetOpt();
    if (i == -1) break;
 
    switch (i) {
      case 'C':
        fmt::print("args\tflags\tcommand\treplacement\n");
        for (const CmdStruct *cs = _cmd_structs; cs < endof(_cmd_structs); cs++) {
          char flags;
          if (cs->proc == EmitGender) {
            flags = 'g'; // Command needs number of parameters defined by number of genders
          } else if (cs->proc == EmitPlural) {
            flags = 'p'; // Command needs number of parameters defined by plural value
          } else if (cs->flags & C_DONTCOUNT) {
            flags = 'i'; // Command may be in the translation when it is not in base
          } else {
            flags = '0'; // Command needs no parameters
          }
          fmt::print("{}\t{:c}\t\"{}\"\t\"{}\"\n", cs->consumes, flags, cs->cmd, strstr(cs->cmd, "STRING") ? "STRING" : cs->cmd);
        }
        return 0;
 
      case 'L':
        fmt::print("count\tdescription\tnames\n");
        for (const PluralForm *pf = _plural_forms; pf < endof(_plural_forms); pf++) {
          fmt::print("{}\t\"{}\"\t{}\n", pf->plural_count, pf->description, pf->names);
        }
        return 0;
 
      case 'P':
        fmt::print("name\tflags\tdefault\tdescription\n");
        for (size_t j = 0; j < lengthof(_pragmas); j++) {
          fmt::print("\"{}\"\t{}\t\"{}\"\t\"{}\"\n",
              _pragmas[j][0], _pragmas[j][1], _pragmas[j][2], _pragmas[j][3]);
        }
        return 0;
 
      case 't':
        _show_todo |= 1;
        break;
 
      case 'w':
        _show_todo |= 2;
        break;
 
      case 'h':
        fmt::print(
          "strgen\n"
          " -t | --todo       replace any untranslated strings with '<TODO>'\n"
          " -w | --warning    print a warning for any untranslated strings\n"
          " -h | -? | --help  print this help message and exit\n"
          " -s | --source_dir search for english.txt in the specified directory\n"
          " -d | --dest_dir   put output file in the specified directory, create if needed\n"
          " -export-commands  export all commands and exit\n"
          " -export-plurals   export all plural forms and exit\n"
          " -export-pragmas   export all pragmas and exit\n"
          " Run without parameters and strgen will search for english.txt and parse it,\n"
          " creating strings.h. Passing an argument, strgen will translate that language\n"
          " file using english.txt as a reference and output <language>.lng.\n"
        );
        return 0;
 
      case 's':
        src_dir = mgo.opt;
        break;
 
      case 'd':
        dest_dir = mgo.opt;
        break;
 
      case -2:
        fmt::print(stderr, "Invalid arguments\n");
        return 0;
    }
  }
 
  if (dest_dir.empty()) dest_dir = src_dir; // if dest_dir is not specified, it equals src_dir
 
  try {
    /* strgen has two modes of operation. If no (free) arguments are passed
     * strgen generates strings.h to the destination directory. If it is supplied
     * with a (free) parameter the program will translate that language to destination
     * directory. As input english.txt is parsed from the source directory */
    if (mgo.numleft == 0) {
      std::filesystem::path input_path = src_dir;
      input_path /= "english.txt";
 
      /* parse master file */
      StringData data(TEXT_TAB_END);
      FileStringReader master_reader(data, input_path, true, false);
      master_reader.ParseFile();
      if (_errors != 0) return 1;
 
      /* write strings.h */
      std::filesystem::path output_path = dest_dir;
      std::filesystem::create_directories(dest_dir);
      output_path /= "strings.h";
 
      HeaderFileWriter writer(output_path);
      writer.WriteHeader(data);
      writer.Finalise(data);
      if (_errors != 0) return 1;
    } else if (mgo.numleft >= 1) {
      std::filesystem::path input_path = src_dir;
      input_path /= "english.txt";
 
      StringData data(TEXT_TAB_END);
      /* parse master file and check if target file is correct */
      FileStringReader master_reader(data, input_path, true, false);
      master_reader.ParseFile();
 
      for (int i = 0; i < mgo.numleft; i++) {
        data.FreeTranslation();
 
        std::filesystem::path lang_file = mgo.argv[i];
        FileStringReader translation_reader(data, lang_file, false, lang_file.filename() != "english.txt");
        translation_reader.ParseFile(); // target file
        if (_errors != 0) return 1;
 
        /* get the targetfile, strip any directories and append to destination path */
        std::filesystem::path output_file = dest_dir;
        output_file /= lang_file.filename();
        output_file.replace_extension("lng");
 
        LanguageFileWriter writer(output_file);
        writer.WriteLang(data);
        writer.Finalise();
 
        /* if showing warnings, print a summary of the language */
        if ((_show_todo & 2) != 0) {
          fmt::print("{} warnings and {} errors for {}\n", _warnings, _errors, output_file);
        }
      }
    }
  } catch (...) {
    return 2;
  }
 
  return 0;
}