From 02ec8ea01254637facb30f77b7cb74be3b735c0d Mon Sep 17 00:00:00 2001
From: Joey Hess <joey@kitenet.net>
Date: Fri, 27 Jul 2012 15:33:24 -0400
Subject: much better webapp startup of the assistant

This avoids forking another process, avoids polling, fixes a race,
and avoids a rare forkProcess thread hang that I saw once time
when starting the webapp.
---
 Command/Watch.hs  |  2 +-
 Command/WebApp.hs | 43 ++++++++++++++-----------------------------
 2 files changed, 15 insertions(+), 30 deletions(-)

(limited to 'Command')

diff --git a/Command/Watch.hs b/Command/Watch.hs
index 744844c4d..61c859106 100644
--- a/Command/Watch.hs
+++ b/Command/Watch.hs
@@ -34,5 +34,5 @@ start :: Bool -> Bool -> Bool -> CommandStart
 start assistant foreground stopdaemon = notBareRepo $ do
 	if stopdaemon
 		then stopDaemon
-		else startDaemon assistant foreground -- does not return
+		else startDaemon assistant foreground Nothing -- does not return
 	stop
diff --git a/Command/WebApp.hs b/Command/WebApp.hs
index 7d0a310d4..1635ac044 100644
--- a/Command/WebApp.hs
+++ b/Command/WebApp.hs
@@ -12,12 +12,8 @@ import Command
 import Assistant
 import Utility.WebApp
 import Utility.Daemon (checkDaemon)
-import qualified Annex
 import Option
 
-import Control.Concurrent
-import System.Posix.Process
-
 def :: [Command]
 def = [withOptions [restartOption] $
         command "webapp" paramNothing seek "launch webapp"]
@@ -34,31 +30,20 @@ start restart = notBareRepo $ do
 	if restart
 		then do
 			stopDaemon
-			nuke =<< fromRepo gitAnnexPidFile
-			startassistant f
-		else unlessM (checkpid f) $
+			void $ liftIO . nukeFile =<< fromRepo gitAnnexPidFile
 			startassistant f
-	let url = "file://" ++ f
-	ifM (liftIO $ runBrowser url)
-		( stop
-		, error $ "failed to start web browser on url " ++ url
-		)
+		else ifM (checkpid <&&> checkshim f) $
+			( liftIO $ go f 
+			, startassistant f
+			)
+	stop
 	where
-		nuke f = void $ liftIO $ catchMaybeIO $ removeFile f
-		checkpid f = do
+		checkpid = do
 			pidfile <- fromRepo gitAnnexPidFile
-			liftIO $
-				doesFileExist f <&&> (isJust <$> checkDaemon pidfile)
-		startassistant f = do
-			nuke f
-			{- Fork a separate process to run the assistant,
-			 - with a copy of the Annex state. -}
-			state <- Annex.getState id
-			liftIO $ void $ forkProcess $
-				Annex.eval state $ startDaemon True False
-			waitdaemon f (1000 :: Int)
-		waitdaemon _ 0 = error "failed to start git-annex assistant"
-		waitdaemon f n = unlessM (checkpid f) $ do
-			-- wait 0.1 seconds before retry
-			liftIO $ threadDelay 100000
-			waitdaemon f (n - 1)
+			liftIO $ isJust <$> checkDaemon pidfile
+		checkshim f = liftIO $ doesFileExist f
+		startassistant = startDaemon True False . Just . go
+		go f = unlessM (runBrowser url) $
+			error $ "failed to start web browser on url " ++ url
+			where
+				url = "file://" ++ f
-- 
cgit v1.2.3