hap-cli 0.5.0__py3-none-any.whl

hap_cli/commands/worksheet_cmd.py

@@ -0,0 +1,342 @@
+"""CLI commands for worksheet operations."""
+
+import json
+
+import click
+
+from hap_cli.context import pass_context
+from hap_cli.core import worksheet as ws_mod
+from hap_cli.utils.formatting import output_json, output_table, output_kv
+
+
+@click.group()
+def worksheet():
+    """Worksheet operations."""
+    pass
+
+
+@worksheet.command("info")
+@click.argument("worksheet_id")
+@click.option("--app-id", "-a", default="", help="Application ID")
+@pass_context
+def worksheet_info(ctx, worksheet_id, app_id):
+    """Get worksheet information."""
+    try:
+        session = ctx.get_session()
+        data = ws_mod.get_worksheet_info(session, worksheet_id, app_id)
+        ctx.output(data, lambda d: output_kv(d))
+    except Exception as e:
+        ctx.handle_error(e)
+
+
+@worksheet.command("fields")
+@click.argument("worksheet_id")
+@click.option("--app-id", "-a", default="", help="Application ID")
+@pass_context
+def worksheet_fields(ctx, worksheet_id, app_id):
+    """List worksheet fields/controls."""
+    try:
+        session = ctx.get_session()
+        controls = ws_mod.get_worksheet_controls(session, worksheet_id, app_id)
+        ctx.output(
+            controls,
+            lambda d: output_table(
+                d,
+                ["controlId", "controlName", "type", "required"],
+                ["Control ID", "Name", "Type", "Required"],
+            ),
+        )
+    except Exception as e:
+        ctx.handle_error(e)
+
+
+@worksheet.command("views")
+@click.argument("worksheet_id")
+@click.option("--app-id", "-a", default="", help="Application ID")
+@pass_context
+def worksheet_views(ctx, worksheet_id, app_id):
+    """List worksheet views."""
+    try:
+        session = ctx.get_session()
+        views = ws_mod.get_worksheet_views(session, worksheet_id, app_id)
+        ctx.output(
+            views,
+            lambda d: output_table(
+                d,
+                ["viewId", "name", "viewType"],
+                ["View ID", "Name", "Type"],
+            ),
+        )
+    except Exception as e:
+        ctx.handle_error(e)
+
+
+# ── Worksheet CRUD ───────────────────────────────────────────────────────
+
+
+@worksheet.command("create")
+@click.argument("app_id")
+@click.argument("name")
+@click.option("--project-id", default="", help="Project ID")
+@click.option("--section-id", default="", help="Section ID")
+@pass_context
+def worksheet_create(ctx, app_id, name, project_id, section_id):
+    """Create a new worksheet."""
+    try:
+        session = ctx.get_session()
+        data = ws_mod.create_worksheet(session, app_id, name, project_id, section_id)
+        ctx.output(data, lambda d: output_json(d))
+    except Exception as e:
+        ctx.handle_error(e)
+
+
+@worksheet.command("copy")
+@click.argument("worksheet_id")
+@click.argument("name")
+@click.option("--app-id", "-a", default="", help="Application ID")
+@click.option("--project-id", default="", help="Project ID")
+@click.option("--copy-rows", is_flag=True, default=False, help="Copy rows as well")
+@pass_context
+def worksheet_copy(ctx, worksheet_id, name, app_id, project_id, copy_rows):
+    """Copy a worksheet."""
+    try:
+        session = ctx.get_session()
+        data = ws_mod.copy_worksheet(session, worksheet_id, name, app_id, project_id, copy_rows)
+        ctx.output(data, lambda d: output_json(d))
+    except Exception as e:
+        ctx.handle_error(e)
+
+
+@worksheet.command("update")
+@click.argument("worksheet_id")
+@click.option("--alias", default="", help="New alias for the worksheet")
+@click.option("--desc", default="", help="New description for the worksheet")
+@click.option("--app-id", "-a", default="", help="Application ID")
+@pass_context
+def worksheet_update(ctx, worksheet_id, alias, desc, app_id):
+    """Update worksheet alias and/or description."""
+    try:
+        session = ctx.get_session()
+        data = ws_mod.update_worksheet(session, worksheet_id, alias, desc, app_id)
+        ctx.output(data, lambda d: output_json(d))
+    except Exception as e:
+        ctx.handle_error(e)
+
+
+# ── Field type reference ─────────────────────────────────────────────────
+
+
+@worksheet.command("field-types")
+@pass_context
+def worksheet_field_types(ctx):
+    """Show field type constants as a reference table."""
+    try:
+        rows = [{"name": k, "type_id": v} for k, v in ws_mod.FIELD_TYPES.items()]
+        ctx.output(
+            rows,
+            lambda d: output_table(d, ["name", "type_id"], ["Field Type", "Type ID"]),
+        )
+    except Exception as e:
+        ctx.handle_error(e)
+
+
+# ── Field/Control management ─────────────────────────────────────────────
+
+
+@worksheet.command("save-fields")
+@click.argument("worksheet_id")
+@click.option("--controls", required=True, help="JSON array of the complete controls list (get current list via 'fields' command first)")
+@pass_context
+def worksheet_save_fields(ctx, worksheet_id, controls):
+    """Save the complete fields/controls list for a worksheet.
+
+    This replaces add-field/edit-field/delete-field with a single full-replacement
+    approach: fetch current fields via 'fields' command, modify the list as needed
+    (add new entries, update existing ones, or remove entries), then pass the
+    complete modified list here.
+
+    Example workflow:
+      hap worksheet fields <worksheet_id> --json > controls.json
+      # edit controls.json as needed
+      hap worksheet save-fields <worksheet_id> --controls "$(cat controls.json)"
+    """
+    try:
+        session = ctx.get_session()
+        controls_data = json.loads(controls)
+        data = ws_mod.save_controls(session, worksheet_id, controls_data)
+        ctx.output(data, lambda d: output_json(d))
+    except Exception as e:
+        ctx.handle_error(e)
+
+
+# ── View management ──────────────────────────────────────────────────────
+
+
+@worksheet.command("add-view")
+@click.argument("worksheet_id")
+@click.argument("name")
+@click.option("--view-type", type=int, default=0, help="View type (default: 0)")
+@click.option("--app-id", "-a", default="", help="Application ID")
+@pass_context
+def worksheet_add_view(ctx, worksheet_id, name, view_type, app_id):
+    """Create a new worksheet view."""
+    try:
+        session = ctx.get_session()
+        data = ws_mod.save_view(session, worksheet_id, name, view_type, app_id=app_id)
+        ctx.output(data, lambda d: output_json(d))
+    except Exception as e:
+        ctx.handle_error(e)
+
+
+@worksheet.command("delete-view")
+@click.argument("worksheet_id")
+@click.argument("view_id")
+@click.option("--app-id", "-a", default="", help="Application ID")
+@pass_context
+def worksheet_delete_view(ctx, worksheet_id, view_id, app_id):
+    """Delete a worksheet view."""
+    try:
+        session = ctx.get_session()
+        data = ws_mod.delete_view(session, worksheet_id, view_id, app_id)
+        ctx.output(data, lambda d: output_json(d))
+    except Exception as e:
+        ctx.handle_error(e)
+
+
+@worksheet.command("copy-view")
+@click.argument("worksheet_id")
+@click.argument("view_id")
+@click.argument("name")
+@pass_context
+def worksheet_copy_view(ctx, worksheet_id, view_id, name):
+    """Copy a worksheet view."""
+    try:
+        session = ctx.get_session()
+        data = ws_mod.copy_view(session, worksheet_id, view_id, name)
+        ctx.output(data, lambda d: output_json(d))
+    except Exception as e:
+        ctx.handle_error(e)
+
+
+@worksheet.command("sort-views")
+@click.argument("worksheet_id")
+@click.argument("view_ids", nargs=-1, required=True)
+@pass_context
+def worksheet_sort_views(ctx, worksheet_id, view_ids):
+    """Reorder worksheet views. Pass view IDs in desired order."""
+    try:
+        session = ctx.get_session()
+        data = ws_mod.sort_views(session, worksheet_id, list(view_ids))
+        ctx.output(data, lambda d: output_json(d))
+    except Exception as e:
+        ctx.handle_error(e)
+
+
+# ── Business Rules ───────────────────────────────────────────────────────
+
+
+@worksheet.command("rules")
+@click.argument("worksheet_id")
+@click.option("--control-id", default="", help="Filter by control ID")
+@pass_context
+def worksheet_rules(ctx, worksheet_id, control_id):
+    """Get business rules for worksheet controls."""
+    try:
+        session = ctx.get_session()
+        data = ws_mod.get_control_rules(session, worksheet_id, control_id)
+        ctx.output(data, lambda d: output_json(d))
+    except Exception as e:
+        ctx.handle_error(e)
+
+
+@worksheet.command("save-rule")
+@click.argument("worksheet_id")
+@click.option("--rules", required=True, help="JSON array of rule definitions")
+@click.option("--control-id", default="", help="Control ID to associate rules with")
+@pass_context
+def worksheet_save_rule(ctx, worksheet_id, rules, control_id):
+    """Save business rules for worksheet controls."""
+    try:
+        session = ctx.get_session()
+        rules_data = json.loads(rules)
+        data = ws_mod.save_control_rule(session, worksheet_id, rules_data, control_id)
+        ctx.output(data, lambda d: output_json(d))
+    except Exception as e:
+        ctx.handle_error(e)
+
+
+# ── Custom Buttons ───────────────────────────────────────────────────────
+
+
+@worksheet.command("buttons")
+@click.argument("worksheet_id")
+@pass_context
+def worksheet_buttons(ctx, worksheet_id):
+    """Get custom action buttons for a worksheet."""
+    try:
+        session = ctx.get_session()
+        data = ws_mod.get_buttons(session, worksheet_id)
+        ctx.output(data, lambda d: output_json(d))
+    except Exception as e:
+        ctx.handle_error(e)
+
+
+@worksheet.command("save-button")
+@click.argument("worksheet_id")
+@click.option("--config", "btn_config", required=True, help="JSON object of button configuration")
+@click.option("--btn-id", default="", help="Button ID (for updating existing button)")
+@pass_context
+def worksheet_save_button(ctx, worksheet_id, btn_config, btn_id):
+    """Create or update a custom action button."""
+    try:
+        session = ctx.get_session()
+        btn_data = json.loads(btn_config)
+        data = ws_mod.save_button(session, worksheet_id, btn_data, btn_id)
+        ctx.output(data, lambda d: output_json(d))
+    except Exception as e:
+        ctx.handle_error(e)
+
+
+@worksheet.command("delete-button")
+@click.argument("worksheet_id")
+@click.argument("btn_id")
+@pass_context
+def worksheet_delete_button(ctx, worksheet_id, btn_id):
+    """Delete a custom action button."""
+    try:
+        session = ctx.get_session()
+        data = ws_mod.delete_button(session, worksheet_id, btn_id)
+        ctx.output(data, lambda d: output_json(d))
+    except Exception as e:
+        ctx.handle_error(e)
+
+
+# ── Feature Switches ────────────────────────────────────────────────────
+
+
+@worksheet.command("switches")
+@click.argument("worksheet_id")
+@pass_context
+def worksheet_switches(ctx, worksheet_id):
+    """Get feature switches for a worksheet."""
+    try:
+        session = ctx.get_session()
+        data = ws_mod.get_switches(session, worksheet_id)
+        ctx.output(data, lambda d: output_json(d))
+    except Exception as e:
+        ctx.handle_error(e)
+
+
+@worksheet.command("edit-switch")
+@click.argument("worksheet_id")
+@click.argument("switch_id", type=int)
+@click.argument("value", type=bool)
+@pass_context
+def worksheet_edit_switch(ctx, worksheet_id, switch_id, value):
+    """Edit a single feature switch."""
+    try:
+        session = ctx.get_session()
+        data = ws_mod.edit_switch(session, worksheet_id, switch_id, value)
+        ctx.output(data, lambda d: output_json(d))
+    except Exception as e:
+        ctx.handle_error(e)

