sweatstack 0.56.0__py3-none-any.whl → 0.57.0__py3-none-any.whl

sweatstack/streamlit.py

@@ -114,6 +114,9 @@ class StreamlitAuth:
 
         self.redirect_uri = redirect_uri or os.environ.get("SWEATSTACK_REDIRECT_URI")
 
+        self._proxy_mode = False
+        self._logout_uri = None
+
         self.api_key = st.session_state.get("sweatstack_api_key")
         self.refresh_token = st.session_state.get("sweatstack_refresh_token")
         self.client = Client(
@@ -124,13 +127,97 @@ class StreamlitAuth:
             client_secret=self.client_secret,
         )
 
+    @classmethod
+    def behind_proxy(
+        cls,
+        redirect_uri: str,
+        header_name: str = "X-SweatStack-Token",
+        logout_uri: str = "/logout",
+    ) -> "StreamlitAuth":
+        """Create a StreamlitAuth instance for use behind a proxy.
+
+        Use this method when your Streamlit app runs behind a proxy that handles
+        authentication and passes the SweatStack access token via an HTTP header.
+
+        Args:
+            redirect_uri: The URI to redirect to after login (used by proxy).
+            header_name: The HTTP header name containing the access token.
+                         Defaults to "X-SweatStack-Token".
+            logout_uri: The URI to redirect to for logout.
+                        Defaults to "/logout".
+
+        Returns:
+            StreamlitAuth: An instance configured for proxy mode.
+
+        Example:
+            auth = StreamlitAuth.behind_proxy(
+                redirect_uri="https://myapp.example.com/app",
+            )
+
+            if not auth.is_authenticated():
+                st.error("Missing authentication header")
+                st.stop()
+
+            activities = auth.client.get_activities()
+        """
+        instance = cls(redirect_uri=redirect_uri)
+        instance._proxy_mode = True
+        instance._logout_uri = logout_uri
+
+        token = st.context.headers.get(header_name)
+        if token:
+            instance.api_key = token
+            instance.client = Client(token, streamlit_compatible=True)
+
+        return instance
+
+    def _show_styled_link_button(self, label: str, url: str):
+        """Displays a styled link button with hover effects.
+
+        Args:
+            label: Text to display on the button.
+            url: The URL to navigate to when clicked.
+        """
+        st.markdown(
+            f"""
+            <style>
+                .animated-button {{
+                }}
+                .animated-button:hover {{
+                    transform: scale(1.05);
+                }}
+                .animated-button:active {{
+                    transform: scale(1);
+                }}
+            </style>
+            <a href="{url}"
+                target="_top"
+                class="animated-button"
+                style="display: inline-block;
+                    padding: 10px 20px;
+                    background-color: #EF2B2D;
+                    color: white;
+                    text-decoration: none;
+                    border-radius: 6px;
+                    border: none;
+                    transition: all 0.3s ease;
+                    cursor: pointer;"
+                >{label}</a>
+            """,
+            unsafe_allow_html=True,
+        )
+
     def logout_button(self):
         """Displays a logout button and handles user logout.
 
-        When clicked, this button clears the stored API key from session state,
-        resets the client, and triggers a Streamlit rerun to update the UI.
+        In standard mode, clears the stored API key from session state,
+        resets the client, and triggers a Streamlit rerun.
+
+        In proxy mode, displays a styled link that redirects to the logout URI.
         """
-        if st.button("Logout"):
+        if self._proxy_mode:
+            self._show_styled_link_button("Logout", self._logout_uri)
+        elif st.button("Logout"):
             self.api_key = None
             self.refresh_token = None
             self.client = Client(streamlit_compatible=True)
@@ -151,34 +238,7 @@ class StreamlitAuth:
         authorization_url = self.get_authorization_url()
         login_label = login_label or "Connect with SweatStack"
         if not self._running_on_streamlit_cloud():
