modules = Builtin

modules = Builtin, PrimIO

	, Control.Console
	, Control.IO

module PrimIO
{-
Taken from Idris2 STDLIB
Copyright (c) 2019 Edwin Brady
    School of Computer Science, University of St Andrews
All rights reserved.
This code is derived from software written by Edwin Brady
(ecb10@st-andrews.ac.uk).
Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions
are met:
1. Redistributions of source code must retain the above copyright
   notice, this list of conditions and the following disclaimer.
2. Redistributions in binary form must reproduce the above copyright
   notice, this list of conditions and the following disclaimer in the
   documentation and/or other materials provided with the distribution.
3. None of the names of the copyright holders may be used to endorse
   or promote products derived from this software without specific
   prior written permission.
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS ``AS IS'' AND ANY
EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT HOLDERS BE
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
-}
import public Builtin
%default total
public export
data IORes : Type -> Type where
  MkIORes : a -> %World -> IORes a
public export
PrimIO : Type -> Type
PrimIO a = %World -> IORes a
export
prim__io_pure : a -> PrimIO a
prim__io_pure x w = MkIORes x w
export
prim__io_bind : PrimIO a -> (a -> PrimIO b) -> PrimIO b
prim__io_bind fn k w = let MkIORes x' w' := fn w in k x' w'

module Control.IO
import public PrimIO
import Algebra.Control.Functor
import Algebra.Control.Applicative
import Algebra.Control.Monad
import Control.Console
%default total
public export
data IO : Type -> Type where
  MkIO : PrimIO a -> IO a
public export
Functor IO where
  map f (MkIO p) = MkIO (p `prim__io_bind` (prim__io_pure . f))
  proofIdentity    = believe_me ()
  proofComposition = believe_me ()
public export
Applicative IO where
  pure x = MkIO $ prim__io_pure x
  (MkIO pf) <*> (MkIO px) = MkIO (
    pf `prim__io_bind` (\f =>
    px `prim__io_bind` (\x =>
      prim__io_pure $ f x)))
  proofIdentity     = believe_me ()
  proofHomomorphism = believe_me ()
  proofInterchange  = believe_me ()
  proofComposition  = believe_me ()
public export
Monad IO where
  bind (MkIO px) f = MkIO (
    px `prim__io_bind` (\x =>
      let MkIO p := f x in p))
  proofLeftIdentity  = believe_me ()
  proofRightIdentity = believe_me ()
  proofAssociativity = believe_me ()
%foreign "C:getchar,libc 6"
%extern
prim__getChar : PrimIO Char
%foreign "C:idris2_getStr, libidris2_support, idris_support.h"
%extern
prim__getStr : PrimIO String
%foreign "C:putchar,libc 6"
%extern
prim__putChar : Char -> PrimIO ()
%foreign "C:idris2_putStr, libidris2_support, idris_support.h"
%extern
prim__putStr : String -> PrimIO ()
public export
Console IO where
  getChar = MkIO prim__getChar
  getLine = MkIO prim__getStr
  putChar   c = MkIO $ prim__putChar c
  putString s = MkIO $ prim__putStr s

module Control.Console
import public Builtin
%default total
public export
interface Console t where
  getChar : t Char
  getLine : t String
  putChar : Char -> t ()
  putString : String -> t ()
public export
putLine : Console t => String -> t ()
putLine = putString -- TODO: Append newline

public export
data Unit = MkUnit