create-ironclaws 1.0.0

package/template/container/tools/gdrive_tool.py

@@ -0,0 +1,185 @@
+"""
+Standalone Google Drive tool for Argus agents.
+Uses OAuth refresh token — no google-auth library needed, just requests.
+
+Usage:
+  # List files in the standup folder
+  python3 gdrive_tool.py list
+
+  # List only files not yet in processed_transcripts.txt
+  python3 gdrive_tool.py list --new-only
+
+  # Read a Google Doc as plain text
+  python3 gdrive_tool.py read --file-id 1abc123...
+
+  # Mark a file as processed (appends ID to processed_transcripts.txt)
+  python3 gdrive_tool.py mark-processed --file-id 1abc123...
+
+Environment variables (required):
+  GOOGLE_CLIENT_ID
+  GOOGLE_CLIENT_SECRET
+  GOOGLE_REFRESH_TOKEN
+  GOOGLE_DRIVE_STANDUP_FOLDER_ID
+
+Optional:
+  GDRIVE_PROCESSED_FILE   path to processed transcripts list
+                          default: /workspace/group/processed_transcripts.txt
+"""
+
+import argparse
+import json
+import os
+import sys
+
+try:
+    import requests
+except ImportError:
+    print("ERROR: 'requests' is not installed.", file=sys.stderr)
+    sys.exit(1)
+
+TOKEN_URL = "https://oauth2.googleapis.com/token"
+DRIVE_API = "https://www.googleapis.com/drive/v3"
+
+PROCESSED_FILE = os.environ.get(
+    "GDRIVE_PROCESSED_FILE", "/workspace/group/processed_transcripts.txt"
+)
+
+
+def get_access_token():
+    """Get Google access token.
+
+    Uses GOOGLE_ACCESS_TOKEN injected by the NanoClaw host (short-lived, ~1h TTL).
+    The OAuth refresh credentials stay on the host — never in the container.
+    """
+    token = os.environ.get("GOOGLE_ACCESS_TOKEN", "").strip()
+    if not token:
+        print(
+            "ERROR: GOOGLE_ACCESS_TOKEN must be set. "
+            "The NanoClaw host generates this token — check that Google OAuth "
+            "credentials (GOOGLE_CLIENT_ID/SECRET/REFRESH_TOKEN) are in the host .env.",
+            file=sys.stderr,
+        )
+        sys.exit(1)
+    return token
+
+
+def get_folder_id():
+    folder_id = os.environ.get("GOOGLE_DRIVE_STANDUP_FOLDER_ID", "").strip()
+    if not folder_id:
+        print("ERROR: GOOGLE_DRIVE_STANDUP_FOLDER_ID must be set.", file=sys.stderr)
+        sys.exit(1)
+    return folder_id
+
+
+def load_processed():
+    """Return set of already-processed file IDs."""
+    if not os.path.exists(PROCESSED_FILE):
+        return set()
+    with open(PROCESSED_FILE) as f:
+        return set(line.strip() for line in f if line.strip())
+
+
+def mark_processed(file_id):
+    """Append a file ID to the processed list."""
+    processed = load_processed()
+    if file_id not in processed:
+        with open(PROCESSED_FILE, "a") as f:
+            f.write(file_id + "\n")
+        print(f"Marked {file_id} as processed.")
+    else:
+        print(f"{file_id} was already marked as processed.")
+
+
+def cmd_list(args):
+    token = get_access_token()
+    folder_id = get_folder_id()
+
+    resp = requests.get(
+        f"{DRIVE_API}/files",
+        headers={"Authorization": f"Bearer {token}"},
+        params={
+            "q": f"'{folder_id}' in parents and trashed=false",
+            "fields": "files(id,name,createdTime,mimeType)",
+            "orderBy": "createdTime asc",
+            "pageSize": 1000,
+            "supportsAllDrives": True,
+            "includeItemsFromAllDrives": True,
+        },
+        timeout=30,
+    )
+    resp.raise_for_status()
+    files = resp.json().get("files", [])
+
+    if args.new_only:
+        processed = load_processed()
+        files = [f for f in files if f["id"] not in processed]
+
+    if not files:
+        print("No files found." if not args.new_only else "No new files to process.")
+        return
+
+    print(json.dumps(files, indent=2))
+
+
+def cmd_read(args):
+    token = get_access_token()
+
+    # Try exporting as plain text (works for Google Docs)
+    resp = requests.get(
+        f"{DRIVE_API}/files/{args.file_id}/export",
+        headers={"Authorization": f"Bearer {token}"},
+        params={"mimeType": "text/plain"},
+        timeout=60,
+    )
+
+    if resp.status_code == 400:
+        # Not a Google Doc — try downloading directly
+        resp = requests.get(
+            f"{DRIVE_API}/files/{args.file_id}?alt=media",
+            headers={"Authorization": f"Bearer {token}"},
+            timeout=60,
+        )
+
+    resp.raise_for_status()
+    print(resp.text)
+
+
+def cmd_mark_processed(args):
+    mark_processed(args.file_id)
+
+
+def parse_args():
+    p = argparse.ArgumentParser(description="Google Drive tool for Argus agents")
+    sub = p.add_subparsers(dest="command")
+
+    l = sub.add_parser("list", help="List files in the standup folder")
+    l.add_argument("--new-only", action="store_true", help="Only show unprocessed files")
+
+    r = sub.add_parser("read", help="Read a Google Doc as plain text")
+    r.add_argument("--file-id", required=True)
+
+    m = sub.add_parser("mark-processed", help="Mark a file as processed")
+    m.add_argument("--file-id", required=True)
+
+    return p.parse_args()
+
+
+def main():
+    args = parse_args()
+    if not args.command:
+        print("ERROR: specify a command. Use --help for usage.", file=sys.stderr)
+        sys.exit(1)
+    try:
+        {"list": cmd_list, "read": cmd_read, "mark-processed": cmd_mark_processed}[
+            args.command
+        ](args)
+    except requests.HTTPError as e:
+        print(f"ERROR: Google Drive API error: {e}\n{e.response.text}", file=sys.stderr)
+        sys.exit(1)
+    except Exception as e:
+        print(f"ERROR: {e}", file=sys.stderr)
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()

