-- | -- Copyright: (c) 2016 Ertugrul Söylemez -- License: BSD3 -- Maintainer: Ertugrul Söylemez <esz@posteo.de> -- Stability: experimental -- -- This module provides a rapid prototyping suite for GHCi that can be -- used standalone or integrated into editors. You can hot-reload -- individual running components as you make changes to their code. It -- is designed to shorten the development cycle during the development -- of long-running programs like servers, web applications and -- interactive user interfaces. -- -- It can also be used in the context of batch-style programs: Keep -- resources that are expensive to create in memory and reuse them -- across module reloads instead of reloading/recomputing them after -- every code change. -- -- Technically this package is a safe and convenient wrapper around -- <https://hackage.haskell.org/package/foreign-store foreign-store>. -- -- __Read the "Safety and securty" section before using this module!__ {-# LANGUAGE RankNTypes #-} module Rapid ( -- * Introduction -- $intro -- ** Communication -- $communication -- ** Reusing expensive resources -- $reusing -- ** Cabal notes -- $cabal -- ** Emacs integration -- $emacs -- ** Safety and security -- $safety -- * Hot code reloading Rapid, rapid, -- * Threads restart, restartWith, start, startWith, stop, -- * Communication createRef, deleteRef, writeRef ) where import Control.Concurrent.Async import Control.Concurrent.STM import Control.Exception import Data.Dynamic import Data.Map.Strict (Map) import qualified Data.Map.Strict as M import Data.Word import Foreign.Store -- | Handle to the current Rapid state. data Rapid k = Rapid { forall k. Rapid k -> TVar Bool rLock :: TVar Bool, -- ^ Lock on the current state. forall k. Rapid k -> TVar (Map k Dynamic) rRefs :: TVar (Map k Dynamic), -- ^ Mutable variables. forall k. Rapid k -> TVar (Map k (Async ())) rThreads :: TVar (Map k (Async ())) -- ^ Active threads. } -- | Cancel the given thread and wait for it to finish. cancelAndWait :: Async a -> IO () cancelAndWait :: forall a. Async a -> IO () cancelAndWait Async a tv = do forall a. Async a -> IO () cancel Async a tv () forall (f :: * -> *) a b. Functor f => a -> f b -> f a <$ forall a. Async a -> IO (Either SomeException a) waitCatch Async a tv -- | Get the value of the mutable variable with the given name. If it -- does not exist, it is created and initialised with the value returned -- by the given action. -- -- Mutable variables should only be used with values that can be -- garbage-collected, for example communication primitives like -- 'Control.Concurrent.MVar.MVar' and 'TVar', but also pure run-time -- information that is expensive to generate, for example the parsed -- contents of a file. createRef :: (Ord k, Typeable a) => Rapid k -- ^ Rapid state handle. -> k -- ^ Name of the mutable variable. -> IO a -- ^ Action to create. -> IO a createRef :: forall k a. (Ord k, Typeable a) => Rapid k -> k -> IO a -> IO a createRef Rapid k r k k IO a gen = forall k a. Ord k => Rapid k -> k -> (Maybe Dynamic -> IO (Maybe Dynamic, a)) -> IO a withRef Rapid k r k k forall a b. (a -> b) -> a -> b $ \Maybe Dynamic mxd -> case Maybe Dynamic mxd of Maybe Dynamic Nothing -> forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b fmap (\a x -> (forall a. a -> Maybe a Just (forall a. Typeable a => a -> Dynamic toDyn a x), a x)) IO a gen Just Dynamic xd | Just a x <- forall a. Typeable a => Dynamic -> Maybe a fromDynamic Dynamic xd -> forall (f :: * -> *) a. Applicative f => a -> f a pure (forall a. a -> Maybe a Just Dynamic xd, a x) | Bool otherwise -> forall e a. Exception e => e -> IO a throwIO (String -> IOError userError String "createRef: Wrong reference type") -- | Delete the mutable variable with the given name, if it exists. deleteRef :: (Ord k) => Rapid k -- ^ Rapid state handle. -> k -- ^ Name of the mutable variable. -> IO () deleteRef :: forall k. Ord k => Rapid k -> k -> IO () deleteRef Rapid k r k k = forall k a. Ord k => Rapid k -> k -> (Maybe Dynamic -> IO (Maybe Dynamic, a)) -> IO a withRef Rapid k r k k (\Maybe Dynamic _ -> forall (f :: * -> *) a. Applicative f => a -> f a pure (forall a. Maybe a Nothing, ())) -- | Retrieve the current Rapid state handle, and pass it to the given -- continuation. If the state handle doesn't exist, it is created. The -- key type @k@ is used for naming reloadable services like threads. -- -- __Warning__: The key type must not change during a session. If you -- need to change the key type, currently the safest option is to -- restart GHCi. -- -- This function uses the -- <https://hackage.haskell.org/package/foreign-store foreign-store library> -- to establish a state handle that survives GHCi reloads and is -- suitable for hot reloading. -- -- The first argument is the 'Store' index. If you do not use the -- /foreign-store/ library in your development workflow, just use 0, -- otherwise use any unused index. rapid :: forall k r. Word32 -- ^ Store index (if in doubt, use 0). -> (Rapid k -> IO r) -- ^ Action on the Rapid state. -> IO r rapid :: forall k r. Word32 -> (Rapid k -> IO r) -> IO r rapid Word32 stNum Rapid k -> IO r k = forall b. ((forall a. IO a -> IO a) -> IO b) -> IO b mask forall a b. (a -> b) -> a -> b $ \forall a. IO a -> IO a unmask -> forall a. Word32 -> IO (Maybe (Store a)) lookupStore Word32 stNum forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b >>= forall b a. b -> (a -> b) -> Maybe a -> b maybe (forall a. Store a -> IO a -> IO a storeAction forall k. Store (Rapid k) store forall {k}. IO (Rapid k) create) (\Store Any _ -> forall a. Store a -> IO a readStore forall k. Store (Rapid k) store) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b >>= forall {b}. (IO r -> IO b) -> Rapid k -> IO b pass forall a. IO a -> IO a unmask where create :: IO (Rapid k) create = forall (f :: * -> *) a. Applicative f => a -> f a pure forall k. TVar Bool -> TVar (Map k Dynamic) -> TVar (Map k (Async ())) -> Rapid k Rapid forall (f :: * -> *) a b. Applicative f => f (a -> b) -> f a -> f b <*> forall a. a -> IO (TVar a) newTVarIO Bool False forall (f :: * -> *) a b. Applicative f => f (a -> b) -> f a -> f b <*> forall a. a -> IO (TVar a) newTVarIO forall k a. Map k a M.empty forall (f :: * -> *) a b. Applicative f => f (a -> b) -> f a -> f b <*> forall a. a -> IO (TVar a) newTVarIO forall k a. Map k a M.empty pass :: (IO r -> IO b) -> Rapid k -> IO b pass IO r -> IO b unmask Rapid k r = do forall a. STM a -> IO a atomically forall a b. (a -> b) -> a -> b $ do forall a. TVar a -> STM a readTVar (forall k. Rapid k -> TVar Bool rLock Rapid k r) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b >>= Bool -> STM () check forall b c a. (b -> c) -> (a -> b) -> a -> c . Bool -> Bool not forall a. TVar a -> a -> STM () writeTVar (forall k. Rapid k -> TVar Bool rLock Rapid k r) Bool True IO r -> IO b unmask (Rapid k -> IO r k Rapid k r) forall a b. IO a -> IO b -> IO a `finally` forall a. STM a -> IO a atomically (forall a. TVar a -> a -> STM () writeTVar (forall k. Rapid k -> TVar Bool rLock Rapid k r) Bool False) store :: Store (Rapid k) store :: forall k. Store (Rapid k) store = forall a. Word32 -> Store a Store Word32 stNum -- | Create a thread with the given name that runs the given action. -- -- The thread is restarted each time an update occurs. restart :: (Ord k) => Rapid k -- ^ Rapid state handle. -> k -- ^ Name of the thread. -> IO () -- ^ Action the thread runs. -> IO () restart :: forall k. Ord k => Rapid k -> k -> IO () -> IO () restart = forall k. Ord k => (forall a. IO a -> IO (Async a)) -> Rapid k -> k -> IO () -> IO () restartWith forall a. IO a -> IO (Async a) async -- | Create a thread with the given name that runs the given action. -- -- The thread is restarted each time an update occurs. -- -- The first argument is the function used to create the thread. It can -- be used to select between 'async', 'asyncBound' and 'asyncOn'. restartWith :: (Ord k) => (forall a. IO a -> IO (Async a)) -- ^ Thread creation function. -> Rapid k -- ^ Rapid state handle. -> k -- ^ Name of the thread. -> IO () -- ^ Action the thread runs. -> IO () restartWith :: forall k. Ord k => (forall a. IO a -> IO (Async a)) -> Rapid k -> k -> IO () -> IO () restartWith forall a. IO a -> IO (Async a) myAsync Rapid k r k k IO () action = forall k. Ord k => Rapid k -> k -> (Maybe (Async ()) -> IO (Maybe (Async ()))) -> IO () withThread Rapid k r k k forall a b. (a -> b) -> a -> b $ \Maybe (Async ()) mtv -> do forall (t :: * -> *) (m :: * -> *) a b. (Foldable t, Monad m) => (a -> m b) -> t a -> m () mapM_ forall a. Async a -> IO () cancelAndWait Maybe (Async ()) mtv forall a. a -> Maybe a Just forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b <$> forall a. IO a -> IO (Async a) myAsync IO () action -- | Create a thread with the given name that runs the given action. -- -- When an update occurs and the thread is currently not running, it is -- started. start :: (Ord k) => Rapid k -- ^ Rapid state handle. -> k -- ^ Name of the thread. -> IO () -- ^ Action the thread runs. -> IO () start :: forall k. Ord k => Rapid k -> k -> IO () -> IO () start = forall k. Ord k => (forall a. IO a -> IO (Async a)) -> Rapid k -> k -> IO () -> IO () startWith forall a. IO a -> IO (Async a) async -- | Create a thread with the given name that runs the given action. -- -- When an update occurs and the thread is currently not running, it is -- started. -- -- The first argument is the function used to create the thread. It can -- be used to select between 'async', 'asyncBound' and 'asyncOn'. startWith :: (Ord k) => (forall a. IO a -> IO (Async a)) -- ^ Thread creation function. -> Rapid k -- ^ Rapid state handle. -> k -- ^ Name of the thread. -> IO () -- ^ Action the thread runs. -> IO () startWith :: forall k. Ord k => (forall a. IO a -> IO (Async a)) -> Rapid k -> k -> IO () -> IO () startWith forall a. IO a -> IO (Async a) myAsync Rapid k r k k IO () action = forall k. Ord k => Rapid k -> k -> (Maybe (Async ()) -> IO (Maybe (Async ()))) -> IO () withThread Rapid k r k k forall a b. (a -> b) -> a -> b $ forall b a. b -> (a -> b) -> Maybe a -> b maybe (forall a. a -> Maybe a Just forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b <$> forall a. IO a -> IO (Async a) myAsync IO () action) (\Async () tv -> forall a. Async a -> IO (Maybe (Either SomeException a)) poll Async () tv forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b >>= forall b a. b -> (a -> b) -> Maybe a -> b maybe (forall (f :: * -> *) a. Applicative f => a -> f a pure (forall a. a -> Maybe a Just Async () tv)) (\Either SomeException () _ -> forall a. a -> Maybe a Just forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b <$> forall a. IO a -> IO (Async a) myAsync IO () action)) -- | Delete the thread with the given name. -- -- When an update occurs and the thread is currently running, it is -- cancelled. stop :: (Ord k) => Rapid k -> k -> x -> IO () stop :: forall k x. Ord k => Rapid k -> k -> x -> IO () stop Rapid k r k k x _ = forall k. Ord k => Rapid k -> k -> (Maybe (Async ()) -> IO (Maybe (Async ()))) -> IO () withThread Rapid k r k k forall a b. (a -> b) -> a -> b $ \Maybe (Async ()) mtv -> forall a. Maybe a Nothing forall (f :: * -> *) a b. Functor f => a -> f b -> f a <$ forall (t :: * -> *) (m :: * -> *) a b. (Foldable t, Monad m) => (a -> m b) -> t a -> m () mapM_ forall a. Async a -> IO () cancelAndWait Maybe (Async ()) mtv -- | Apply the given transform to the reference with the given name. withRef :: (Ord k) => Rapid k -> k -> (Maybe Dynamic -> IO (Maybe Dynamic, a)) -> IO a withRef :: forall k a. Ord k => Rapid k -> k -> (Maybe Dynamic -> IO (Maybe Dynamic, a)) -> IO a withRef Rapid k r k k Maybe Dynamic -> IO (Maybe Dynamic, a) f = do (Maybe Dynamic mx, a y) <- forall a. STM a -> IO a atomically (forall k a. Ord k => k -> Map k a -> Maybe a M.lookup k k forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b <$> forall a. TVar a -> STM a readTVar (forall k. Rapid k -> TVar (Map k Dynamic) rRefs Rapid k r)) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b >>= Maybe Dynamic -> IO (Maybe Dynamic, a) f forall a. STM a -> IO a atomically forall a b. (a -> b) -> a -> b $ forall a. TVar a -> (a -> a) -> STM () modifyTVar' (forall k. Rapid k -> TVar (Map k Dynamic) rRefs Rapid k r) (forall b a. b -> (a -> b) -> Maybe a -> b maybe (forall k a. Ord k => k -> Map k a -> Map k a M.delete k k) (forall k a. Ord k => k -> a -> Map k a -> Map k a M.insert k k) Maybe Dynamic mx) forall (f :: * -> *) a. Applicative f => a -> f a pure a y -- | Apply the given transform to the thread with the given name. withThread :: (Ord k) => Rapid k -> k -> (Maybe (Async ()) -> IO (Maybe (Async ()))) -> IO () withThread :: forall k. Ord k => Rapid k -> k -> (Maybe (Async ()) -> IO (Maybe (Async ()))) -> IO () withThread Rapid k r k k Maybe (Async ()) -> IO (Maybe (Async ())) f = forall a. STM a -> IO a atomically (forall k a. Ord k => k -> Map k a -> Maybe a M.lookup k k forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b <$> forall a. TVar a -> STM a readTVar (forall k. Rapid k -> TVar (Map k (Async ())) rThreads Rapid k r)) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b >>= Maybe (Async ()) -> IO (Maybe (Async ())) f forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b >>= forall a. STM a -> IO a atomically forall b c a. (b -> c) -> (a -> b) -> a -> c . forall a. TVar a -> (a -> a) -> STM () modifyTVar' (forall k. Rapid k -> TVar (Map k (Async ())) rThreads Rapid k r) forall b c a. (b -> c) -> (a -> b) -> a -> c . forall b a. b -> (a -> b) -> Maybe a -> b maybe (forall k a. Ord k => k -> Map k a -> Map k a M.delete k k) (forall k a. Ord k => k -> a -> Map k a -> Map k a M.insert k k) -- | Overwrite the mutable variable with the given name with the value -- returned by the given action. If the mutable variable does not -- exist, it is created. -- -- This function may be used to change the value type of a mutable -- variable. writeRef :: (Ord k, Typeable a) => Rapid k -- ^ Rapid state handle. -> k -- ^ Name of the mutable variable. -> IO a -- ^ Value action. -> IO a writeRef :: forall k a. (Ord k, Typeable a) => Rapid k -> k -> IO a -> IO a writeRef Rapid k r k k IO a gen = forall k a. Ord k => Rapid k -> k -> (Maybe Dynamic -> IO (Maybe Dynamic, a)) -> IO a withRef Rapid k r k k forall a b. (a -> b) -> a -> b $ \Maybe Dynamic _ -> forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b fmap (\a x -> (forall a. a -> Maybe a Just (forall a. Typeable a => a -> Dynamic toDyn a x), a x)) IO a gen {- $cabal In general a Cabal project should not have this library as a build-time dependency. However, in certain environments (like Nix-based development) it may be beneficial to include it in the @.cabal@ file regardless. A simple solution is to add a flag: > flag Devel > default: False > description: Enable development dependencies > manual: True > > library > build-depends: > base >= 4.8 && < 5, > {- ... -} > if flag(devel) > build-depends: rapid > {- ... -} Now you can configure your project with @-fdevel@ during development and have this module available. -} {- $communication If you need your background threads to communicate with each other, for example by using concurrency primitives, some additional support is required. You cannot just create a 'TVar' within your @update@ action. It would be a different one for every invocation, so threads that are restarted would not communicate with already running threads, because they would use a fresh @TVar@, while the old threads would still use the old one. To solve this, you need to wrap your 'newTVar' action with 'createRef'. The @TVar@ created this way will survive reloads in the same way as background threads do. In particular, if there is already one from an older invocation of @update@, it will be reused: > import Control.Concurrent.STM > import Control.Monad > import Rapid > > update = > rapid 0 $ \r -> do > mv1 <- createRef r "var1" newEmptyTMVarIO > mv2 <- createRef r "var2" newEmptyTMVarIO > > start r "producer" $ > mapM_ (atomically . putTMVar mv1) [0 :: Integer ..] > > restart r "consumer" $ > forever . atomically $ do > x <- takeTMVar mv1 > putTMVar mv2 (x, "blah") > > -- For debugging the update action: > replicateM_ 3 $ > atomically (takeTMVar mv2) >>= print You can now change the string @"blah"@ in the consumer thread and then run @update@. You will notice that the numbers in the left component of the tuples keep increasing even after a reload, while the string in the right component changes. That means the producer thread was not restarted, but the consumer thread was. Yet the restarted consumer thread still refers to the same @TVar@ as before, so it still receives from the producer. -} {- $emacs This library integrates well with <https://haskell.github.io/haskell-mode/manual/latest/Interactive-Haskell.html haskell-interactive-mode>, particularly with its somewhat hidden @haskell-process-reload-devel-main@ function. This function finds your @DevelMain@ module by looking for a buffer named @DevelMain.hs@, loads or reloads it in your current project's interactive session and then runs @update@. Assuming that you are already using /haskell-interactive-mode/ all you need to do to use it is to keep your @DevelMain@ module open in a buffer and type @M-x haskell-process-reload-devel-main RET@ when you want to hot-reload. You may want to bind it to a key: > (define-key haskell-mode-map (kbd "C-c m") 'haskell-process-reload-devel-main) Since you will likely always reload the current module before running @update@, you can save a few keystrokes by defining a small function that does both and bind that one to a key instead: > (defun my-haskell-run-devel () > "Reloads the current module and then hot-reloads code via DevelMain.update." > (interactive) > (haskell-process-load-file) > (haskell-process-reload-devel-main)) > > (define-key haskell-mode-map (kbd "C-c m") 'my-haskell-run-devel) -} {- $intro While working on a project you may want to have your code running in the background and restart parts of it as you make changes. The premise of this introduction is that you already have such a project, for example a web application, and that you use a persistent GHCi session (either standalone or built into your editor). To use this library in your project create a module conventionally named @DevelMain@ that exports an action conventionally named @update@: > module DevelMain (update) where > > import Rapid > > update :: IO () > update = > rapid 0 $ \r -> > -- We'll list our components here shortly. > pure () The idea is that within a GHCi session you run this @update@ action whenever you want to reload your project during development. In the simplest case, like in a web application, your project consists of a single HTTP server thread that is just restarted each time you reload. Here is an example using the Snap Framework: > import qualified Data.Text as T > import Rapid > import Snap.Core > import Snap.Http.Server > > update = > rapid 0 $ \r -> > restart r "webserver" $ > quickHttpServe (writeText (T.pack "Hello world!")) Once you run @update@ in a GHCi session, a server is started (port 8000) that keeps running in the background, even when you reload modules. The REPL is fully responsive, so you can continue working. When you want to apply the changes you have made, you reload the @DevelMain@ module and run @update@ again. To see this in action, change the text string in the example, reload the module and then run @update@. Also observe that nothing is changed until you actually run @update@. When you want to stop a running background thread, replace 'restart' within the @update@ action by 'stop' and run @update@. The action given to 'stop' is actually ignored. It only takes the action argument for your convenience. You can run multiple threads at the same time and also have threads that are not restarted during a reload, but are only started and then kept running: > import MyProject.MyDatabase > import MyProject.MyBackgroundWorker > import MyProject.MyWebServer > import Rapid > > update = > rapid 0 $ \r -> do > start r "database" myDatabase > start r "worker" myBackgroundWorker > restart r "webserver" myWebServer Usually you would put 'restart' in front of the component that you are currently working on, while using 'start' with all others. Note that even though you are working on the code in @MyProject.MyWebServer@ you are always reloading the @DevelMain@ module. There is nothing wrong with loading and reloading other modules, but only this module gives you access to your @update@ action. -} {- $reusing Mutable references as introduced in the previous section can also be used to shorten the development cycle in the case when an expensive resource has to be created. As an example imagine that you need to parse a huge file into a data structure. You can keep the result of that in memory across reloads. Example with parsing JSON: > import Control.Exception > import Data.Aeson > import qualified Data.ByteString as B > > update = > rapid 0 $ \r -> > value <- createRef r "file" $ > B.readFile "blah.json" >>= > either (throwIO . userError) pure . eitherDecode > > -- You can now reuse 'value' across reloads. If you want to recreate the value at some point, you can just change 'createRef' to 'writeRef' and then run @update@. Keep in mind to change it back @createRef@ afterward. Use 'deleteRef' to remove values you no longer need, so they can be garbage-collected. -} {- $safety It's easy to crash your GHCi session with this library. In order to prevent that, you must follow these rules: * Do not change your service name type (the type argument of 'Rapid', i.e. the second argument to 'restart', 'start' and 'stop') within a session. The simplest way to do that is to resist the temptation to define a custom name type, and just use strings instead. If you do change the name type, you should restart GHCi. * Be careful with mutable variable created with 'createRef': If the value type changes (e.g. constructors or fields were changed), the variable must be recreated, for example by using 'writeRef' once. This most likely entails restarting all threads that were using the variable. Again the safest option is to just restart GHCi. * If any package in the current environment changes (especially this library itself), for example by updating a package via @cabal@ or @stack@, the @update@ action is likely to crash or go wrong in subtle ways due to binary incompatibility. If packages change, restart GHCi. * __This library is a development tool! Do not even think of using it to hot-reload in a productive environment!__ There are much safer and more appropriate ways to hot-reload code in production, for example by using a plugin system. The reason for this unsafety is that the underlying /foreign-store/ library is itself very unsafe in nature and requires that we maintain binary compatibility. This library hides most of that unsafety, but still requires that you follow the rules above. Please take the last rule seriously and never ever use this library in production! If something goes wrong during a reload, we do not get a convenient run-time exception; we get a memory violation, which can cause anything from a segfault to a remotely exploitable security hole. -}