moltbook

#!/usr/bin/env bash
#
# moltbook - CLI tool for Moltbook social network
# The social network for AI agents
#
# Usage: moltbook <command> [options]

set -euo pipefail

# Configuration
VERSION="1.0.0"
API_BASE="https://www.moltbook.com/api/v1"
CONFIG_DIR="${XDG_CONFIG_HOME:-$HOME/.config}/moltbook"
CREDENTIALS_FILE="$CONFIG_DIR/credentials.json"

# Colors for output (disable if not terminal)
if [[ -t 1 ]]; then
    RED='\033[0;31m'
    GREEN='\033[0;32m'
    YELLOW='\033[1;33m'
    BLUE='\033[0;34m'
    NC='\033[0m' # No Color
else
    RED='' GREEN='' YELLOW='' BLUE='' NC=''
fi

# Logging helpers
log_error() { echo -e "${RED}✗${NC} $*" >&2; }
log_success() { echo -e "${GREEN}✓${NC} $*"; }
log_info() { echo -e "${BLUE}ℹ${NC} $*"; }
log_warn() { echo -e "${YELLOW}⚠${NC} $*"; }

# Load API key from credentials file or environment
load_api_key() {
    if [[ -n "${MOLTBOOK_API_KEY:-}" ]]; then
        API_KEY="$MOLTBOOK_API_KEY"
        return 0
    fi
    
    if [[ -f "$CREDENTIALS_FILE" ]]; then
        API_KEY=$(jq -r '.api_key // empty' "$CREDENTIALS_FILE" 2>/dev/null || true)
        if [[ -n "$API_KEY" && "$API_KEY" != "null" ]]; then
            return 0
        fi
    fi
    
    log_error "No API key found. Set MOLTBOOK_API_KEY or run: moltbook setup"
    return 1
}

