main.py

import typer
import os
import threading
import json
from rich.console import Console
from rich.table import Table
from datetime import datetime, timedelta, timezone
from dotenv import load_dotenv
from api import get_user_id, get_user_profile, get_user_posts, get_post_insights, fetch_all_posts, create_post, get_post_replies, get_post_replies_count
from utils import convert_to_locale

app = typer.Typer()
console = Console()
load_dotenv()

ACCESS_TOKEN = os.getenv("ACCESS_TOKEN")
HEADERS = {
    'Authorization': f'Bearer {ACCESS_TOKEN}'
}
DRAFTS_FILE = 'drafts.json'
SERVER_PROCESS_TIME = 10

@app.command()
def get_profile():
    """
    Retrieve and display user profile information, including the last post made by the user.
    """
    user_id = get_user_id(HEADERS)
    profile = get_user_profile(user_id, HEADERS)
    last_post = get_user_posts(user_id, HEADERS, limit=1)[0]

    profile_table = Table(title=f'{profile["username"]}\'s Profile')
    profile_table.add_column("Field", style="cyan", no_wrap=True)
    profile_table.add_column("Value", style="magenta")

    profile_table.add_row("ID", profile.get("id", "N/A"))
    profile_table.add_row("Username", profile.get("username", "N/A"))
    profile_table.add_row("Profile Picture URL", profile.get("threads_profile_picture_url", "N/A"))
    profile_table.add_row("Biography", profile.get("threads_biography", "N/A"))
    if last_post:
        profile_table.add_row("Last Post ID", last_post.get("id", "N/A"))
        profile_table.add_row("Post Type", last_post.get("media_type", "N/A"))
        profile_table.add_row("Post Text", last_post.get("text", "N/A"))
        profile_table.add_row("Post Permalink", last_post.get("permalink", "N/A"))
        profile_table.add_row("Post Timestamp", convert_to_locale(last_post.get("timestamp", "N/A")))
    else:
        profile_table.add_row("Message", "No posts found")

    console.print(profile_table)

@app.command()
def get_recent_posts(limit: int = 5):
    """
    Retrieve the most recent posts.
    """
    user_id = get_user_id(HEADERS)
    posts = get_user_posts(user_id, HEADERS, limit=limit)

    table = Table(title="Recent Posts")
    table.add_column("ID", style="cyan", no_wrap=True)
    table.add_column("Username", style="cyan", no_wrap=True)
    table.add_column("Timestamp", style="magenta")
    table.add_column("Type", style="green")
    table.add_column("Text", style="yellow")
    table.add_column("Permalink", style="blue")
    table.add_column("Replies", style="red")

    for post in posts:
        if post.get('media_type') == 'REPOST_FACADE':
            continue
        timestamp = convert_to_locale(post.get('timestamp', 'N/A'))
        replies_count = get_post_replies_count(post['id'], HEADERS)
        table.add_row(
            post.get('id', 'N/A'),
            post.get('username', 'N/A'),
            timestamp,
            post.get('media_type', 'N/A'),
            post.get('text', 'N/A'),
            post.get('permalink', 'N/A'),
            str(replies_count)
        )

    console.print(table)

@app.command()
def get_top_liked_posts(limit: int = 5, time_range: str = None):
    """
    Retrieve the top liked posts of all time or within a specific time range.
    """
    user_id = get_user_id(HEADERS)
    all_posts = fetch_all_posts(user_id, HEADERS)

    if time_range:
        now = datetime.now(timezone.utc)
        if time_range.endswith('w'):
            weeks = int(time_range[:-1])
            start_time = now - timedelta(weeks=weeks)
        elif time_range.endswith('d'):
            days = int(time_range[:-1])
            start_time = now - timedelta(days=days)
        elif time_range.endswith('h'):
            hours = int(time_range[:-1])
            start_time = now - timedelta(hours=hours)
        elif time_range.endswith('m'):
            months = int(time_range[:-1])
            start_time = now - timedelta(days=30 * months)
        else:
            typer.echo("Invalid time range format. Use '2w' for 2 weeks, '7d' for 7 days, '24h' for 24 hours, or '7m' for 7 months.")
            return

        all_posts = [post for post in all_posts if datetime.strptime(post['timestamp'], '%Y-%m-%dT%H:%M:%S%z') >= start_time]

    posts_with_likes = []
    for post in all_posts:
        if post.get('media_type') == 'REPOST_FACADE':
            continue
        insights = get_post_insights(post['id'], HEADERS)
        if 'likes' in insights:
            posts_with_likes.append((post, insights['likes']))

    posts_with_likes.sort(key=lambda x: x[1], reverse=True)
    top_liked_posts = posts_with_likes[:limit]

    table = Table(title="Top Liked Posts")
    table.add_column("Username", style="cyan", no_wrap=True)
    table.add_column("Timestamp", style="magenta")
    table.add_column("Type", style="green")
    table.add_column("Text", style="yellow")
    table.add_column("Permalink", style="blue")
    table.add_column("Likes", style="red")
    table.add_column("Replies", style="green")
    table.add_column("Reposts", style="blue")
    table.add_column("Quotes", style="yellow")
    table.add_column("Views", style="cyan")

    for post, likes in top_liked_posts:
        timestamp = convert_to_locale(post.get('timestamp', 'N/A'))
        insights = get_post_insights(post['id'], HEADERS)
        table.add_row(
            post.get('username', 'N/A'),
            timestamp,
            post.get('media_type', 'N/A'),
            post.get('text', 'N/A'),
            post.get('permalink', 'N/A'),
            str(insights.get('likes', 'N/A')),
            str(insights.get('replies', 'N/A')),
            str(insights.get('reposts', 'N/A')),
            str(insights.get('quotes', 'N/A')),
            str(insights.get('views', 'N/A'))
        )

    console.print(table)

