import json
from flask import Flask, request, jsonify
from database import DatabaseConfig, init_database
import pymysql

# Load configuration
with open('config.json') as f:
    config = json.load(f)

app = Flask(__name__)

# Initialize database
db_config = DatabaseConfig()
init_database()  # Initialize database and tables if they don't exist

@app.route('/api/v1/save', methods=['POST'])
def save_data():
    """API endpoint to save data to the database"""
    try:
        # Get JSON data from request
        data = request.get_json()
        
        if not data:
            return jsonify({'error': 'No data provided'}), 400
        
        # Extract data fields
        data_key = data.get('key', '')
        data_value = data.get('value', '')
        
        if not data_key:
            return jsonify({'error': 'Key is required'}), 400
        
        # Connect to database
        connection = db_config.get_connection()
        if not connection:
            return jsonify({'error': 'Database connection failed'}), 500
        
        try:
            with connection.cursor() as cursor:
                # Insert data into the table
                sql = "INSERT INTO admin_data (data_key, data_value) VALUES (%s, %s)"
                cursor.execute(sql, (data_key, data_value))
                connection.commit()
                
                # Get the ID of the inserted record
                record_id = cursor.lastrowid
                
                return jsonify({
                    'message': 'Data saved successfully',
                    'id': record_id
                }), 201
                
        except Exception as e:
            return jsonify({'error': str(e)}), 500
        finally:
            connection.close()
            
    except Exception as e:
        return jsonify({'error': str(e)}), 500

if __name__ == '__main__':
    # Get server configuration
    server_config = config['server']
    app.run(host=server_config['host'], port=server_config['port'], debug=server_config['debug'])