package/template/container/tools/jira_tool.py

@@ -0,0 +1,433 @@
+"""
+Jira tool for Argus agents.
+
+Usage:
+  # Get next ticket to work on (highest priority, oldest first)
+  python3 jira_tool.py next
+
+  # Get specific ticket
+  python3 jira_tool.py get --issue IAI-42
+
+  # Search tickets
+  python3 jira_tool.py search --status "To Do"
+  python3 jira_tool.py search --status "In Progress" --assignee "user@example.com"
+
+  # Create a ticket
+  python3 jira_tool.py create --summary "Bug in BobAIA" --description "..." --type Bug
+  python3 jira_tool.py create --summary "Fix login" --type Bug --epic IAI-10
+
+  # Create an epic
+  python3 jira_tool.py create-epic --summary "Salesforce Integration" --description "..."
+  python3 jira_tool.py create-epic --summary "Salesforce Integration" --start-date 2026-04-15 --due-date 2026-06-30
+
+  # Search epics
+  python3 jira_tool.py search-epics
+  python3 jira_tool.py search-epics --query "Salesforce"
+
+  # Update a ticket
+  python3 jira_tool.py update --issue IAI-42 --status "In Progress"
+  python3 jira_tool.py update --issue IAI-42 --description "Updated description"
+  python3 jira_tool.py update --issue IAI-42 --comment "Working on this now"
+  python3 jira_tool.py update --issue IAI-42 --assignee-email "user@yourcompany.com"
+
+  # Find Jira account ID by email
+  python3 jira_tool.py user --email "user@yourcompany.com"
+
+Environment variables (required):
+  JIRA_BASE_URL      e.g. https://yourcompany.atlassian.net
+  JIRA_EMAIL         Atlassian account email
+  JIRA_API_TOKEN     Atlassian API token
+  JIRA_PROJECT_KEY   e.g. IAI
+"""
+
+import argparse
+import json
+import os
+import sys
+
+try:
+    import requests
+    from requests.auth import HTTPBasicAuth
+except ImportError:
+    print("ERROR: 'requests' is not installed. Run: pip3 install requests", file=sys.stderr)
+    sys.exit(1)
+
+
+PRIORITY_ORDER = {"Critical": 0, "High": 1, "Medium": 2, "Low": 3, "Lowest": 4}
+
+
+def get_client():
+    base_url = os.environ.get("JIRA_BASE_URL", "").rstrip("/")
+    project = os.environ.get("JIRA_PROJECT_KEY", "")
+
+    if not base_url or not project:
+        print(
+            "ERROR: JIRA_BASE_URL and JIRA_PROJECT_KEY must be set.",
+            file=sys.stderr,
+        )
+        sys.exit(1)
+
+    email = os.environ.get("JIRA_EMAIL", "")
+    token = os.environ.get("JIRA_API_TOKEN", "")
+    # auth is optional — if absent, OneCLI HTTPS proxy injects Authorization header.
+    auth = HTTPBasicAuth(email, token) if email and token else None
+    return base_url, auth, project
+
+
+def jira_get(base_url, auth, path, params=None):
+    resp = requests.get(
+        f"{base_url}/rest/api/3{path}",
+        auth=auth,
+        headers={"Accept": "application/json"},
+        params=params,
+        timeout=30,
+    )
+    resp.raise_for_status()
+    return resp.json()
+
+
+def jira_search(base_url, auth, jql, max_results=50, fields=None):
+    """POST-based search — returns the issues list directly."""
+    if fields is None:
+        fields = ["summary", "status", "priority", "issuetype",
+                  "assignee", "reporter", "description", "created"]
+    body = {"jql": jql, "maxResults": max_results, "fields": fields}
+    resp = requests.post(
+        f"{base_url}/rest/api/3/search/jql",
+        auth=auth,
+        headers={"Accept": "application/json", "Content-Type": "application/json"},
+        json=body,
+        timeout=30,
+    )
+    resp.raise_for_status()
+    return resp.json().get("issues", [])
+
+
+def jira_post(base_url, auth, path, body):
+    resp = requests.post(
+        f"{base_url}/rest/api/3{path}",
+        auth=auth,
+        headers={"Accept": "application/json", "Content-Type": "application/json"},
+        json=body,
+        timeout=30,
+    )
+    resp.raise_for_status()
+    return resp.json()
+
+
+def jira_put(base_url, auth, path, body):
+    resp = requests.put(
+        f"{base_url}/rest/api/3{path}",
+        auth=auth,
+        headers={"Accept": "application/json", "Content-Type": "application/json"},
+        json=body,
+        timeout=30,
+    )
+    resp.raise_for_status()
+    return resp
+
+
+def jira_transition(base_url, auth, issue_key, status_name):
+    """Move an issue to a new status by name."""
+    transitions = jira_get(base_url, auth, f"/issue/{issue_key}/transitions")
+    match = next(
+        (t for t in transitions["transitions"] if t["name"].lower() == status_name.lower()),
+        None,
+    )
+    if not match:
+        available = [t["name"] for t in transitions["transitions"]]
+        print(
+            f"ERROR: Status '{status_name}' not found. Available: {available}",
+            file=sys.stderr,
+        )
+        sys.exit(1)
+    jira_post(base_url, auth, f"/issue/{issue_key}/transitions", {"transition": {"id": match["id"]}})
+
+
+def format_issue(issue):
+    fields = issue.get("fields", {})
+    priority = (fields.get("priority") or {}).get("name", "None")
+    status = (fields.get("status") or {}).get("name", "?")
+    assignee = (fields.get("assignee") or {}).get("emailAddress", "unassigned")
+    reporter = (fields.get("reporter") or {}).get("emailAddress", "?")
+    issue_type = (fields.get("issuetype") or {}).get("name", "?")
+    summary = fields.get("summary", "?")
+    key = issue.get("key", "?")
+
+    # Extract plain text from description (Atlassian Document Format)
+    desc_raw = fields.get("description") or {}
+    description = extract_adf_text(desc_raw)
+
+    return {
+        "key": key,
+        "type": issue_type,
+        "status": status,
+        "priority": priority,
+        "summary": summary,
+        "description": description,
+        "reporter": reporter,
+        "assignee": assignee,
+        "url": f"{os.environ.get('JIRA_BASE_URL', '').rstrip('/')}/browse/{key}",
+    }
+
+
+def extract_adf_text(node):
+    """Extract plain text from Atlassian Document Format (ADF) JSON."""
+    if not node:
+        return ""
+    if isinstance(node, str):
+        return node
+    if node.get("type") == "text":
+        return node.get("text", "")
+    parts = []
+    for child in node.get("content", []):
+        parts.append(extract_adf_text(child))
+    return " ".join(p for p in parts if p).strip()
+
+
+def cmd_get(args):
+    base_url, auth, _ = get_client()
+    issue = jira_get(base_url, auth, f"/issue/{args.issue}")
+    print(json.dumps(format_issue(issue), indent=2))
+
+
+def cmd_next(args):
+    base_url, auth, project = get_client()
+    jql = f'project = {project} AND status = "To Do" ORDER BY priority ASC, created ASC'
+    issues = jira_search(base_url, auth, jql, max_results=10)
+    if not issues:
+        print("No tickets in To Do.")
+        return
+    def sort_key(i):
+        p = (i.get("fields", {}).get("priority") or {}).get("name", "Low")
+        return (PRIORITY_ORDER.get(p, 99), i.get("fields", {}).get("created", ""))
+    issues.sort(key=sort_key)
+    print(json.dumps(format_issue(issues[0]), indent=2))
+
+
+def cmd_search(args):
+    base_url, auth, project = get_client()
+    conditions = [f"project = {project}"]
+    if args.status:
+        conditions.append(f'status = "{args.status}"')
+    if args.assignee:
+        conditions.append(f'assignee = "{args.assignee}"')
+    jql = " AND ".join(conditions) + " ORDER BY priority ASC, created ASC"
+    issues = jira_search(base_url, auth, jql)
+    print(json.dumps([format_issue(i) for i in issues], indent=2))
+
+
+def cmd_create(args):
+    base_url, auth, project = get_client()
+
+    # Build description in ADF format
+    description_adf = {
+        "type": "doc",
+        "version": 1,
+        "content": [
+            {
+                "type": "paragraph",
+                "content": [{"type": "text", "text": args.description or ""}],
+            }
+        ],
+    }
+
+    body = {
+        "fields": {
+            "project": {"key": project},
+            "summary": args.summary,
+            "description": description_adf,
+            "issuetype": {"name": args.type or "Task"},
+        }
+    }
+
+    # Set reporter by account ID if provided
+    if args.reporter_account_id:
+        body["fields"]["reporter"] = {"id": args.reporter_account_id}
+
+    # Link to parent epic if provided
+    if args.epic:
+        body["fields"]["parent"] = {"key": args.epic}
+
+    result = jira_post(base_url, auth, "/issue", body)
+    key = result.get("key")
+    url = f"{base_url}/browse/{key}"
+    print(json.dumps({"key": key, "url": url, "epic": args.epic or None}))
+
+
+def cmd_create_epic(args):
+    base_url, auth, project = get_client()
+
+    description_adf = {
+        "type": "doc",
+        "version": 1,
+        "content": [
+            {
+                "type": "paragraph",
+                "content": [{"type": "text", "text": args.description or ""}],
+            }
+        ],
+    }
+
+    body = {
+        "fields": {
+            "project": {"key": project},
+            "summary": args.summary,
+            "description": description_adf,
+            "issuetype": {"name": "Epic"},
+        }
+    }
+
+    if args.reporter_account_id:
+        body["fields"]["reporter"] = {"id": args.reporter_account_id}
+
+    if args.start_date:
+        # customfield_10015 is the standard Epic start date in Jira Cloud
+        body["fields"]["customfield_10015"] = args.start_date
+
+    if args.due_date:
+        body["fields"]["duedate"] = args.due_date
+
+    result = jira_post(base_url, auth, "/issue", body)
+    key = result.get("key")
+    url = f"{base_url}/browse/{key}"
+    print(json.dumps({"key": key, "url": url, "type": "Epic"}))
+
+
+def cmd_search_epics(args):
+    base_url, auth, project = get_client()
+    conditions = [f"project = {project}", "issuetype = Epic"]
+    if args.query:
+        # Escape quotes in query for JQL
+        safe_query = args.query.replace('"', '\\"')
+        conditions.append(f'summary ~ "{safe_query}"')
+    jql = " AND ".join(conditions) + " ORDER BY created DESC"
+    issues = jira_search(base_url, auth, jql)
+    print(json.dumps([format_issue(i) for i in issues], indent=2))
+
+
+def cmd_update(args):
+    base_url, auth, _ = get_client()
+
+    if args.status:
+        jira_transition(base_url, auth, args.issue, args.status)
+        print(f"Moved {args.issue} to '{args.status}'")
+
+    if args.description:
+        description_adf = {
+            "type": "doc",
+            "version": 1,
+            "content": [
+                {
+                    "type": "paragraph",
+                    "content": [{"type": "text", "text": args.description}],
+                }
+            ],
+        }
+        jira_put(base_url, auth, f"/issue/{args.issue}", {"fields": {"description": description_adf}})
+        print(f"Updated description on {args.issue}")
+
+    if args.comment:
+        comment_adf = {
+            "type": "doc",
+            "version": 1,
+            "content": [
+                {
+                    "type": "paragraph",
+                    "content": [{"type": "text", "text": args.comment}],
+                }
+            ],
+        }
+        jira_post(base_url, auth, f"/issue/{args.issue}/comment", {"body": comment_adf})
+        print(f"Added comment to {args.issue}")
+
+    if args.assignee_email:
+        result = jira_get(base_url, auth, "/user/search", params={"query": args.assignee_email})
+        matches = [u for u in result if u.get("emailAddress", "").lower() == args.assignee_email.lower()]
+        if not matches:
+            print(f"ERROR: No Jira user found for email '{args.assignee_email}'", file=sys.stderr)
+            sys.exit(1)
+        account_id = matches[0]["accountId"]
+        jira_put(base_url, auth, f"/issue/{args.issue}", {"fields": {"assignee": {"id": account_id}}})
+        print(f"Assigned {args.issue} to {args.assignee_email}")
+
+
+def cmd_user(args):
+    """Find Jira account ID by email — used to set reporter/assignee."""
+    base_url, auth, _ = get_client()
+    result = jira_get(base_url, auth, "/user/search", params={"query": args.email})
+    matches = [u for u in result if u.get("emailAddress", "").lower() == args.email.lower()]
+    if not matches:
+        print(f"ERROR: No Jira user found for email '{args.email}'", file=sys.stderr)
+        sys.exit(1)
+    user = matches[0]
+    print(json.dumps({
+        "account_id": user.get("accountId"),
+        "display_name": user.get("displayName"),
+        "email": user.get("emailAddress"),
+    }))
+
+
+def parse_args():
+    p = argparse.ArgumentParser(description="Jira tool for Argus agents")
+    sub = p.add_subparsers(dest="command")
+
+    sub.add_parser("next", help="Get next To Do ticket by priority")
+
+    g = sub.add_parser("get", help="Get a specific ticket")
+    g.add_argument("--issue", required=True)
+
+    s = sub.add_parser("search", help="Search tickets")
+    s.add_argument("--status")
+    s.add_argument("--assignee")
+
+    c = sub.add_parser("create", help="Create a ticket")
+    c.add_argument("--summary", required=True)
+    c.add_argument("--description", default="")
+    c.add_argument("--type", default="Task", help="Bug, Task, Story, etc.")
+    c.add_argument("--reporter-account-id", dest="reporter_account_id", default=None)
+    c.add_argument("--epic", default=None, help="Parent epic key (e.g. IAI-10) to link this ticket under")
+
+    ce = sub.add_parser("create-epic", help="Create an epic")
+    ce.add_argument("--summary", required=True)
+    ce.add_argument("--description", default="")
+    ce.add_argument("--reporter-account-id", dest="reporter_account_id", default=None)
+    ce.add_argument("--start-date", dest="start_date", default=None, help="Start date (YYYY-MM-DD)")
+    ce.add_argument("--due-date", dest="due_date", default=None, help="Due date (YYYY-MM-DD)")
+
+    se = sub.add_parser("search-epics", help="Search epics in the project")
+    se.add_argument("--query", default=None, help="Search term to filter epic summaries")
+
+    u = sub.add_parser("update", help="Update a ticket")
+    u.add_argument("--issue", required=True)
+    u.add_argument("--status", help="Transition to this status")
+    u.add_argument("--description", help="Update description")
+    u.add_argument("--comment", help="Add a comment")
+    u.add_argument("--assignee-email", dest="assignee_email", help="Assign to this email address")
+
+    usr = sub.add_parser("user", help="Find Jira account ID by email")
+    usr.add_argument("--email", required=True)
+
+    return p.parse_args()
+
+
+def main():
+    args = parse_args()
+    if not args.command:
+        print("ERROR: specify a command. Use --help for usage.", file=sys.stderr)
+        sys.exit(1)
+    try:
+        {"next": cmd_next, "get": cmd_get, "search": cmd_search,
+         "create": cmd_create, "create-epic": cmd_create_epic,
+         "search-epics": cmd_search_epics,
+         "update": cmd_update, "user": cmd_user}[args.command](args)
+    except requests.HTTPError as e:
+        print(f"ERROR: Jira API error: {e}\n{e.response.text}", file=sys.stderr)
+        sys.exit(1)
+    except Exception as e:
+        print(f"ERROR: {e}", file=sys.stderr)
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()