@app.command()
def create_text_post(text: str):
    """
    Create a post with text.
    """
    user_id = get_user_id(HEADERS)
    payload = {
        "media_type": "TEXT",
        "text": text
    }
    post = create_post(user_id, HEADERS, payload)
    typer.echo(f"Post created with ID: {post['id']}")

@app.command()
def create_image_post(text: str, image_url: str):
    """
    Create a post with an image.
    """
    user_id = get_user_id(HEADERS)
    payload = {
        "media_type": "IMAGE",
        "image_url": image_url,
        "text": text
    }
    post = create_post(user_id, HEADERS, payload)
    typer.echo(f"Post created with ID: {post['id']}")

@app.command()
def get_latest_replies(media_id: str, limit: int = 5):
    """
    Retrieve the latest replies for a specific media post.
    """
    replies = get_post_replies(media_id, HEADERS, limit=limit)

    table = Table(title="Latest Replies")
    table.add_column("Username", style="cyan", no_wrap=True)
    table.add_column("Media ID", style="cyan", no_wrap=True)
    table.add_column("Timestamp", style="magenta")
    table.add_column("Text", style="yellow")
    table.add_column("Permalink", style="blue")

    for reply in replies:
        timestamp = convert_to_locale(reply.get('timestamp', 'N/A'))
        table.add_row(
            reply.get('username', 'N/A'),
            reply.get('id', 'N/A'),
            timestamp,
            reply.get('text', 'N/A'),
            reply.get('permalink', 'N/A')
        )

    console.print(table)

@app.command()
def send_reply(media_id: str, text: str):
    """
    Send a reply to a specific media post.
    """
    user_id = get_user_id(HEADERS)
    payload = {
        "media_type": "TEXT",
        "text": text,
        "reply_to_id": media_id
    }
    reply = create_post(user_id, HEADERS, payload)
    typer.echo(f"Reply created with ID: {reply['id']}")

def job_create_text_post(text: str):
    """
    Job function to create a post with text.
    """
    create_text_post(text)

@app.command()
def schedule_post(text: str, post_time: str):
    """
    Schedule a post with text at a specific time.
    """
    post_time_dt = datetime.strptime(post_time, '%Y-%m-%d %H:%M:%S')
    current_time = datetime.now()
    delay = (post_time_dt - current_time).total_seconds()

    if delay <= 0:
        typer.echo("Scheduled time must be in the future.")
        return

    timer = threading.Timer(delay, job_create_text_post, [text])
    timer.start()
    typer.echo(f"Post scheduled for {post_time} with text: '{text}'")

@app.command()
def create_draft(text: str, drafts_file: str = DRAFTS_FILE):
    '''
    Create a draft with the given text and save it to the drafts file.
    '''
    if os.path.exists(drafts_file):
        with open(drafts_file, 'r') as file:
            drafts = json.load(file)
    else:
        drafts = []

    next_id = max([draft['id'] for draft in drafts], default=0) + 1

    draft = {
        "id": next_id,
        "text": text,
        "timestamp": datetime.now().strftime('%Y-%m-%d %H:%M:%S')
    }
    drafts.append(draft)

    with open(drafts_file, 'w') as file:
        json.dump(drafts, file, indent=4)

    typer.echo(f"Draft created with ID: {next_id}")

@app.command()
def get_drafts(drafts_file: str = DRAFTS_FILE):
    '''
    Get all drafts from the drafts file.
    '''
    if not os.path.exists(drafts_file):
        typer.echo("No drafts found.")
        return

    with open(drafts_file, 'r') as file:
        drafts = json.load(file)

    table = Table(title="Drafts")
    table.add_column("ID", style="cyan", no_wrap=True)
    table.add_column("Text", style="yellow")
    table.add_column("Timestamp", style="magenta")

    for draft in drafts:
        table.add_row(
            str(draft['id']),
            draft['text'],
            draft['timestamp']
        )

    console.print(table)

@app.command()
def send_draft(draft_id: int, drafts_file: str = DRAFTS_FILE):
    '''
    Send a draft with the given ID and remove it from the drafts file.
    '''
    if not os.path.exists(drafts_file):
        typer.echo("No drafts found.")
        raise typer.Exit(1)

    with open(drafts_file, 'r') as file:
        drafts = json.load(file)

    draft = next((draft for draft in drafts if draft['id'] == draft_id), None)

    if draft is None:
        typer.echo(f"Draft with ID {draft_id} not found.")
        raise typer.Exit(1)

    create_text_post(draft['text'])

    drafts = [draft for draft in drafts if draft['id'] != draft_id]

    with open(drafts_file, 'w') as file:
        json.dump(drafts, file, indent=4)

    typer.echo(f"Draft with ID {draft_id} sent and removed from drafts.")

def main():
    app()

if __name__ == "__main__":
    main()