from flask import Flask, jsonify, request
import os

app = Flask(__name__)

# Get the directory where this script is located
script_dir = os.path.dirname(os.path.abspath(__file__))

# Import our archive parser
from .parse_archive import ArchiveParser

# Initialize the archive parser
archive_parser = ArchiveParser()


@app.route("/posts", methods=["GET"])
def get_initial_posts():
    """Get the first 10 posts for initial load"""
    try:
        # Use archive parser to fetch posts
        posts = archive_parser.get_posts(count=10, start_index=0)
        return jsonify(posts)
    except Exception:
        return jsonify({"error": "Failed to fetch posts"}), 500


@app.route("/posts/more", methods=["GET"])
def get_more_posts():
    """Get next 10 posts based on the count parameter"""
    try:
        # Get the count parameter from query string, default to 10
        count = request.args.get("count", 10, type=int)

        # Use archive parser to fetch posts
        posts = archive_parser.get_posts(count=10, start_index=count)

        return jsonify(posts)
    except Exception:
        return jsonify({"error": "Failed to fetch more posts"}), 500


@app.route("/posts/total", methods=["GET"])
def get_total_posts():
    """Get the total number of posts available"""
    try:
        # Get total from archive parser
        total = archive_parser.get_total_posts()
        return jsonify({"total": total})
    except Exception as e:
        return jsonify({"error": "Failed to fetch total count"}), 500


@app.route("/")
def home():
    """Home endpoint to verify server is running"""
    return jsonify(
        {
            "message": "Reddit Clone API Server is running",
            "endpoints": {
                "GET /posts": "Get first 10 posts",
                "GET /posts/more?count=N": "Get next 10 posts, starting from index N",
                "GET /posts/total": "Get total number of posts",
            },
        }
    )


if __name__ == "__main__":
    app.run(debug=True, host="0.0.0.0", port=6006)