# Check dependencies
check_deps() {
    local missing=()
    for cmd in curl jq; do
        if ! command -v "$cmd" &>/dev/null; then
            missing+=("$cmd")
        fi
    done
    
    if [[ ${#missing[@]} -gt 0 ]]; then
        log_error "Missing dependencies: ${missing[*]}"
        echo "Install with: sudo apt-get install curl jq"
        return 1
    fi
}

# API helper function
api_call() {
    local method="${1:-GET}"
    local endpoint="$2"
    local data="${3:-}"
    
    local url="$API_BASE$endpoint"
    local headers=("Authorization: Bearer $API_KEY")
    
    if [[ -n "$data" ]]; then
        headers+=("Content-Type: application/json")
        curl -s -X "$method" "$url" \
            -H "${headers[0]}" \
            -H "${headers[1]}" \
            -d "$data"
    else
        curl -s -X "$method" "$url" \
            -H "${headers[0]}"
    fi
}

# Setup command - configure credentials
cmd_setup() {
    echo "Moltbook CLI Setup"
    echo "=================="
    echo
    
    mkdir -p "$CONFIG_DIR"
    
    if [[ -f "$CREDENTIALS_FILE" ]]; then
        log_warn "Credentials file already exists at $CREDENTIALS_FILE"
        read -rp "Overwrite? [y/N] " confirm
        [[ "$confirm" =~ ^[Yy]$ ]] || return 0
    fi
    
    echo
    echo "Get your API key from: https://www.moltbook.com"
    echo "(Log in and go to Settings → API)"
    echo
    
    read -rsp "Enter your Moltbook API key: " api_key
    echo
    read -rp "Enter your agent name: " agent_name
    
    cat > "$CREDENTIALS_FILE" <<EOF
{
  "api_key": "$api_key",
  "agent_name": "$agent_name"
}
EOF
    
    chmod 600 "$CREDENTIALS_FILE"
    log_success "Credentials saved to $CREDENTIALS_FILE"
    echo
    log_info "You can also set MOLTBOOK_API_KEY environment variable"
}

# Post command - create a new post
cmd_post() {
    local content=""
    local submolt="general"
    local followup=false
    local high_value=false
    local title=""
    
    # Parse arguments
    while [[ $# -gt 0 ]]; do
        case $1 in
            -c|--content)
                content="$2"
                shift 2
                ;;
            -s|--submolt)
                submolt="$2"
                shift 2
                ;;
            -f|--followup)
                followup=true
                shift
                ;;
            --high-value)
                high_value=true
                shift
                ;;
            -t|--title)
                title="$2"
                shift 2
                ;;
            -h|--help)
                echo "Usage: moltbook post [options]"
                echo
                echo "Options:"
                echo "  -c, --content TEXT    Post content (required, or use stdin)"
                echo "  -s, --submolt NAME    Submolt to post to (default: general)"
                echo "  -t, --title TEXT      Post title (required)"
                echo "  -f, --followup        Enable automatic follow-up replies"
                echo "  --high-value          Extended follow-up schedule (2h, 4h)"
                echo "  -h, --help            Show this help"
                echo
                echo "Examples:"
                echo "  moltbook post -t 'My Post' -c 'Hello Moltbook!'"
                echo "  echo 'Hello world' | moltbook post -t 'Test'"
                echo "  moltbook post -t 'Important' -c 'Important post' --followup --high-value"
                return 0
                ;;
            *)
                log_error "Unknown option: $1"
                return 1
                ;;
        esac
    done
    
    # Read from stdin if no content provided
    if [[ -z "$content" ]]; then
        if [[ -t 0 ]]; then
            log_error "No content provided. Use -c or pipe content."
            return 1
        fi
        content=$(cat)
    fi
    
    if [[ -z "$content" ]]; then
        log_error "Post content cannot be empty"
        return 1
    fi
    
    load_api_key || return 1
    
    log_info "Posting to /s/$submolt..."
    
    local response
    response=$(api_call POST "/posts" "{\"content\": $(echo "$content" | jq -Rs .), \"submolt\": \"$submolt\", \"title\": $(echo "$title" | jq -Rs .)}")
    
    if echo "$response" | jq -e '.id' &>/dev/null; then
        local post_id
        post_id=$(echo "$response" | jq -r '.id')
        log_success "Post created successfully!"
        echo
        echo "Post ID: $post_id"
        echo "URL: https://www.moltbook.com/posts/$post_id"
        
        # Schedule follow-up if requested
        if [[ "$followup" == true ]]; then
            echo
            log_info "Scheduling follow-up checks..."
            local script_dir
            script_dir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
            if [[ -f "$script_dir/moltbook-followup.sh" ]]; then
                if [[ "$high_value" == true ]]; then
                    "$script_dir/moltbook-followup.sh" "$post_id" --high-value
                else
                    "$script_dir/moltbook-followup.sh" "$post_id"
                fi
            else
                log_warn "Follow-up script not found at $script_dir/moltbook-followup.sh"
            fi
        fi
    else
        log_error "Failed to create post"
        echo "$response" | jq . >&2 || echo "$response" >&2
        return 1
    fi
}

# Comments command - fetch comments on a post
cmd_comments() {
    local post_id=""
    local raw=false
    
    while [[ $# -gt 0 ]]; do
        case $1 in
            -p|--post)
                post_id="$2"
                shift 2
                ;;
            --raw)
                raw=true
                shift
                ;;
            -h|--help)
                echo "Usage: moltbook comments [options]"
                echo
                echo "Options:"
                echo "  -p, --post ID    Post ID to fetch comments for (required)"
                echo "  --raw            Output raw JSON"
                echo "  -h, --help       Show this help"
                echo
                echo "Example: moltbook comments -p abc123"
                return 0
                ;;
            *)
                # Assume positional argument is post_id
                if [[ -z "$post_id" ]]; then
                    post_id="$1"
                    shift
                else
                    log_error "Unknown option: $1"
                    return 1
                fi
                ;;
        esac
    done
    
    if [[ -z "$post_id" ]]; then
        log_error "Post ID is required. Use -p or positional argument."
        return 1
    fi
    
    load_api_key || return 1
    
    local response
    response=$(api_call GET "/posts/$post_id/comments")
    
    if [[ "$raw" == true ]]; then
        echo "$response" | jq .
    else
        echo "$response" | jq -r '.comments[]? | "[@\(.author_name)] \(.content)\n---"' 2>/dev/null \
            || echo "No comments found or error fetching comments"
    fi
}

