meerschaum 2.9.5__py3-none-any.whl → 3.0.0rc2__py3-none-any.whl

meerschaum/api/dash/tokens.py

@@ -0,0 +1,606 @@
+#! /usr/bin/env python3
+# -*- coding: utf-8 -*-
+# vim:fenc=utf-8
+
+"""
+Dash utility functions for constructing tokens components.
+"""
+
+from __future__ import annotations
+
+from datetime import datetime, timezone, timedelta
+from typing import Any, List
+
+import meerschaum as mrsm
+from meerschaum.api import debug, CHECK_UPDATE, get_api_connector
+from meerschaum.api.dash.connectors import get_web_connector
+from meerschaum.api.dash.components import alert_from_success_tuple
+from meerschaum.api.dash.sessions import get_user_from_session
+from meerschaum.utils.typing import WebState, SuccessTuple, List, Tuple
+from meerschaum.utils.packages import attempt_import, import_html, import_dcc
+from meerschaum.utils.misc import interval_str
+from meerschaum.utils.dtypes import value_is_null, round_time
+from meerschaum._internal.static import STATIC_CONFIG
+from meerschaum.core import Token
+from meerschaum.utils.daemon import get_new_daemon_name
+dcc, html = import_dcc(check_update=CHECK_UPDATE), import_html(check_update=CHECK_UPDATE)
+dbc = attempt_import('dash_bootstrap_components', lazy=False, check_update=CHECK_UPDATE)
+
+
+def get_tokens_table(session_id: Optional[str] = None) -> Tuple[dbc.Table, List[dbc.Alert]]:
+    """
+    Return the main tokens table.
+    """
+    conn = get_api_connector()
+    user = get_user_from_session(session_id) if session_id is not None else None
+    alerts = []
+    try:
+        tokens = conn.get_tokens(user=user, debug=debug)
+    except Exception as e:
+        tokens = []
+        alerts = [alert_from_success_tuple((False, f"Failed to fetch tokens from '{conn}':\n{e}"))]
+
+    if not tokens:
+        return tokens, alerts
+
+    table_header = [
+        html.Thead(
+            html.Tr([
+                html.Th("Label"),
+                html.Th("Client ID"),
+                html.Th("Created"),
+                html.Th("Expires in"),
+                html.Th("Is Valid"),
+                html.Th(""),
+            ]),
+        ),
+    ]
+
+    rows = [
+        html.Tr([
+            html.Td(str(token.label)),
+            html.Td(html.Pre(str(token.id))),
+            html.Td(get_creation_string(token).replace('Created ', '')),
+            html.Td(get_expiration_string(token).replace('Expires in ', '')),
+            html.Td("✅" if token.is_valid else "❌"),
+            html.Td([
+                dbc.Button(
+                    html.B("⠇"),
+                    color='link',
+                    size='sm',
+                    id={'type': 'tokens-context-button', 'index': str(token.id)},
+                    style={'text-decoration': 'none'},
+                ),
+                build_manage_token_popover(token),
+                build_edit_token_modal(token),
+                build_invalidate_token_modal(token),
+                build_delete_token_modal(token),
+            ]),
+        ])
+        for token in tokens
+    ]
+
+    table_body = [html.Tbody(rows)]
+    table = dbc.Table(table_header + table_body)
+    return table, alerts
+
+
+def build_manage_token_popover(token: Token) -> dbc.Popover:
+    """
+    Return the "Manage token" popover.
+    """
+    return dbc.Popover(
+        [
+            dbc.PopoverHeader(["Manage token"]),
+            dbc.PopoverBody([
+                dbc.ButtonGroup(
+                    ([
+                        dbc.Button(
+                            "Edit",
+                            outline=True,
+                            color='light',
+                            id={
+                                'type': 'tokens-edit-button',
+                                'index': str(token.id),
+                            },
+                        ),
+                        dbc.Button(
+                            "Invalidate",
+                            outline=True,
+                            color='warning',
+                            id={
+                                'type': 'tokens-invalidate-button',
+                                'index': str(token.id),
+                            },
+                        ),
+                    ] if token.is_valid else []) + [
+                    dbc.Button(
+                        "Delete",
+                        color='danger',
+                        outline=True,
+                        id={
+                            'type': 'tokens-delete-button',
+                            'index': str(token.id),
+                        },
+                    ),
+                ]),
+            ]),
+        ],
+        body=True,
+        trigger='legacy',
+        autohide=True,
+        id={'type': 'tokens-context-popover', 'index': str(token.id)},
+        target={'type': 'tokens-context-button', 'index': str(token.id)},
+    )
+
+
+def build_edit_token_modal(token: Token) -> dbc.Modal:
+    """
+    Return the Modal for editing the token.
+    """
+    return dbc.Modal(
+        [
+            dbc.ModalHeader([
+                html.H4([
+                    "Edit token ",
+                    html.B(str(token.label))
+                ]),
+            ]),
+            dbc.ModalBody([
+                html.Div(id={'type': 'tokens-edit-alerts-div', 'index': str(token.id)}),
+
+                dbc.Form([
+                    dbc.Row(
+                        ([
+                            dbc.Label("Name", width='auto'),
+                            dbc.Col(
+                                [
+                                    dbc.Input(
+                                        placeholder="Enter token's label",
+                                        value=str(token.label),
+                                        id={
+                                            'type': 'tokens-name-input',
+                                            'index': str(token.id),
+                                        },
+                                    ),
+                                ],
+                                className="me-3",
+                            ),
+                            dbc.Label("Expiration", width='auto'),
+                            dbc.Col(
+                                dcc.DatePickerSingle(
+                                    date=(
+                                        token.expiration.to_pydatetime()
+                                        if hasattr(token.expiration, 'to_pydatetime')
+                                        else token.expiration
+                                    ),
+                                    clearable=True,
+                                    min_date_allowed=datetime.today().date(),
+                                    display_format="YYYY-MM-DD",
+                                    id={
+                                        'type': 'tokens-expiration-datepickersingle',
+                                        'index': str(token.id),
+                                    },
+                                )
+                            ),
+                        ] if token.is_valid else []) + [
+                            dbc.Col([
+                            ]),
+                        ],
+                        className='g-2',
+                    ),
+                    html.Br(),
+                    dbc.Row([
+                    ]),
+                    html.Div([
+                        dbc.Button(
+                            "Deselect all",
+                            size='sm',
+                            color='link',
+                            id={
+                                'type': "tokens-deselect-scopes-button",
+                                'index': str(token.id),
+                            },
+                            style={'text-decoration': 'none'},
+                        ),
+                        html.Br(),
+                        dbc.Row([
+                            dbc.Label("Scopes", width='auto'),
+                            dbc.Col([
+                                dbc.Checklist(
+                                    options=[
+                                        {"label": scope, "value": scope}
+                                        for scope in STATIC_CONFIG['tokens']['scopes']
+                                    ],
+                                    value=token.scopes,
+                                    id={
+                                        'type': "tokens-scopes-checklist",
+                                        'index': str(token.id),
+                                    },
+                                    style={'columnCount': 3},
+                                ),
+                            ]),
+                        ]),
+                    ] if token.is_valid else [], id={
+                        'type': 'tokens-scopes-checklist-div',
+                        'index': str(token.id),
+                    }),
+                ]),
+
+            ]),
+            dbc.ModalFooter(
+                [
+                    html.Small(str(token.id)),
+                    dbc.Button(
+                        "Submit",
+                        id={'type': 'tokens-edit-submit-button', 'index': str(token.id)},
+                    ),
+                ],
+                className='d-flex justify-content-between',
+            ),
+        ],
+        size='lg',
+        is_open=False,
+        id={'type': 'tokens-edit-modal', 'index': str(token.id)},
+    )
+
+
+def build_invalidate_token_modal(token: Token) -> dbc.Modal:
+    """
+    Return the Invalidate token modal.
+    """
+    return dbc.Modal(
+        [
+            dbc.ModalHeader([
+                html.H4([
+                    "Invalidate token ",
+                    html.B(token.label),
+                    "?"
+                ]),
+            ]),
+            dbc.ModalBody([
+                html.Div(
+                    id={
+                        'type': 'tokens-invalidate-alerts-div',
+                        'index': str(token.id)
+                    },
+                ),
+                html.P(
+                    [
+                        "Are you sure you want to invalidate token ",
+                        html.B(token.label),
+                        " (",
+                        html.I(str(token.id)),
+                        ")?",
+                    ],
+                ),
+                html.P([html.B("This action cannot be undone!")]),
+            ]),
+            dbc.ModalFooter([
+                dbc.Button(
+                    "Invalidate",
+                    color='danger',
+                    id={
+                        'type': 'tokens-invalidate-confirm-button',
+                        'index': str(token.id),
+                    },
+                ),
+            ]),
+        ],
+        id={
+            'type': 'tokens-invalidate-modal',
+            'index': str(token.id),
+        },
+    )
+
+
+def build_delete_token_modal(token: Token) -> dbc.Modal:
+    """
+    Return the delete token modal.
+    """
+    return dbc.Modal(
+        [
+            dbc.ModalHeader([
+                html.H4([
+                    "Delete token ",
+                    html.B(token.label),
+                    "?"
+                ]),
+            ]),
+            dbc.ModalBody([
+                html.Div(
+                    id={
+                        'type': 'tokens-delete-alerts-div',
+                        'index': str(token.id)
+                    },
+                ),
+                html.P(
+                    [
+                        "Are you sure you want to delete token ",
+                        html.B(token.label),
+                        " (",
+                        html.I(str(token.id)),
+                        ")?",
+                    ],
+                ),
+                html.P([html.B("This action cannot be undone!")]),
+            ]),
+            dbc.ModalFooter([
+                dbc.Button(
+                    "Delete",
+                    color='danger',
+                    id={
+                        'type': 'tokens-delete-confirm-button',
+                        'index': str(token.id),
+                    },
+                ),
+            ]),
+        ],
+        id={
+            'type': 'tokens-delete-modal',
+            'index': str(token.id),
+        },
+    )
+
+
+def get_tokens_cards(session_id: Optional[str] = None) -> Tuple[List[dbc.Card], List[dbc.Alert]]:
+    """
+    Return the cards and alerts for tokens.
+    """
+    cards, alerts = [], [] 
+    conn = get_api_connector()
+    user = get_user_from_session(session_id) if session_id is not None else None
+    try:
+        tokens = conn.get_tokens(user=user, debug=debug)
+    except Exception as e:
+        tokens = []
+        alerts = [alert_from_success_tuple((False, f"Failed to fetch tokens from '{conn}':\n{e}"))]
+
+    for token in tokens:
+        try:
+            cards.append(
+                dbc.Card([
+                    dbc.CardHeader(
+                        [
+                            html.H5(token.label),
+                        ]
+                    ),
+                    dbc.CardBody(
+                        [
+                            html.Code(str(token.id), style={'color': '#999999'}),
+                        ]
+                    ),
+                    dbc.CardFooter(
+                        [
+                            html.P(
+                                get_creation_string(token),
+                                style={'color': '#999999'},
+                            ),
+                            html.P(
+                                get_expiration_string(token),
+                                style={'color': '#999999'},
+                            ),
+                        ]
+                    ),
+                ])
+            )
+        except Exception as e:
+            alerts.append(
+                alert_from_success_tuple((False, f"Failed to load metadata for token:\n{e}"))
+            )
+
+    return cards, alerts
+
+
+def get_creation_string(token: mrsm.core.Token) -> str:
+    """
+    Return the formatted string to represent the token's creation timestamp.
+    """
+    creation = token.creation
+    if value_is_null(str(creation)):
+        return ''
+    now = datetime.now(timezone.utc)
+    return 'Created ' + interval_str(creation - now, round_unit=True)
+
+
+def get_expiration_string(token: mrsm.core.Token) -> str:
+    """
+    Return the formatted string to represent the token's expiration timestamp.
+    """
+    expiration = token.expiration
+    if value_is_null(str(expiration)):
+        return 'Does not expire'
+    now = datetime.now(timezone.utc)
+    return 'Expires in ' + interval_str(expiration - now, round_unit=True)
+
+
+def build_tokens_register_input_modal() -> dbc.Modal:
+    """
+    Return the layout for the tokens register input modal.
+    """
+    now = datetime.now(timezone.utc)
+    default_expiration_days = mrsm.get_config(
+        'system', 'api', 'tokens', 'default_expiration_days',
+    ) or 366
+    default_expiration = round_time(
+        now + timedelta(days=default_expiration_days),
+        timedelta(days=1),
+    )
+    min_date_allowed = round_time(now + timedelta(days=1), timedelta(days=1))
+
+    return [
+        dbc.ModalHeader(html.H4("Register Token")),
+        dbc.ModalBody([
+            dbc.Form([
+                dbc.Row(
+                    [
+                        dbc.Label("Name", width='auto'),
+                        dbc.Col(
+                            [
+                                dbc.Input(
+                                    placeholder="Enter token's label",
+                                    value=get_new_daemon_name(),
+                                    id='tokens-name-input'
+                                ),
+                            ],
+                            className="me-3",
+                        ),
+                        dbc.Label("Expiration", width='auto'),
+                        dbc.Col(
+                            dcc.DatePickerSingle(
+                                date=default_expiration,
+                                clearable=True,
+                                min_date_allowed=min_date_allowed,
+                                display_format="YYYY-MM-DD",
+                                id='tokens-expiration-datepickersingle',
+                            )
+                        ),
+                        dbc.Col(
+                            dbc.Switch(
+                                id="tokens-toggle-scopes-switch",
+                                label="Grant all scopes",
+                                value=True,
+                            ),
+                            className="me-3",
+                        ),
+                    ],
+                    className='g-2',
+                ),
+                html.Br(),
+                dbc.Row([
+                ]),
+                html.Div([
+                    dbc.Button(
+                        "Deselect all",
+                        size='sm',
+                        color='link',
+                        id="tokens-deselect-scopes-button",
+                        style={'text-decoration': 'none'},
+                    ),
+                    html.Br(),
+                    dbc.Row([
+                        dbc.Label("Scopes", width='auto'),
+                        dbc.Col([
+                            dbc.Checklist(
+                                options=[
+                                    {"label": scope, "value": scope}
+                                    for scope in STATIC_CONFIG['tokens']['scopes']
+                                ],
+                                value=list(STATIC_CONFIG['tokens']['scopes']),
+                                id="tokens-scopes-checklist",
+                                style={'columnCount': 3},
+                            ),
+                        ]),
+                    ]),
+                ], id='tokens-scopes-checklist-div', style={'display': 'none'}),
+            ]),
+        ]),
+        dbc.ModalFooter([
+            dbc.Button('Register', id='tokens-register-button'),
+        ]),
+    ]
+
+
+def build_register_table_from_token(token: Token) -> dbc.Table:
+    """
+    Return a table with the token's metadata.
+    """
+    table_header = [html.Thead(html.Tr([html.Th("Attribute"), html.Th("Value")]))]
+    table_header = []
+    pre_style = {'white-space': 'pre-wrap', 'word-break': 'break-all'}
+    rows = [
+        html.Tr([
+            html.Td(html.B("Client ID")),
+            html.Td(html.Pre(
+                str(token.id),
+                style=pre_style,
+                id='token-id-pre',
+            )),
+        ]),
+        html.Tr([
+            html.Td(html.B("Client Secret")),
+            html.Td(html.Pre(
+                str(token.secret),
+                style=pre_style,
+                id='token-secret-pre',
+            )),
+        ]),
+        html.Tr([
+            html.Td(html.B("API Key")),
+            html.Td(html.Pre(token.get_api_key(), style=pre_style)),
+        ]),
+        html.Tr([
+            html.Td(html.B("Expiration")),
+            html.Td(
+                html.Pre(token.expiration.isoformat(), style=pre_style)
+                if token.expiration is not None
+                else "Does not expire"
+            ),
+        ]),
+        html.Tr([
+            html.Td(html.B("Scopes")),
+            html.Td(html.P(' '.join(token.scopes), style={'word-break': 'normal'})),
+        ]),
+        html.Tr([
+            html.Td(html.B("User")),
+            html.Td(token.user.username if token.user is not None else ""),
+        ]),
+    ]
+    table_body = [html.Tbody(rows)]
+    table = dbc.Table(
+        table_header + table_body,
+        id='tokens-register-table',
+    )
+    return table
+
+
+def build_tokens_register_output_modal(token: Token) -> List[Any]:
+    """
+    Return the layout for the tokens register output modal.
+    """
+    success, msg = token.register(debug=debug)
+    header_text = (
+        "Registered token "
+        if success
+        else "Failed to register token "
+    )
+    body_children = (
+        [
+            dbc.Stack(
+                [
+                    html.Div([
+                        html.P(html.B("Copy the token's details to dismiss.")),
+                        html.P(html.I("The secret and API key will not be shown again.")),
+                    ]),
+                    html.Div([
+                        dbc.Button(
+                            "Copy token details",
+                            id='tokens-register-copy-button',
+                        ),
+                    ], className="ms-auto"),
+                    html.Div([
+                        dcc.Clipboard(id="tokens-register-clipboard"),
+                    ]),
+                ],
+                direction='horizontal',
+                gap=2,
+            ),
+            build_register_table_from_token(token),
+        ]
+        if success
+        else alert_from_success_tuple((False, msg))
+    )
+    return [
+        dbc.ModalHeader(
+            html.H4([header_text, html.B(token.label)]),
+            close_button=False,
+        ),
+        dbc.ModalBody(body_children),
+        dbc.ModalFooter([
+            dbc.Button(
+                "Close",
+                id='tokens-close-register-output-modal-button',
+                disabled=True,
+            ),
+        ]),
+    ]

