139 lines
3.7 KiB
Python
139 lines
3.7 KiB
Python
import click
|
|
from rich import print as rprint
|
|
from rich.panel import Panel
|
|
from rich.table import Table
|
|
|
|
from ..config import Config
|
|
from ..database import Database
|
|
from ..generator import ScriptGenerator
|
|
from ..playback import WorkflowPlayback
|
|
|
|
|
|
@click.group()
|
|
def workflow():
|
|
"""Manage workflows."""
|
|
pass
|
|
|
|
|
|
@workflow.command()
|
|
def list():
|
|
"""List all workflows."""
|
|
db = Database()
|
|
workflows = db.get_all_workflows()
|
|
db.close()
|
|
|
|
if not workflows:
|
|
rprint("[yellow]No workflows recorded yet[/yellow]")
|
|
return
|
|
|
|
table = Table(title="Workflows")
|
|
table.add_column("ID", style="cyan")
|
|
table.add_column("Name", style="magenta")
|
|
table.add_column("Commands", style="green")
|
|
table.add_column("Usage", style="yellow")
|
|
|
|
for w in workflows:
|
|
table.add_row(str(w.id), w.name, str(len(w.commands)), str(w.usage_count))
|
|
|
|
rprint(Panel(table, title=f"{len(workflows)} Workflows"))
|
|
|
|
|
|
@workflow.command()
|
|
@click.argument("workflow_id", type=int)
|
|
def show(workflow_id):
|
|
"""Show workflow details."""
|
|
db = Database()
|
|
workflow = db.get_workflow(workflow_id)
|
|
db.close()
|
|
|
|
if not workflow:
|
|
rprint(f"[yellow]Workflow {workflow_id} not found[/yellow]")
|
|
return
|
|
|
|
table = Table(title=f"Workflow: {workflow.name}")
|
|
table.add_column("Property", style="cyan")
|
|
table.add_column("Value", style="magenta")
|
|
|
|
table.add_row("ID", str(workflow.id))
|
|
table.add_row("Description", workflow.description or "N/A")
|
|
table.add_row("Commands", str(len(workflow.commands)))
|
|
table.add_row("Usage Count", str(workflow.usage_count))
|
|
|
|
rprint(Panel(table, title="Workflow Details"))
|
|
|
|
|
|
@workflow.command()
|
|
@click.argument("workflow_id", type=int)
|
|
@click.option("--speed", "-s", default=1.0, help="Playback speed multiplier")
|
|
@click.option("--dry-run", "-n", is_flag=True, help="Dry run (no execution)")
|
|
def play(workflow_id, speed, dry_run):
|
|
"""Playback a workflow."""
|
|
db = Database()
|
|
playback = WorkflowPlayback(db=db)
|
|
|
|
try:
|
|
result = playback.playback(
|
|
workflow_id,
|
|
speed=speed,
|
|
dry_run=dry_run,
|
|
)
|
|
|
|
rprint(f"[green]Workflow playback completed[/green]")
|
|
rprint(f" Executed: {result['executed']}")
|
|
rprint(f" Succeeded: {result['succeeded']}")
|
|
rprint(f" Failed: {result['failed']}")
|
|
except ValueError as e:
|
|
rprint(f"[yellow]{str(e)}[/yellow]")
|
|
|
|
db.close()
|
|
|
|
|
|
@workflow.command()
|
|
@click.argument("workflow_id", type=int)
|
|
def preview(workflow_id):
|
|
"""Preview workflow commands."""
|
|
db = Database()
|
|
playback = WorkflowPlayback(db=db)
|
|
|
|
try:
|
|
preview = playback.preview_workflow(workflow_id)
|
|
|
|
table = Table(title=f"Workflow Preview")
|
|
table.add_column("#", style="cyan")
|
|
table.add_column("Command", style="magenta")
|
|
|
|
for item in preview:
|
|
table.add_row(str(item["index"]), item["command"])
|
|
|
|
rprint(Panel(table, title="Preview"))
|
|
except ValueError as e:
|
|
rprint(f"[yellow]{str(e)}[/yellow]")
|
|
|
|
db.close()
|
|
|
|
|
|
@workflow.command()
|
|
@click.argument("workflow_id", type=int)
|
|
@click.option("--name", "-n", help="Script name")
|
|
@click.option("--output", "-o", help="Output directory")
|
|
def generate(workflow_id, name, output):
|
|
"""Generate script from workflow."""
|
|
db = Database()
|
|
generator = ScriptGenerator(db=db)
|
|
workflow = db.get_workflow(workflow_id)
|
|
|
|
if not workflow:
|
|
rprint(f"[yellow]Workflow {workflow_id} not found[/yellow]")
|
|
db.close()
|
|
return
|
|
|
|
script_path = generator.generate_script(
|
|
workflow,
|
|
name=name,
|
|
output_dir=output,
|
|
)
|
|
|
|
db.close()
|
|
|
|
rprint(f"[green]Script generated: {script_path}[/green]")
|