extend transferrer protocol to send progress bar total size updates

New protocol is not back-compat with old one, but it's never been
released so that's ok.

Messages/Progress.hs

@@ -25,6 +25,7 @@ import Messages.Internal
 import qualified System.Console.Regions as Regions
 import qualified System.Console.Concurrent as Console
 import Control.Monad.IO.Class (MonadIO)
+import Data.IORef
 
 {- Class of things from which a size can be gotten to display a progress
  - meter. -}
@@ -115,8 +116,16 @@ metered' st othermeter msize showoutput a = go st
 			a meter (combinemeter m)
 		| otherwise = nometer
 	go (MessageState { outputType = SerializedOutput h _ }) = do
-		liftIO $ outputSerialized h $ StartProgressMeter msize
-		meter <- liftIO $ mkMeter msize $ \_ _ _old new ->
+		liftIO $ outputSerialized h $ BeginProgressMeter msize
+		szv <- liftIO $ newIORef msize
+		meter <- liftIO $ mkMeter msize $ \_ msize' _old new -> do
+			case msize' of
+				Just sz | msize' /= msize -> do
+					psz <- readIORef szv
+					when (msize' /= psz) $ do
+						writeIORef szv msize'
+						outputSerialized h $ UpdateProgressMeterTotalSize sz
+				_ -> noop
 			outputSerialized h $ UpdateProgressMeter $
 				meterBytesProcessed new
 		m <- liftIO $ rateLimitMeterUpdate minratelimit meter $

Messages/Serialized.hs

@@ -20,7 +20,7 @@ import Messages
 import Messages.Internal
 import Messages.Progress
 import qualified Messages.JSON as JSON
-import Utility.Metered (BytesProcessed)
+import Utility.Metered (BytesProcessed, setMeterTotalSize)
 
 import Control.Monad.IO.Class (MonadIO)
 
@@ -63,12 +63,12 @@ relaySerializedOutput getso sendsor meterreport runannex = go Nothing
 					outputSerialized h $ JSONObject b
 				_ -> q
 			loop st
-		Left (StartProgressMeter sz) -> do
+		Left (BeginProgressMeter sz) -> do
 			ost <- runannex (Annex.getState Annex.output)
 			-- Display a progress meter while running, until
 			-- the meter ends or a final value is returned.
 			metered' ost Nothing sz (runannex showOutput) 
-				(\_meter meterupdate -> loop (Just meterupdate))
+				(\meter meterupdate -> loop (Just (meter, meterupdate)))
 				>>= \case
 					Right r -> return (Right r)
 					-- Continue processing serialized
@@ -80,12 +80,18 @@ relaySerializedOutput getso sendsor meterreport runannex = go Nothing
 			return (Left st)
 		Left (UpdateProgressMeter n) -> do
 			case st of
-				Just meterupdate -> do
+				Just (_, meterupdate) -> do
 					meterreport (Just n)
 					liftIO $ meterupdate n
 				Nothing -> noop
 			loop st
-		Left StartPrompt -> do
+		Left (UpdateProgressMeterTotalSize sz) -> do
+			case st of
+				Just (meter, _) -> liftIO $
+					setMeterTotalSize meter sz
+				Nothing -> noop
+			loop st
+		Left BeginPrompt -> do
 			prompter <- runannex mkPrompter
 			v <- prompter $ do
 				sendsor ReadyPrompt