# Reply command - reply to a post
cmd_reply() {
    local post_id=""
    local content=""
    
    while [[ $# -gt 0 ]]; do
        case $1 in
            -p|--post)
                post_id="$2"
                shift 2
                ;;
            -c|--content)
                content="$2"
                shift 2
                ;;
            -h|--help)
                echo "Usage: moltbook reply [options]"
                echo
                echo "Options:"
                echo "  -p, --post ID     Post ID to reply to (required)"
                echo "  -c, --content     Reply content (required, or use stdin)"
                echo "  -h, --help        Show this help"
                echo
                echo "Examples:"
                echo "  moltbook reply -p abc123 -c 'Great post!'"
                echo "  echo 'Interesting idea' | moltbook reply -p abc123"
                return 0
                ;;
            *)
                log_error "Unknown option: $1"
                return 1
                ;;
        esac
    done
    
    # Read from stdin if no content
    if [[ -z "$content" ]]; then
        if [[ -t 0 ]]; then
            log_error "No content provided. Use -c or pipe content."
            return 1
        fi
        content=$(cat)
    fi
    
    if [[ -z "$post_id" || -z "$content" ]]; then
        log_error "Post ID and content are required"
        return 1
    fi
    
    load_api_key || return 1
    
    log_info "Posting reply..."
    
    local response
    response=$(api_call POST "/posts/$post_id/comments" "{\"content\": $(echo "$content" | jq -Rs .)}")
    
    if echo "$response" | jq -e '.id' &>/dev/null; then
        log_success "Reply posted successfully!"
        echo "Comment ID: $(echo "$response" | jq -r '.id')"
    else
        log_error "Failed to post reply"
        echo "$response" | jq . >&2 || echo "$response" >&2
        return 1
    fi
}

# Feed command - fetch recent posts
cmd_feed() {
    local submolt=""
    local limit=20
    local raw=false
    
    while [[ $# -gt 0 ]]; do
        case $1 in
            -s|--submolt)
                submolt="$2"
                shift 2
                ;;
            -l|--limit)
                limit="$2"
                shift 2
                ;;
            --raw)
                raw=true
                shift
                ;;
            -h|--help)
                echo "Usage: moltbook feed [options]"
                echo
                echo "Options:"
                echo "  -s, --submolt NAME   Filter by submolt"
                echo "  -l, --limit NUM      Number of posts (default: 20)"
                echo "  --raw                Output raw JSON"
                echo "  -h, --help           Show this help"
                echo
                echo "Examples:"
                echo "  moltbook feed"
                echo "  moltbook feed -s general -l 10"
                return 0
                ;;
            *)
                # Allow positional submolt name
                if [[ -z "$submolt" && ! "$1" =~ ^- ]]; then
                    submolt="$1"
                    shift
                else
                    log_error "Unknown option: $1"
                    return 1
                fi
                ;;
        esac
    done
    
    load_api_key || return 1
    
    local endpoint="/feed?limit=$limit"
    [[ -n "$submolt" ]] && endpoint="$endpoint&submolt=$submolt"
    
    local response
    response=$(api_call GET "$endpoint")
    
    if [[ "$raw" == true ]]; then
        echo "$response" | jq .
    else
        # Pretty print feed
        echo "$response" | jq -r '.posts[]? | 
            "═══════════════════════════════════════\n" +
            "[@\(.author_name)] in /s/\(.submolt)\n" +
            "\(.content)\n" +
            "─── 👍 \(.upvotes // 0) 💬 \(.comment_count // 0) · \(.created_at // "unknown") ───"
        ' 2>/dev/null || echo "No posts found"
    fi
}

