Finally catching? - Polysemy

instance Member (Final IO) r => Catch.MonadCatch (Sem r) where
  catch m handler = withStrategicToFinal @IO $ do
    m' <- runS m
    st <- getInitialStateS
    handler' <- bindS handler
    pure $ m' `Catch.catch` \e -> handler' $ e <$ st

instance Member (Final IO) r => Catch.MonadCatch (Sem r) where
  catch m handler =
    withWeavingToFinal \ s wv _ ->
      wv (m <$ s) `Catch.catch` \ e ->
        wv (handler e <$ s)

mask :: Member (Final IO) r
     => ((forall a. Sem r a -> Sem r a) -> Sem r b)
     -> Sem r b
mask f = withWeavingToFinal $ \st lower ins ->
  lower $ f (\z -> fmap (fromJust . ins) $ embedFinal @IO $ X.mask $ \y -> y $ lower $ z <$ st) <$ st

mask ::
  forall r a .
  Member (Final IO) r =>
  ((forall x . Sem r x -> Sem r x) -> Sem r a) ->
  Sem r a
mask f =
  withStrategicToFinal @IO $
    controlS' \ lowerStrat ->
      X.mask \ restore ->
        lowerStrat do
          let
            res :: forall x . Sem r x -> Sem r x
            res mx =
              controlF \ lower -> restore (lower mx)
          controlS \ lowerOuter -> lowerOuter (f res)