sstable-1.0: src/sstable.hs

{-# LANGUAGE DeriveDataTypeable #-}

module Main where

import System.Console.CmdArgs
import qualified Data.ByteString as B
import qualified Data.ByteString.Char8 as B8
import System.IO (isEOF)
import Data.Char (isSpace)
import Control.Monad (forM_, unless)

import Data.SSTable.Writer
import Data.SSTable.Reader

data Args 
  = Write { path :: FilePath }
  | Read  { path :: FilePath }
  deriving (Show, Data, Typeable)

writeMode = mode $ Write { path = def &= argPos 0 & typFile }
readMode  = mode $ Read  { path = def &= argPos 0 & typFile }

main = doit =<< cmdArgs "sstable" [writeMode, readMode]

doit (Write path) = withWriter path go
  where
    go writer = do
      eof <- isEOF
      unless eof $ parseLine writer >> go writer

    strip :: B.ByteString -> B.ByteString
    strip = B8.dropWhile isSpace
    keyValue line = let (k, v) = B8.break isSpace line in (strip k, strip v)

    parseLine writer = writeEntry writer . keyValue =<< B.getLine

doit (Read path) = withReader path enum
  where 
    enum reader = do
      entries <- scan reader B.empty
      forM_ entries $ \(key, value) -> do
        B.putStr key
        putStr " => "
        B.putStrLn value