meerschaum/api/dash/websockets.py

@@ -8,7 +8,7 @@ Functions for interacting via WebSockets.
 
 import asyncio, sys
 from meerschaum.api._websockets import websockets
-from meerschaum.config.static import STATIC_CONFIG
+from meerschaum._internal.static import STATIC_CONFIG
 
 def ws_url_from_href(href: str) -> str:
     """

meerschaum/api/dash/webterm.py

@@ -31,6 +31,10 @@ def get_webterm(state: WebState) -> Tuple[Any, Any]:
     """
     Start the webterm and return its iframe.
     """
+    from meerschaum.api import _include_webterm
+    if not _include_webterm:
+        return console_div, []
+
     session_id = state['session-store.data'].get('session-id', None)
     username = get_username_from_session(session_id)
     if not is_session_authenticated(session_id):

meerschaum/api/models/__init__.py

@@ -6,6 +6,26 @@
 Create and manipulate SQL tables with ORM
 """
 
-from meerschaum.api.models._pipes import MetaPipe
-#  from meerschaum.api.models._metrics import Metric
-#  from meerschaum.api.models._locations import Location
+import meerschaum as mrsm
+import meerschaum.models
+
+from meerschaum.api.models._pipes import (
+    FetchPipesKeysResponseModel,
+    SyncPipeRequestModel,
+)
+from meerschaum.api.models._actions import SuccessTupleResponseModel
+from meerschaum.api.models._tokens import (
+    GetTokensResponseModel,
+    RegisterTokenResponseModel,
+    RegisterTokenRequestModel,
+    GetTokenResponseModel,
+)
+
+__all__ = (
+    'FetchPipesKeysResponseModel',
+    'SyncPipeRequestModel',
+    'SuccessTupleResponseModel',
+    'RegisterTokenResponseModel',
+    'RegisterTokenRequestModel',
+    'GetTokenResponseModel',
+)

meerschaum/api/models/_actions.py

@@ -0,0 +1,22 @@
+#! /usr/bin/env python3
+# vim:fenc=utf-8
+
+"""
+Define actions response models.
+"""
+
+from typing import Tuple
+from pydantic import RootModel, ConfigDict
+from meerschaum.utils.typing import SuccessTuple
+
+
+class SuccessTupleResponseModel(RootModel[SuccessTuple]):
+    """
+    A response model for a tuple of a boolean and a string.
+    E.g. `[true, "Success"]`
+    """
+    model_config = ConfigDict(
+        json_schema_extra={
+            'example': [True, "Success"],
+        },
+    )