Allow selecting languages to test with args

glotcode · Apr 23, 2022 · 0aeba6e · 0aeba6e
1 parent c1fd497
commit 0aeba6e
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/Glot/Language.hs b/Glot/Language.hs
@@ -10,6 +10,7 @@
 module Glot.Language
     ( Id
     , idToText
+    , idFromText
     , EditorConfig(..)
     , RunConfig(..)
     , Language(..)

diff --git a/app/image-tester.hs b/app/image-tester.hs
@@ -18,19 +18,39 @@ import qualified Data.Aeson as Aeson
 import qualified Glot.Language as Language
 import qualified Glot.DockerRun as DockerRun
 import qualified Glot.Snippet as Snippet
+import qualified System.Environment as Env
 
 
 main :: IO ()
 main =
     let
+        prepareIds :: [String] -> [Language.Id]
+        prepareIds languageIds =
+            languageIds
+                & map Text.pack
+                & map Language.idFromText
+                & Maybe.catMaybes
+
+        filterLanguages :: [Language.Id] -> [Language] -> [Language]
+        filterLanguages languageIds languages =
+            if null languageIds then
+                languages
+            else
+                languages
+                    & filter (\lang -> elem (Language.identifier lang) languageIds)
+
         prepareLanguages :: [Language] -> [LanguageData]
         prepareLanguages languages =
             languages
                 & map languageDataFromLanguage
                 & Maybe.catMaybes
     in do
+    languageIdsToRun <- Env.getArgs
     languages <- Language.readLanguages
-    mapM_ runLanguage (prepareLanguages languages)
+    languages
+        & filterLanguages (prepareIds languageIdsToRun)
+        & prepareLanguages
+        & mapM_ runLanguage
 
 
 data LanguageData = LanguageData