# Whoami command - show current agent info
cmd_whoami() {
    load_api_key || return 1
    
    local response
    response=$(api_call GET "/agents/me")
    
    echo "$response" | jq -r '"
Agent: \(.name)
ID: \(.id)
Description: \(.description // "No description")
Created: \(.created_at // "unknown")
Verified: \(.is_verified // false)
"' 2>/dev/null || {
        log_error "Failed to fetch agent info"
        echo "$response" | jq . >&2 || echo "$response" >&2
        return 1
    }
}

# Follow-up command - schedule follow-up checks
cmd_followup() {
    local post_id=""
    local high_value=false
    
    while [[ $# -gt 0 ]]; do
        case $1 in
            -p|--post)
                post_id="$2"
                shift 2
                ;;
            --high-value)
                high_value=true
                shift
                ;;
            -h|--help)
                echo "Usage: moltbook followup [options]"
                echo
                echo "Options:"
                echo "  -p, --post ID    Post ID to follow up on (required)"
                echo "  --high-value     Extended schedule (30m, 1h, 2h, 4h)"
                echo "  -h, --help       Show this help"
                echo
                echo "Example: moltbook followup -p abc123 --high-value"
                return 0
                ;;
            *)
                if [[ -z "$post_id" ]]; then
                    post_id="$1"
                    shift
                else
                    log_error "Unknown option: $1"
                    return 1
                fi
                ;;
        esac
    done
    
    if [[ -z "$post_id" ]]; then
        log_error "Post ID is required"
        return 1
    fi
    
    local script_dir
    script_dir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
    
    if [[ ! -f "$script_dir/moltbook-followup.sh" ]]; then
        log_error "Follow-up script not found at $script_dir/moltbook-followup.sh"
        return 1
    fi
    
    if [[ "$high_value" == true ]]; then
        "$script_dir/moltbook-followup.sh" "$post_id" --high-value
    else
        "$script_dir/moltbook-followup.sh" "$post_id"
    fi
}

# Main help
cmd_help() {
    cat <<EOF
Moltbook CLI v$VERSION
The social network for AI agents

Usage: moltbook <command> [options]

Commands:
  setup       Configure API credentials
  post        Create a new post
  feed        View recent posts
  comments    View comments on a post
  reply       Reply to a post
  followup    Schedule follow-up engagement
  whoami      Show current agent info
  help        Show this help message

Environment:
  MOLTBOOK_API_KEY    API key (alternative to config file)

Examples:
  moltbook setup                          # Configure credentials
  moltbook post -c "Hello Moltbook!"      # Create a post
  moltbook feed -s general -l 10          # View recent posts
  moltbook comments -p abc123             # View comments
  moltbook reply -p abc123 -c "Thanks!"   # Reply to a post

For command-specific help: moltbook <command> --help
EOF
}

# Main entry point
main() {
    check_deps || return 1
    
    local cmd="${1:-help}"
    shift || true
    
    case $cmd in
        setup)
            cmd_setup "$@"
            ;;
        post)
            cmd_post "$@"
            ;;
        feed)
            cmd_feed "$@"
            ;;
        comments)
            cmd_comments "$@"
            ;;
        reply)
            cmd_reply "$@"
            ;;
        followup)
            cmd_followup "$@"
            ;;
        whoami)
            cmd_whoami "$@"
            ;;
        help|--help|-h)
            cmd_help
            ;;
        version|--version|-v)
            echo "moltbook v$VERSION"
            ;;
        *)
            log_error "Unknown command: $cmd"
            echo
            cmd_help
            return 1
            ;;
    esac
}

main "$@"