mirror of
https://github.com/github/semantic.git
synced 2024-11-28 10:15:55 +03:00
68 lines
2.6 KiB
Haskell
68 lines
2.6 KiB
Haskell
module Split where
|
|
|
|
import Diff
|
|
import Patch
|
|
import Syntax
|
|
import Term
|
|
import Unified
|
|
import Control.Comonad.Cofree
|
|
import Control.Monad.Free
|
|
import qualified Data.Map as Map
|
|
import qualified Data.Set as Set
|
|
import Rainbow
|
|
|
|
type ClassName = String
|
|
type Element a = Cofree (Syntax a) (Maybe ClassName, String)
|
|
|
|
data HTML =
|
|
Text String
|
|
| Span (Maybe ClassName) String
|
|
| Ul (Maybe ClassName) [HTML]
|
|
| Dl (Maybe ClassName) [HTML]
|
|
| Dt String
|
|
deriving (Eq, Show)
|
|
|
|
split :: Diff a Info -> String -> String -> IO ByteString
|
|
split _ _ _ = return mempty
|
|
|
|
splitDiff :: Diff a Info -> String -> String -> Patch (HTML, Range)
|
|
splitDiff diff before after = iter toElements $ splitPatch before after <$> diff
|
|
where
|
|
toElements (Annotated (left, right) (Leaf _)) = Replace (leafToElement before left) (leafToElement after right)
|
|
|
|
leafToElement source (Info range _ categories) = (Span (classify categories) $ substring range source, range)
|
|
|
|
splitPatch :: String -> String -> Patch (Term a Info) -> Patch (HTML, Range)
|
|
splitPatch before after (Replace a b) = Replace (splitTerm before a) (splitTerm after b)
|
|
splitPatch _ after (Insert b) = Insert $ splitTerm after b
|
|
splitPatch before _ (Delete a) = Delete $ splitTerm before a
|
|
|
|
splitTerm :: String -> Term a Info -> (HTML, Range)
|
|
splitTerm source = cata toElement where
|
|
toElement (Info range lineRange categories) (Leaf _) = (Span (classify categories) $ substring range source, range)
|
|
toElement (Info range lineRange categories) (Indexed i) = makeList i range categories
|
|
toElement (Info range lineRange categories) (Fixed i) = makeList i range categories
|
|
toElement (Info range lineRange categories) (Keyed k) = makeMap (Map.toList k) range categories
|
|
|
|
accumulate (children, previous) (child, range) = (children ++ [ subtext previous $ start range, child ], end range)
|
|
accumulateFromMap (children, previous) (key, (child, range)) = (children ++ [ subtext previous $ start range, Dt key, child ], end range)
|
|
|
|
makeList i range categories = (Ul (classify categories) items, range)
|
|
where
|
|
(children, previous) = foldl accumulate ([], start range) i
|
|
items = children ++ [ subtext previous $ end range ]
|
|
|
|
makeMap k range categories = (Dl (classify categories) items, range)
|
|
where
|
|
(children, previous) = foldl accumulateFromMap ([], start range) k
|
|
items = children ++ [ subtext previous $ end range ]
|
|
|
|
subtext :: Int -> Int -> HTML
|
|
subtext start end = Text $ substring (Range start end) source
|
|
|
|
splitHTMLIntoLines :: HTML -> [HTML]
|
|
splitHTMLIntoLines (Text string) = Text <$> lines string
|
|
|
|
classify :: Set.Set Category -> Maybe ClassName
|
|
classify = foldr (const . Just) Nothing
|