-            st.markdown(
-                f"""
-                <style>
-                    .animated-button {{
-                    }}
-                    .animated-button:hover {{
-                        transform: scale(1.05);
-                    }}
-                    .animated-button:active {{
-                        transform: scale(1);
-                    }}
-                </style>
-                <a href="{authorization_url}"
-                    target="_top"
-                    class="animated-button"
-                    style="display: inline-block;
-                        padding: 10px 20px;
-                        background-color: #EF2B2D;
-                        color: white;
-                        text-decoration: none;
-                        border-radius: 6px;
-                        border: none;
-                        transition: all 0.3s ease;
-                        cursor: pointer;"
-                    >{login_label}</a>
-                """,
-                unsafe_allow_html=True,
-            )
+            self._show_styled_link_button(login_label, authorization_url)
         else:
             st.link_button(login_label, authorization_url)
 
@@ -275,12 +335,23 @@ class StreamlitAuth:
         to the Streamlit app with an authorization code, which is exchanged for an
         access token.
 
+        In proxy mode, this method only shows the login button if not authenticated.
+        The proxy handles the OAuth callback and token exchange.
+
         Args:
             login_label: The label to display on the login button. Defaults to "Login with SweatStack".
 
         Returns:
             None
         """
+        if self._proxy_mode:
+            if self.is_authenticated():
+                if show_logout:
+                    self.logout_button()
+            else:
+                self._show_sweatstack_login(login_label)
+            return
+
         if self.is_authenticated():
             if not st.session_state.get("sweatstack_auth_toast_shown", False):
                 st.toast("SweatStack authentication successful!", icon="✅")

{sweatstack-0.56.0.dist-info → sweatstack-0.57.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sweatstack
-Version: 0.56.0
+Version: 0.57.0
 Summary: The official Python client for SweatStack
 Author-email: Aart Goossens <aart@gssns.io>
 Requires-Python: >=3.9

{sweatstack-0.56.0.dist-info → sweatstack-0.57.0.dist-info}/RECORD

@@ -7,11 +7,11 @@ sweatstack/jupyterlab_oauth2_startup.py,sha256=YcjXvzeZ459vL_dCkFi1IxX_RNAu80ZX9
 sweatstack/openapi_schemas.py,sha256=VvquBdbssdB9D1KeJYQCx51hy1Df4SS0PjzGWXcUaew,46221
 sweatstack/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sweatstack/schemas.py,sha256=Xh9E8DjFx5NIEBnVqS6ixFVb0E06ANZbdOlnMofCpZw,4481
-sweatstack/streamlit.py,sha256=fVTgTyb8a2c7IC-lCs32ocILvyj1dFXP4iQaTV9D4Is,17287
+sweatstack/streamlit.py,sha256=wnabWhife9eMAdkECPjRKkzE82KZoi_H8YzucZl_m9s,19604
 sweatstack/sweatshell.py,sha256=MYLNcWbOdceqKJ3S0Pe8dwHXEeYsGJNjQoYUXpMTftA,333
 sweatstack/utils.py,sha256=AwHRdC1ziOZ5o9RBIB21Uxm-DoClVRAJSVvgsmSmvps,1801
 sweatstack/Sweat Stack examples/Getting started.ipynb,sha256=k2hiSffWecoQ0VxjdpDcgFzBXDQiYEebhnAYlu8cgX8,6335204
-sweatstack-0.56.0.dist-info/METADATA,sha256=N732fwIaJz4h5uVfOCZUwSaMbaLk7CB1leB6dHpvpw4,852
-sweatstack-0.56.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-sweatstack-0.56.0.dist-info/entry_points.txt,sha256=kCzOUQI3dqbTpEYqtgYDeiKFaqaA7BMlV6D24BMzCFU,208
-sweatstack-0.56.0.dist-info/RECORD,,
+sweatstack-0.57.0.dist-info/METADATA,sha256=0amcxzS2bzmC9qy3v_9wZ9_DH2BBAkqfvN8gECa7BZ0,852
+sweatstack-0.57.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+sweatstack-0.57.0.dist-info/entry_points.txt,sha256=kCzOUQI3dqbTpEYqtgYDeiKFaqaA7BMlV6D24BMzCFU,208
+sweatstack-0.57.0.dist-info/RECORD,,

{sweatstack-0.56.0.dist-info → sweatstack-0.57.0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.27.0
+Generator: hatchling 1.28.0
 Root-Is-Purelib: true
 Tag: py3-none-any