Add --keep-source-strings

MiguelDomingues · Nov 14, 2022 · 27e1485 · 27e1485
1 parent adc7966
commit 27e1485
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 4 deletions.
diff --git a/src/MarkdownLocalize.CLI/Program.cs b/src/MarkdownLocalize.CLI/Program.cs
@@ -96,6 +96,9 @@ public static int Main(string[] args)
         [Option("--add-front-matter-key", "Add key:value to front-matter.", CommandOptionType.MultipleValue)]
         public List<string> AddFrontMatter { get; } = null;
 
+        [Option("--keep-source-strings", "Keep source strings for non-translated strings.", CommandOptionType.NoValue)]
+        public bool KeepSourceStrings { get; } = false;
+
         private int OnExecute()
         {
             InitMarkdownParserOptions();
@@ -157,7 +160,7 @@ private void Translate(string inputMarkdown, string outputMarkdown, string input
             var catalog = POT.Load(po);
             TranslationInfo info;
             string relativeToSource = PathUtils.GetRelativePath(outputMarkdown, inputMarkdown, true);
-            string translatedMarkdown = POT.Translate(catalog, md, inputMarkdown, relativeToSource, out info);
+            string translatedMarkdown = POT.Translate(catalog, md, inputMarkdown, relativeToSource, KeepSourceStrings, out info);
             Log(string.Format(TRANSLATION_INFO, info.TranslatedCount, info.TotalCount));
             int ratio = info.TotalCount > 0 ? (int)(info.TranslatedCount * 1.0 / info.TotalCount * 100) : 0;
             if (info.TotalCount > 0)

diff --git a/src/MarkdownLocalize.POT/POT.cs b/src/MarkdownLocalize.POT/POT.cs
@@ -148,12 +148,16 @@ static POCatalog GenerateCatalog()
         return catalog;
     }
 
-    public static string Translate(POCatalog catalog, string markdown, string fileName, string pathToSource, out TranslationInfo info)
+    public static string Translate(POCatalog catalog, string markdown, string fileName, string pathToSource, bool keepSourceStrings, out TranslationInfo info)
     {
         string translatedMarkdown = MarkdownParser.Translate(markdown, (si) =>
         {
             var key = new POKey(NormalizeLineBreaks(si.String), null, si.Context);
             string translation = catalog.GetTranslation(key);
+            if (keepSourceStrings && translation == null)
+            {
+                translation = si.String;
+            }
             return translation != null ? translation.Trim() : null;
         }, fileName, pathToSource, catalog.Language, out info);
 

diff --git a/test/MarkdownLocalize.Tests/TranslateMarkdown.cs b/test/MarkdownLocalize.Tests/TranslateMarkdown.cs
@@ -26,7 +26,7 @@ public void TranslateSimple(string poFile, string originalMarkdown, string trans
         });
         var catalog = POT.Load(ReadPO(poFile));
         TranslationInfo info;
-        string md = POT.Translate(catalog, originalMarkdown, null, null, out info);
+        string md = POT.Translate(catalog, originalMarkdown, null, null, false, out info);
 
         Assert.Equal(translatedMarkdown, md);
         Assert.Equal(expectedTotalCount, info.TotalCount);
@@ -44,10 +44,22 @@ public void TranslateHtml(string poFile, string originalMarkdown, string transla
         });
         var catalog = POT.Load(ReadPO(poFile));
         TranslationInfo info;
-        string md = POT.Translate(catalog, originalMarkdown, null, null, out info);
+        string md = POT.Translate(catalog, originalMarkdown, null, null, false, out info);
 
         Assert.Equal(translatedMarkdown, md);
         Assert.Equal(expectedTotalCount, info.TotalCount);
         Assert.Equal(expectedTranslatedCount, info.TranslatedCount);
     }
+
+    [Fact]
+    public void TranslateKeepSource()
+    {
+        var catalog = POT.Load(ReadPO("headings.pt-PT.po"));
+        TranslationInfo info;
+        string md = POT.Translate(catalog, "# Heading\n\n##New Heading", null, null, true, out info);
+
+        Assert.Equal("# Título\n\n##New Heading", md);
+        Assert.Equal(2, info.TotalCount);
+        Assert.Equal(2, info.TranslatedCount);
+    }
 }