apaxianparent.hs

import Data.List (intercalate)
import Data.List.NonEmpty (NonEmpty ((:|)), (<|))
import qualified Data.List.NonEmpty as NE

main :: IO ()
main = interact (formatOutput . solve . parseInput)

-- helpers for parsing
splitOn' :: Char -> String -> NonEmpty String
splitOn' sep s = case s of { [] -> "" :| []; (part : remainder) -> if part == sep then "" <| splitOn' sep remainder else let head :| tail = splitOn' sep remainder in (part : head) :| tail}

splitOn :: Char -> String -> [String]
splitOn sep s = NE.toList (splitOn' sep s)

-- Define input and output here
type Input = (String, String)

type Output = String

formatOutput :: Output -> String
formatOutput solution = solution ++ "\n"

parseInput :: String -> Input
parseInput rawInput = let items = (splitOn ' ' (head $ lines rawInput)) in (items !! 0, items !! 1)

solve :: Input -> Output
solve (a, b) =
  case last a of
    'e' -> a ++ "x" ++ b
    'a' -> init a ++ "ex" ++ b
    'i' -> init a ++ "ex" ++ b
    'u' -> init a ++ "ex" ++ b
    'o' -> init a ++ "ex" ++ b
    'x' -> if last (init a)  == 'e' then a ++ b else a ++ "ex" ++ b
    _   -> a ++ "ex" ++ b