import json
import subprocess
import os
from jinja2 import Template
from flask import Flask, render_template_string, jsonify, url_for, redirect, render_template, request

app = Flask(__name__)

from modules.parse import run_bgp_curl_command, run_arp_curl_command, run_neighbors_curl_command, run_interfaces_curl_command, run_bgp_route_curl_command

from modules.bgp import parse_bgp_data, generate_bgp_json

from modules.arp import parse_arp_data, generate_arp_json

from modules.neighbors import parse_neighbors_data, generate_neighbors_json

from modules.interfaces import parse_interface_data

@app.route('/')
def index():
    bgp_image = os.getenv("BGP_TOOLS_IMAGE")
    return render_template('index.html', bgp_image=bgp_image)

@app.route("/bgp")
def bgp():
    bgp_data = run_bgp_curl_command()
    ipv4_info, ipv4_peers, ipv6_info, ipv6_peers = parse_bgp_data(bgp_data)
    return render_template('bgp.html')

@app.route("/bgp/json")
def bgp_json():
    bgp_data = run_bgp_curl_command()
    ipv4_info, ipv4_peers, ipv6_info, ipv6_peers = parse_bgp_data(bgp_data)
    return jsonify(generate_bgp_json(ipv4_info, ipv4_peers, ipv6_info, ipv6_peers))

@app.route("/arp")
def arp():
    return render_template("arp.html")

@app.route("/arp/json")
def arp_json():
    arp_data = run_arp_curl_command()
    arp_table = parse_arp_data(arp_data)
    return jsonify(generate_arp_json(arp_table))

@app.route("/neighbors")
def neighbors():
    return render_template("neighbors.html")

@app.route("/neighbors/json")
def neighborsp_json():
    neighbors_data = run_neighbors_curl_command()
    neighbors_table = parse_neighbors_data(neighbors_data)
    return jsonify(generate_neighbors_json(neighbors_table))

@app.route('/interfaces')
def interface_table_page():
    return render_template("interfaces.html")

@app.route('/interfaces/json')
def interface_table_summary_json():
    data = run_interfaces_curl_command()
    interface_table = parse_interface_data(data)
    return jsonify({"interface_table": interface_table})
 
@app.route('/bgp-route')
def bgp_route_page():
    return render_template("bgp-route.html")
 
@app.route('/bgp-route/lookup', methods=['POST'])
def bgp_route_lookup():
    data = request.json
    ip_version = data.get('ip_version')
    bgprouteprefix = data.get('bgprouteprefix')

    if not ip_version or not bgprouteprefix:
        return jsonify({"error": "ip_version and bgprouteprefix are required"}), 400

    try:
        result = run_bgp_route_curl_command(ip_version, bgprouteprefix)
        return jsonify(result)
    except Exception as e:
        return jsonify({"error": str(e)}), 500