hap_cli/context.py

@@ -0,0 +1,43 @@
+"""Shared CLI context for hap-cli."""
+
+import sys
+from typing import Optional
+
+import click
+
+from hap_cli.core.session import Session, SessionError, APIError
+from hap_cli.utils.formatting import output_json
+
+
+class Context:
+    """Shared CLI context."""
+
+    def __init__(self):
+        self.json_mode = False
+        self.session: Optional[Session] = None
+
+    def get_session(self) -> Session:
+        if self.session is None:
+            self.session = Session.load()
+        return self.session
+
+    def output(self, data, table_fn=None):
+        """Output data in JSON or human-readable format."""
+        if self.json_mode:
+            output_json(data)
+        elif table_fn:
+            table_fn(data)
+        else:
+            output_json(data)
+
+    def handle_error(self, e: Exception):
+        """Handle and display errors."""
+        msg = str(e)
+        if self.json_mode:
+            output_json({"error": msg, "type": type(e).__name__})
+        else:
+            click.echo(f"Error: {msg}", err=True)
+        sys.exit(1)
+
+
+pass_context = click.make_pass_decorator(Context, ensure=True)

hap_cli/core/__init__.py

@@ -0,0 +1 @@
+"""Core modules for hap-cli harness."""

hap_cli/core/ai.py

