88from  .._utils  import  AsyncCallbacks , CancelCallback , wrap_async 
99from  ..types  import  MISSING , MISSING_TYPE 
1010from  ._button  import  BOOKMARK_ID 
11- from  ._restore_state  import  RestoreContextState 
12- from  ._save_state  import  ShinySaveState 
11+ from  ._restore_state  import  RestoreState 
12+ from  ._save_state  import  BookmarkState 
1313from  ._types  import  BookmarkStore 
1414
1515# TODO: Barret - Bookmark state 
@@ -192,8 +192,7 @@ def _get_bookmark_exclude(self) -> list[str]:
192192    def  on_bookmark (
193193        self ,
194194        callback : (
195-             Callable [[ShinySaveState ], None ]
196-             |  Callable [[ShinySaveState ], Awaitable [None ]]
195+             Callable [[BookmarkState ], None ] |  Callable [[BookmarkState ], Awaitable [None ]]
197196        ),
198197        / ,
199198    ) ->  CancelCallback :
@@ -266,8 +265,7 @@ async def do_bookmark(self) -> None:
266265    def  on_restore (
267266        self ,
268267        callback : (
269-             Callable [[RestoreContextState ], None ]
270-             |  Callable [[RestoreContextState ], Awaitable [None ]]
268+             Callable [[RestoreState ], None ] |  Callable [[RestoreState ], Awaitable [None ]]
271269        ),
272270    ) ->  CancelCallback :
273271        """ 
@@ -281,8 +279,7 @@ def on_restore(
281279    def  on_restored (
282280        self ,
283281        callback : (
284-             Callable [[RestoreContextState ], None ]
285-             |  Callable [[RestoreContextState ], Awaitable [None ]]
282+             Callable [[RestoreState ], None ] |  Callable [[RestoreState ], Awaitable [None ]]
286283        ),
287284    ) ->  CancelCallback :
288285        """ 
@@ -402,8 +399,7 @@ def _get_bookmark_exclude(self) -> list[str]:
402399    def  on_bookmark (
403400        self ,
404401        callback : (
405-             Callable [[ShinySaveState ], None ]
406-             |  Callable [[ShinySaveState ], Awaitable [None ]]
402+             Callable [[BookmarkState ], None ] |  Callable [[BookmarkState ], Awaitable [None ]]
407403        ),
408404        / ,
409405    ) ->  CancelCallback :
@@ -419,17 +415,15 @@ def on_bookmarked(
419415    def  on_restore (
420416        self ,
421417        callback : (
422-             Callable [[RestoreContextState ], None ]
423-             |  Callable [[RestoreContextState ], Awaitable [None ]]
418+             Callable [[RestoreState ], None ] |  Callable [[RestoreState ], Awaitable [None ]]
424419        ),
425420    ) ->  CancelCallback :
426421        return  self ._on_restore_callbacks .register (wrap_async (callback ))
427422
428423    def  on_restored (
429424        self ,
430425        callback : (
431-             Callable [[RestoreContextState ], None ]
432-             |  Callable [[RestoreContextState ], Awaitable [None ]]
426+             Callable [[RestoreState ], None ] |  Callable [[RestoreState ], Awaitable [None ]]
433427        ),
434428    ) ->  CancelCallback :
435429        return  self ._on_restored_callbacks .register (wrap_async (callback ))
@@ -457,14 +451,14 @@ async def do_bookmark(self) -> None:
457451
458452        try :
459453            # ?withLogErrors 
460-             from  ..bookmark ._bookmark  import  ShinySaveState 
454+             from  ..bookmark ._bookmark  import  BookmarkState 
461455            from  ..session  import  session_context 
462456
463-             async  def  root_state_on_save (state : ShinySaveState ) ->  None :
457+             async  def  root_state_on_save (state : BookmarkState ) ->  None :
464458                with  session_context (self ._session_root ):
465459                    await  self ._on_bookmark_callbacks .invoke (state )
466460
467-             root_state  =  ShinySaveState (
461+             root_state  =  BookmarkState (
468462                input = self ._session_root .input ,
469463                exclude = self ._get_bookmark_exclude (),
470464                on_save = root_state_on_save ,
@@ -541,7 +535,7 @@ def __init__(self, session_proxy: SessionProxy):
541535        # The goal of this method is to save the scope's values. All namespaced inputs 
542536        # will already exist within the `root_state`. 
543537        @self ._root_bookmark .on_bookmark  
544-         async  def  scoped_on_bookmark (root_state : ShinySaveState ) ->  None :
538+         async  def  scoped_on_bookmark (root_state : BookmarkState ) ->  None :
545539            return  await  self ._scoped_on_bookmark (root_state )
546540
547541        from  ..session  import  session_context 
@@ -557,7 +551,7 @@ async def scoped_on_bookmarked(url: str) -> None:
557551        ns_prefix  =  str (self ._ns  +  self ._ns ._sep )
558552
559553        @self ._root_bookmark .on_restore  
560-         async  def  scoped_on_restore (restore_state : RestoreContextState ) ->  None :
554+         async  def  scoped_on_restore (restore_state : RestoreState ) ->  None :
561555            if  self ._on_restore_callbacks .count () ==  0 :
562556                return 
563557
@@ -567,22 +561,22 @@ async def scoped_on_restore(restore_state: RestoreContextState) -> None:
567561                await  self ._on_restore_callbacks .invoke (scoped_restore_state )
568562
569563        @self ._root_bookmark .on_restored  
570-         async  def  scoped_on_restored (restore_state : RestoreContextState ) ->  None :
564+         async  def  scoped_on_restored (restore_state : RestoreState ) ->  None :
571565            if  self ._on_restored_callbacks .count () ==  0 :
572566                return 
573567
574568            scoped_restore_state  =  restore_state ._state_within_namespace (ns_prefix )
575569            with  session_context (self ._session_proxy ):
576570                await  self ._on_restored_callbacks .invoke (scoped_restore_state )
577571
578-     async  def  _scoped_on_bookmark (self , root_state : ShinySaveState ) ->  None :
572+     async  def  _scoped_on_bookmark (self , root_state : BookmarkState ) ->  None :
579573        # Exit if no user-defined callbacks. 
580574        if  self ._on_bookmark_callbacks .count () ==  0 :
581575            return 
582576
583-         from  ..bookmark ._bookmark  import  ShinySaveState 
577+         from  ..bookmark ._bookmark  import  BookmarkState 
584578
585-         scoped_state  =  ShinySaveState (
579+         scoped_state  =  BookmarkState (
586580            input = self ._session_root .input ,
587581            exclude = self ._root_bookmark .exclude ,
588582            on_save = None ,
@@ -621,8 +615,7 @@ def _create_effects(self) -> NoReturn:
621615    def  on_bookmark (
622616        self ,
623617        callback : (
624-             Callable [[ShinySaveState ], None ]
625-             |  Callable [[ShinySaveState ], Awaitable [None ]]
618+             Callable [[BookmarkState ], None ] |  Callable [[BookmarkState ], Awaitable [None ]]
626619        ),
627620        / ,
628621    ) ->  CancelCallback :
@@ -658,17 +651,15 @@ def store(self) -> BookmarkStore:
658651    def  on_restore (
659652        self ,
660653        callback : (
661-             Callable [[RestoreContextState ], None ]
662-             |  Callable [[RestoreContextState ], Awaitable [None ]]
654+             Callable [[RestoreState ], None ] |  Callable [[RestoreState ], Awaitable [None ]]
663655        ),
664656    ) ->  CancelCallback :
665657        return  self ._on_restore_callbacks .register (wrap_async (callback ))
666658
667659    def  on_restored (
668660        self ,
669661        callback : (
670-             Callable [[RestoreContextState ], None ]
671-             |  Callable [[RestoreContextState ], Awaitable [None ]]
662+             Callable [[RestoreState ], None ] |  Callable [[RestoreState ], Awaitable [None ]]
672663        ),
673664    ) ->  CancelCallback :
674665        return  self ._on_restored_callbacks .register (wrap_async (callback ))
@@ -697,8 +688,7 @@ def _get_bookmark_exclude(self) -> NoReturn:
697688    def  on_bookmark (
698689        self ,
699690        callback : (
700-             Callable [[ShinySaveState ], None ]
701-             |  Callable [[ShinySaveState ], Awaitable [None ]]
691+             Callable [[BookmarkState ], None ] |  Callable [[BookmarkState ], Awaitable [None ]]
702692        ),
703693    ) ->  NoReturn :
704694        raise  NotImplementedError (
@@ -728,8 +718,7 @@ async def do_bookmark(self) -> NoReturn:
728718    def  on_restore (
729719        self ,
730720        callback : (
731-             Callable [[RestoreContextState ], None ]
732-             |  Callable [[RestoreContextState ], Awaitable [None ]]
721+             Callable [[RestoreState ], None ] |  Callable [[RestoreState ], Awaitable [None ]]
733722        ),
734723    ) ->  NoReturn :
735724        raise  NotImplementedError (
@@ -739,8 +728,7 @@ def on_restore(
739728    def  on_restored (
740729        self ,
741730        callback : (
742-             Callable [[RestoreContextState ], None ]
743-             |  Callable [[RestoreContextState ], Awaitable [None ]]
731+             Callable [[RestoreState ], None ] |  Callable [[RestoreState ], Awaitable [None ]]
744732        ),
745733    ) ->  NoReturn :
746734        raise  NotImplementedError (
0 commit comments