mirror of
https://github.com/The1Penguin/org-to-mafiauniverse.git
synced 2024-11-22 02:08:13 +00:00
83 lines
2.4 KiB
Haskell
83 lines
2.4 KiB
Haskell
module Main where
|
|
|
|
import Control.Applicative hiding (some)
|
|
import Control.Monad
|
|
import Data.Functor
|
|
import Data.Void
|
|
import GHC.Stack.CCS (whereFrom)
|
|
import GHC.StaticPtr (StaticPtrInfo)
|
|
import System.Environment
|
|
import System.Exit
|
|
import Text.Megaparsec hiding (satisfy)
|
|
import Text.Megaparsec.Char
|
|
|
|
type Parser = Parsec Void String
|
|
|
|
newtype URL = URL String
|
|
|
|
data Info =
|
|
Bread String |
|
|
List [Info] |
|
|
Header String |
|
|
Subheader String |
|
|
Subsubheader String |
|
|
Image URL |
|
|
Video URL |
|
|
Link URL Info |
|
|
Spoiler Info
|
|
|
|
instance Show Info where
|
|
show :: Info -> String
|
|
show info = case info of
|
|
Bread text -> text
|
|
List texts -> "[LIST]" ++ concatMap show texts ++ "[/LIST]"
|
|
Header text -> "[TITLE]" ++ text ++ "[/TITLE]"
|
|
Subheader text -> "[SIZE=4]" ++ text ++ "[/SIZE]"
|
|
Subsubheader text -> "[SIZE=2]" ++ text ++ "[/SIZE]"
|
|
Image (URL url) -> "[IMG]" ++ url ++ "[/IMG]"
|
|
Video (URL url) -> "[VIDEO]" ++ url ++ "[/VIDEO]"
|
|
Link (URL url) inf -> "[URL=\"" ++ url ++ "\"]" ++ show inf ++ "[/IMG]"
|
|
Spoiler inf -> "[SPOILER]" ++ show inf ++ "[/SPOILER]"
|
|
|
|
newtype Post = Post [Info]
|
|
|
|
parseHelper :: Parser String -> String -> String
|
|
parseHelper parser x = case parse parser "" x of
|
|
Left bundle -> error $ errorBundlePretty bundle
|
|
Right text -> text
|
|
|
|
parseFile :: String -> String
|
|
parseFile = unlines . map (parseHelper fileParser) . lines
|
|
|
|
fileParser :: Parser String
|
|
fileParser = try $
|
|
string "* " <|>
|
|
string "** "
|
|
|
|
parseCli :: [String] -> IO String
|
|
parseCli [] = return usage
|
|
parseCli [x] =
|
|
if head x /= '-'
|
|
then parseFile <$> readFile x
|
|
else return $ useArgs $ parseHelper argParse x
|
|
parseCli [x, _] = return $ useArgs $ parseHelper argParse x
|
|
parseCli _ = fail "Too many arugemnts"
|
|
|
|
argParse :: Parser String
|
|
argParse = string "-" >> some alphaNumChar
|
|
|
|
useArgs :: String -> String
|
|
useArgs [] = ""
|
|
useArgs ('h' : _) = usage
|
|
useArgs ('v' : _) = version
|
|
useArgs (_ : xs) = useArgs xs
|
|
|
|
usage :: String
|
|
usage = "Usage: otm [-hv] [file]"
|
|
|
|
version :: String
|
|
version = "otm 0.1"
|
|
|
|
main :: IO ()
|
|
main = putStr =<< parseCli =<< getArgs
|