import pymongo
import random
import time
from datetime import datetime, timedelta
import json

# Connection ke MongoDB
client = pymongo.MongoClient("mongodb://iot_user:iot_password@localhost:27017/iot_db")
db = client.iot_db
    
def generate_temperature_data(device_id):
    """Generate data sensor suhu dan kelembaban"""
    return {
        'device_id': device_id,
        'timestamp': datetime.utcnow(),
        'sensor_type': 'temperature_humidity',
        'data': {
            'temperature': round(random.uniform(20.0, 35.0), 2),
            'humidity': round(random.uniform(40.0, 80.0), 2),
            'heat_index': round(random.uniform(25.0, 40.0), 2)
        },
        'location': 'Living Room',
        'battery_level': random.randint(60, 100),
        'signal_strength': random.randint(-80, -30)
    }

def generate_motion_data(device_id):
    """Generate data sensor gerak"""
    return {
        'device_id': device_id,
        'timestamp': datetime.utcnow(),
        'sensor_type': 'motion',
        'data': {
            'motion_detected': random.choice([True, False]),
            'motion_intensity': random.randint(1, 10) if random.choice([True, False]) else 0
        },
        'location': 'Bedroom',
        'battery_level': random.randint(70, 100),
        'signal_strength': random.randint(-75, -35)
    }

def generate_light_data(device_id):
    """Generate data sensor cahaya"""
    return {
        'device_id': device_id,
        'timestamp': datetime.utcnow(),
        'sensor_type': 'light',
        'data': {
            'light_intensity': random.randint(0, 1000),
            'uv_index': round(random.uniform(0, 11), 1)
        },
        'location': 'Garden',
        'battery_level': random.randint(50, 100),
        'signal_strength': random.randint(-70, -40)
    }

def generate_soil_moisture_data(device_id):
    """Generate data sensor kelembaban tanah"""
    return {
        'device_id': device_id,
        'timestamp': datetime.utcnow(),
        'sensor_type': 'soil_moisture',
        'data': {
            'soil_moisture': round(random.uniform(0, 100), 2),
            'soil_temperature': round(random.uniform(15.0, 30.0), 2),
            'ph_level': round(random.uniform(5.5, 8.0), 2)
        },
        'location': 'Greenhouse',
        'battery_level': random.randint(40, 100),
        'signal_strength': random.randint(-85, -40)
    }

def generate_air_quality_data(device_id):
    """Generate data sensor kualitas udara"""
    return {
        'device_id': device_id,
        'timestamp': datetime.utcnow(),
        'sensor_type': 'air_quality',
        'data': {
            'pm2_5': round(random.uniform(0, 500), 2),
            'pm10': round(random.uniform(0, 500), 2),
            'co2_level': round(random.uniform(400, 5000), 2)
        },
        'location': 'Office',
        'battery_level': random.randint(50, 100),
        'signal_strength': random.randint(-80, -35)
    }

def generate_alert(device_id, alert_type, message):
    """Generate alert jika ada kondisi tertentu"""
    return {
        'device_id': device_id,
        'timestamp': datetime.utcnow(),
        'alert_type': alert_type,
        'severity': random.choice(['low', 'medium', 'high']),
        'message': message,
        'acknowledged': False,
        'created_at': datetime.utcnow()
    }

def main():
    print("Starting IoT Data Generator...")
    
    try:
        while True:
            # Generate data untuk setiap device
            temp_data = generate_temperature_data('ESP32_001')
            motion_data = generate_motion_data('ESP32_002')
            light_data = generate_light_data('ESP32_003')
            soil_data = generate_soil_moisture_data('ESP32_004')
            air_quality_data = generate_air_quality_data('ESP32_005')
            
            # Insert data ke MongoDB
            db.sensor_data.insert_one(temp_data)
            db.sensor_data.insert_one(motion_data)
            db.sensor_data.insert_one(light_data)
            db.sensor_data.insert_one(soil_data)
            db.sensor_data.insert_one(air_quality_data)
            
            # Generate alerts berdasarkan kondisi tertentu
            if temp_data['data']['temperature'] > 30:
                alert = generate_alert(
                    'ESP32_001',
                    'high_temperature',
                    f"High temperature detected: {temp_data['data']['temperature']}°C"
                )
                db.alerts.insert_one(alert)
                
            if motion_data['data']['motion_detected'] and motion_data['data']['motion_intensity'] > 7:
                alert = generate_alert(
                    'ESP32_002',
                    'motion_detected',
                    f"High motion intensity detected: {motion_data['data']['motion_intensity']}"
                )
                db.alerts.insert_one(alert)
                
            if light_data['data']['light_intensity'] < 100:
                alert = generate_alert(
                    'ESP32_003',
                    'low_light',
                    f"Low light detected: {light_data['data']['light_intensity']} lux"
                )
                db.alerts.insert_one(alert)

            if soil_data['data']['soil_moisture'] > 90:
                alert = generate_alert(
                    'ESP32_004',
                    'high_soil_moisture',
                    f"High soil moisture detected: {soil_data['data']['soil_moisture']}%"
                )
                db.alerts.insert_one(alert)

            if soil_data['data']['ph_level'] < 6.0 or soil_data['data']['ph_level'] > 7.5:
                alert = generate_alert(
                    'ESP32_004',
                    'abnormal_soil_ph',
                    f"Abnormal soil pH detected: {soil_data['data']['ph_level']}"
                )
                db.alerts.insert_one(alert)

            if air_quality_data['data']['co2_level'] > 1000:
                alert = generate_alert(
                    'ESP32_005',
                    'high_co2',
                    f"High CO₂ level detected: {air_quality_data['data']['co2_level']} ppm"
                )
                db.alerts.insert_one(alert)

            if air_quality_data['data']['pm2_5'] > 150:
                alert = generate_alert(
                    'ESP32_005',
                    'high_pm2_5',
                    f"High PM2.5 level detected: {air_quality_data['data']['pm2_5']} µg/m³"
                )
                db.alerts.insert_one(alert)

            if air_quality_data['data']['pm10'] > 200:
                alert = generate_alert(
                    'ESP32_005',
                    'high_pm10',
                    f"High PM10 level detected: {air_quality_data['data']['pm10']} µg/m³"
                )
                db.alerts.insert_one(alert)
                
            print(f"Data inserted at {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
            time.sleep(2)  # Generate data setiap 2 detik
            
    except KeyboardInterrupt:
        print("\nData generator stopped.")
    finally:
        client.close()

if __name__ == "__main__":
    main()