Changeset 2814

Timestamp:

07/01/08 00:33:32 (6 months ago)

Author:

edmanm

Message:

Make the encoding used by ts2po configurable as well (defaults to utf-8
still), and set the PO header properly.

Files:

• vidalia/trunk/src/tools/ts2po/ts2po.cpp (modified) (10 diffs)

vidalia/trunk/src/tools/ts2po/ts2po.cpp

@@ -12 +12 @@
 #include <QDomDocument>
 #include <QTextStream>
+#include <QTextCodec>
 #include <QDateTime>
 
@@ -32 +33 @@
 }
 
-/** Return a header to be placed at the top of the .po file. */
+/** Return a header to be placed at the top of the .po file. The header will
+ * include <b>encoding</b> in the Content-Type header line. */
 QString
-create_po_header()
+create_po_header(const QString &encoding)
 {
   QString header;
@@ -49 +51 @@
   header.append("\"Language-Team: "TS2PO_LANGUAGE_TEAM"\\n\"\n");
   header.append("\"MIME-Version: 1.0\\n\"\n");
-  header.append("\"Content-Type: text/plain; charset=UTF-8\\n\"\n");
+  header.append("\"Content-Type: text/plain; ");
+  header.append(QString("charset=%1\\n\"\n").arg(encoding));
   header.append("\"Content-Transfer-Encoding: 8bit\\n\"\n");
   header.append("\"Plural-Forms: nplurals=INTEGER; plural=EXPRESSION;\\n\"\n");
@@ -112 +115 @@
 
 /** Convert the TS-formatted document in <b>ts</b> to a PO-formatted document.
- * The output will be written to <b>po</b>. Returns the number of strings
+ * The output will be written to <b>po</b>, including a file header that
+ * specifies <b>encoding</b> as the character set. Returns the number of strings
  * converted on success, or -1 on error and <b>errorMessage</b> will be set. */
 int
-ts2po(const QDomDocument *ts, QString *po, QString *errorMessage)
+ts2po(const QDomDocument *ts, QString *po, const QString &encoding,
+      QString *errorMessage)
 {
   int n_strings = 0;
@@ -130 +135 @@
 
   /* Start with the PO header */
-  *po = create_po_header();
+  *po = create_po_header(encoding);
 
   /* Iterate through all of the translation contexts and build up the PO file
@@ -158 +163 @@
 {
   QTextStream error(stderr);
-  error << "usage: ts2po [-q] -i <infile.ts> -o <outfile.po>\n";
+  error << "usage: ts2po [-q] -i <infile.ts> -o <outfile.po> "
+           "[-c <encoding>]\n";
   error << "  -q (optional)   Quiet mode (errors are still displayed)\n";
   error << "  -i <infile.ts>  Input .ts file\n";
   error << "  -o <outfile.po> Output .po file\n";
+  error << "  -c <encoding>   Text encoding (default: utf-8)\n";
   error.flush();
   exit(1);
@@ -172 +179 @@
   QString errorMessage;
   char *infile, *outfile;
+  QTextCodec *codec = QTextCodec::codecForName("utf-8");
   bool quiet = false;
   
   /* Check for the correct number of input parameters. */
-  if (argc < 5 || argc > 6)
+  if (argc < 5 || argc > 8)
     print_usage_and_exit();
   for (int i = 1; i < argc; i++) {
@@ -185 +193 @@
     else if (!arg.compare("-o", Qt::CaseInsensitive) && ++i < argc)
       outfile = argv[i];
-    else
+    else if (!arg.compare("-c", Qt::CaseInsensitive) && ++i < argc) {
+      codec = QTextCodec::codecForName(argv[i]);
+      if (!codec) {
+        error << "Invalid text encoding specified.\n";
+        return 1;
+      }
+    } else
       print_usage_and_exit(); 
   }
-
+ 
   /* Read and parse the input .ts file. */
   QDomDocument ts;
@@ -209 +223 @@
   /* Convert the input .ts file to a .po formatted file. */
   QString po;
-  int n_strings = ts2po(&ts, &po, &errorMessage);
+  int n_strings = ts2po(&ts, &po, QString(codec->name()), &errorMessage);
   if (n_strings < 0) {
     error << QString("Unable to convert '%1' to '%2': %3\n").arg(infile)
@@ -219 +233 @@
   /* Write the .po output. */
   QTextStream out(&poFile);
-  out.setCodec("UTF-8");
+  out.setCodec(codec);
   out << po;
   poFile.close();