1
1
mirror of https://github.com/github/semantic.git synced 2024-12-22 06:11:49 +03:00

Extract Command.Stdin

This commit is contained in:
Timothy Clem 2017-05-17 10:02:18 -07:00
parent 7ba950cf47
commit fcf7155a35
4 changed files with 55 additions and 49 deletions

View File

@ -20,6 +20,7 @@ library
, Command , Command
, Command.Files , Command.Files
, Command.Git , Command.Git
, Command.Stdin
, Data.Align.Generic , Data.Align.Generic
, Data.Functor.Both , Data.Functor.Both
, Data.Functor.Classes.Eq.Generic , Data.Functor.Classes.Eq.Generic

View File

@ -12,6 +12,7 @@ module Command
import qualified Command.Files as Files import qualified Command.Files as Files
import qualified Command.Git as Git import qualified Command.Git as Git
import qualified Command.Stdin as Stdin
import Control.Monad.Free.Freer import Control.Monad.Free.Freer
import Control.Monad.IO.Class import Control.Monad.IO.Class
import Data.Functor.Both import Data.Functor.Both
@ -59,7 +60,7 @@ readFilesAtSHAs gitDir alternates paths shas = ReadFilesAtSHAs gitDir alternates
runCommand :: Command a -> IO a runCommand :: Command a -> IO a
runCommand = iterFreerA $ \ command yield -> case command of runCommand = iterFreerA $ \ command yield -> case command of
ReadFile path lang -> Files.readFile path lang >>= yield ReadFile path lang -> Files.readFile path lang >>= yield
ReadStdin -> Files.readStdin >>= yield ReadStdin -> Stdin.readStdin >>= yield
ReadFilesAtSHA gitDir alternates paths sha -> Git.readFilesAtSHA gitDir alternates paths sha >>= yield ReadFilesAtSHA gitDir alternates paths sha -> Git.readFilesAtSHA gitDir alternates paths sha >>= yield
ReadFilesAtSHAs gitDir alternates paths shas -> Git.readFilesAtSHAs gitDir alternates paths shas >>= yield ReadFilesAtSHAs gitDir alternates paths shas -> Git.readFilesAtSHAs gitDir alternates paths shas >>= yield
LiftIO io -> io >>= yield LiftIO io -> io >>= yield

View File

@ -1,9 +1,8 @@
{-# LANGUAGE OverloadedStrings, RecordWildCards, DeriveGeneric, DeriveAnyClass, TypeSynonymInstances #-} {-# LANGUAGE OverloadedStrings, TypeSynonymInstances #-}
module Command.Files module Command.Files
( readFile ( readFile
, transcode , transcode
, languageForFilePath , languageForFilePath
, readStdin
) where ) where
import Prologue hiding (readFile) import Prologue hiding (readFile)
@ -15,10 +14,6 @@ import Control.Exception (catch, IOException)
import qualified Data.Text.ICU.Convert as Convert import qualified Data.Text.ICU.Convert as Convert
import qualified Data.Text.ICU.Detect as Detect import qualified Data.Text.ICU.Detect as Detect
import qualified Control.Applicative as A
import Data.String
import Data.Aeson
import Data.Functor.Both
-- | Read a file to a SourceBlob, transcoding to UTF-8 along the way. -- | Read a file to a SourceBlob, transcoding to UTF-8 along the way.
readFile :: FilePath -> Maybe Language -> IO SourceBlob readFile :: FilePath -> Maybe Language -> IO SourceBlob
@ -37,45 +32,3 @@ transcode text = fromText <$> do
-- | Return a language based on a FilePath's extension, or Nothing if extension is not found or not supported. -- | Return a language based on a FilePath's extension, or Nothing if extension is not found or not supported.
languageForFilePath :: FilePath -> Maybe Language languageForFilePath :: FilePath -> Maybe Language
languageForFilePath = languageForType . toS . takeExtension languageForFilePath = languageForType . toS . takeExtension
-- PROTOTYPE
readStdin :: IO [Both SourceBlob]
readStdin = do
input <- B.getContents
let request = decode (toS input) :: Maybe DiffRequest
pure $ maybe mempty sourceBlobs request
newtype DiffRequest = DiffRequest
{ blobs :: [BlobPair]
} deriving (Show, Generic, FromJSON, ToJSON)
data BlobPair = BlobPair
{ path :: String
, before :: Maybe BlobContent
, after :: Maybe BlobContent
} deriving (Show, Generic, FromJSON, ToJSON)
newtype BlobContent = BlobContent { utf8Text :: ByteString }
deriving (Eq, Show)
instance ToJSON BlobContent where
toJSON = String . decodeUtf8 . utf8Text
instance FromJSON BlobContent where
parseJSON (String t) = (pure . BlobContent . encodeUtf8) t
parseJSON _ = A.empty
sourceBlobs :: DiffRequest -> [Both SourceBlob]
sourceBlobs DiffRequest{..} = toSourceBlob <$> blobs
where
toSourceBlob :: BlobPair -> Both SourceBlob
toSourceBlob BlobPair{..} = fmap (sourceBlob' path) (both before after)
sourceBlob' :: FilePath -> Maybe BlobContent -> SourceBlob
sourceBlob' path maybeContent = maybe (emptySourceBlob path) (sourceBlob path (languageForFilePath path)) (source' maybeContent)
source' :: Maybe BlobContent -> Maybe Source
source' = fmap (Source . utf8Text)

51
src/Command/Stdin.hs Normal file
View File

@ -0,0 +1,51 @@
{-# LANGUAGE RecordWildCards, DeriveGeneric, DeriveAnyClass, TypeSynonymInstances #-}
module Command.Stdin
( readStdin
) where
import Command.Files (languageForFilePath)
import qualified Control.Applicative as A
import Data.Aeson
import qualified Data.ByteString as B
import Data.Functor.Both
import Data.String
import Prologue
import Source
readStdin :: IO [Both SourceBlob]
readStdin = do
input <- B.getContents
let request = decode (toS input) :: Maybe BlobDiff
pure $ maybe mempty sourceBlobs request
newtype BlobDiff = BlobDiff { blobs :: [BlobPair] }
deriving (Show, Generic, FromJSON, ToJSON)
data BlobPair = BlobPair
{ path :: String
, before :: Maybe BlobContent
, after :: Maybe BlobContent
} deriving (Show, Generic, FromJSON, ToJSON)
newtype BlobContent = BlobContent { utf8Text :: ByteString }
deriving (Eq, Show)
instance ToJSON BlobContent where
toJSON = String . decodeUtf8 . utf8Text
instance FromJSON BlobContent where
parseJSON (String t) = (pure . BlobContent . encodeUtf8) t
parseJSON _ = A.empty
sourceBlobs :: BlobDiff -> [Both SourceBlob]
sourceBlobs BlobDiff{..} = toSourceBlob <$> blobs
where
toSourceBlob :: BlobPair -> Both SourceBlob
toSourceBlob BlobPair{..} = fmap (sourceBlob' path) (both before after)
sourceBlob' :: FilePath -> Maybe BlobContent -> SourceBlob
sourceBlob' path maybeContent = maybe (emptySourceBlob path) (sourceBlob path (languageForFilePath path)) (source' maybeContent)
source' :: Maybe BlobContent -> Maybe Source
source' = fmap (Source . utf8Text)