@@ -0,0 +1,133 @@
+"""AI assistant and service management for MingDAO HAP."""
+
+from typing import Any, Optional
+
+from hap_cli.core.session import Session
+
+
+# ── AI Assistant ─────────────────────────────────────────────────────────
+
+
+def get_assistants(
+    session: Session,
+    project_id: str,
+    page_index: int = 1,
+    page_size: int = 20,
+) -> dict[str, Any]:
+    """Get AI assistant list."""
+    return session.api_call(
+        "Assistant", "GetList",
+        {"projectId": project_id, "pageIndex": page_index, "pageSize": page_size},
+    )
+
+
+def get_assistant(
+    session: Session,
+    assistant_id: str,
+) -> dict[str, Any]:
+    """Get AI assistant detail."""
+    return session.api_call("Assistant", "Get", {"assistantId": assistant_id})
+
+
+def upsert_assistant(
+    session: Session,
+    config: dict[str, Any],
+) -> dict[str, Any]:
+    """Create or update an AI assistant."""
+    return session.api_call("Assistant", "Upsert", config)
+
+
+def delete_assistant(
+    session: Session,
+    assistant_id: str,
+) -> dict[str, Any]:
+    """Delete an AI assistant."""
+    return session.api_call("Assistant", "Delete", {"assistantId": assistant_id})
+
+
+def set_assistant_status(
+    session: Session,
+    assistant_id: str,
+    status: int,
+) -> dict[str, Any]:
+    """Enable or disable an AI assistant."""
+    return session.api_call(
+        "Assistant", "SetStatus",
+        {"assistantId": assistant_id, "status": status},
+    )
+
+
+# ── Knowledge Base ───────────────────────────────────────────────────────
+
+
+def get_knowledge_bases(
+    session: Session,
+    project_id: str,
+) -> dict[str, Any]:
+    """Get knowledge base list."""
+    return session.api_call("Assistant", "GetListKnowledgeBase", {"projectId": project_id})
+
+
+def get_knowledge_base(
+    session: Session,
+    knowledge_base_id: str,
+) -> dict[str, Any]:
+    """Get knowledge base detail."""
+    return session.api_call("Assistant", "GetKnowledgeBase", {"knowledgeBaseId": knowledge_base_id})
+
+
+def upsert_knowledge_base(
+    session: Session,
+    config: dict[str, Any],
+) -> dict[str, Any]:
+    """Create or update a knowledge base."""
+    return session.api_call("Assistant", "UpsertKnowledgeBase", config)
+
+
+def delete_knowledge_base(
+    session: Session,
+    knowledge_base_id: str,
+) -> dict[str, Any]:
+    """Delete a knowledge base."""
+    return session.api_call("Assistant", "DeleteKnowledgeBase", {"knowledgeBaseId": knowledge_base_id})
+
+
+# ── Chatbot Configuration ────────────────────────────────────────────────
+
+
+def get_chatbot_config(
+    session: Session,
+    process_id: str,
+) -> dict[str, Any]:
+    """Get chatbot configuration for a process/workflow."""
+    return session.workflow_call("getChatbotConfig", {"processId": process_id})
+
+
+def save_chatbot_config(
+    session: Session,
+    process_id: str,
+    config: dict[str, Any],
+) -> dict[str, Any]:
+    """Save chatbot configuration for a process/workflow."""
+    data: dict[str, Any] = {"processId": process_id}
+    data.update(config)
+    return session.workflow_call("saveChatbotConfig", data)
+
+
+# ── AI Service ───────────────────────────────────────────────────────────
+
+
+def get_ai_service_status(
+    session: Session,
+    project_id: str,
+) -> dict[str, Any]:
+    """Get AI service enable/disable status."""
+    return session.api_call("AIService", "GetAIServiceStatus", {"projectId": project_id})
+
+
+def get_models(
+    session: Session,
+    project_id: str,
+) -> dict[str, Any]:
+    """Get available AI models with developer info."""
+    return session.api_call("AIService", "GetDeveloperWithModes", {"projectId": project_id})