"""
Disk Monitor Job - Disk Usage Checks

Runs every 15 minutes to check disk space on:
- Hub SSD
- Cassaforte HDD
- Root partition

Saves results to database.
"""

def run(context):
    """Execute disk monitoring check."""
    from modules.fleet import FleetStatus
    from core.database import DatabaseManager
    import json

    db = DatabaseManager()
    fleet = FleetStatus()

    # Get disk status
    status = fleet.get_status()
    disk = status.get('disk', {})

    # Save disk status
    db.save_fleet_status(
        check_type='disk',
        status=disk.get('status', 'unknown'),
        data=json.dumps(disk.get('disks', [])),
        message=disk.get('message', 'No data')
    )

    return {
        "status": "success",
        "disk_status": disk.get('status'),
        "disks": len(disk.get('disks', []))
    }


if __name__ == '__main__':
    # Execute when run as a module
    import json
    result = run({})
    print(json.dumps(result, indent=2))