import Control.Comonad.Cofree (Cofree (..))
import Data.Foldable (toList)

import Data.Apart (Apart (..))
import Data.Apart.Shape (Shape (..))
import Data.Apart.Combinators (limit, throughout, recover)
import Data.Apart.Transformations (Segmented (..), Scattered (..))
import Data.Apart.Structures.Stack (Stack)

-- part of data structure in some file
scattered :: Scattered Stack Int FilePath
scattered = Apart $ 1 :< Ready (Just $ 2 :< Ready (Just $ 3 :< Converted "Example/piece.txt"))

read_from_file :: FilePath -> IO (Segmented Stack Int)
read_from_file fp = read @(Segmented Stack Int) <$> readFile fp

-- the whole structure in memory
in_memory :: Stack Int
in_memory = 1 :< Just (2 :< Just (3 :< Just (4 :< Just (5 :< Nothing))))

save_to_file :: FilePath -> Segmented Stack Int -> IO FilePath
save_to_file fp structure = writeFile fp (show structure) *> pure fp

main = do
	print "Splitting data structure based on limit, the rest should be putted in file"
	limit 4 (save_to_file "Example/backup.txt") in_memory >>= print . toList . part
	print "Recovering data structure, the rest of structure should be in file"
	recover read_from_file scattered >>= print . toList
	print "Traverse over structure with action, recover segments on the way"
	throughout print read_from_file scattered