Array Literal Generator

ToArray.hs

#!/usr/bin/env stack
-- stack --resolver lts-7.0 --install-ghc runghc --package turtle

{-# LANGUAGE OverloadedStrings #-}

import Data.Text (pack, unpack)
import Turtle

joinM :: (Monoid m) => m -> [m] -> m
joinM separator = foldl1 (\a b -> a `mappend` separator `mappend` b)

wrap :: Char -> String -> String
wrap c = wrap' c c

wrap' :: Char -> Char -> String -> String
wrap' c1 c2 s = [c1] ++ s ++ [c2]

parser :: Parser (Maybe Text)
parser = optional (optText "lang" 'l' "Language (swift|objc|java) - default is swift")

toArrayLiteral :: Text -> [String] -> String
toArrayLiteral "swift" = wrap' '[' ']' . joinM ", " . map (wrap '"')
toArrayLiteral "objc"  = ('@':) . wrap' '[' ']' . joinM ", " . map ('@':) . map (wrap '"')
toArrayLiteral "java"  = swiftArray
toArrayLiteral lang = \_ -> unpack $ "Not supported language type '" <> lang <> "'"

main = do
    mLang <- options "Array Literal Converter" parser
    contents <- getContents
    let lang = case mLang of Nothing -> "swift"
                             Just l -> l
        sink = toArrayLiteral lang
    let result = sink . lines $ contents
    putStrLn result