ImageSource

from smolvm import ImageSource

# Define an image with checksum verification
image = ImageSource(
    name="ubuntu-jammy",
    kernel_url="https://example.com/kernels/jammy-vmlinux.bin",
    kernel_sha256="e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",
    rootfs_url="https://example.com/rootfs/jammy-rootfs.ext4",
    rootfs_sha256="cf83e1357eefb8bdf1542850d66d8007d620e4050b5715dc83f4a921d36ce9ce",
)

print(image.name)  # ubuntu-jammy
print(image.kernel_url)

from smolvm import ImageSource

# No checksum verification
image = ImageSource(
    name="test-image",
    kernel_url="https://example.com/vmlinux.bin",
    rootfs_url="https://example.com/rootfs.ext4",
)

from smolvm import ImageManager, ImageSource

custom_images = {
    "alpine-minimal": ImageSource(
        name="alpine-minimal",
        kernel_url="https://cdn.example.com/alpine-vmlinux.bin",
        kernel_sha256="abc123...",
        rootfs_url="https://cdn.example.com/alpine-rootfs.ext4",
        rootfs_sha256="def456...",
    ),
    "debian-dev": ImageSource(
        name="debian-dev",
        kernel_url="https://cdn.example.com/debian-vmlinux.bin",
        rootfs_url="https://cdn.example.com/debian-rootfs.ext4",
    ),
}

manager = ImageManager(registry=custom_images)
image = manager.ensure_image("alpine-minimal")

from smolvm import ImageManager, VMConfig, SmolVM

manager = ImageManager()
image = manager.ensure_image("hello")

print(type(image))  # <class 'smolvm.images.LocalImage'>
print(image.name)   # hello
print(image.kernel_path)  # PosixPath('/home/user/.smolvm/images/hello/vmlinux.bin')
print(image.rootfs_path)  # PosixPath('/home/user/.smolvm/images/hello/rootfs.ext4')

# Use with VMConfig
config = VMConfig(
    kernel_path=image.kernel_path,
    rootfs_path=image.rootfs_path,
)

with SmolVM(config) as vm:
    vm.start()

from smolvm import ImageSource

image = ImageSource(
    name="test",
    kernel_url="https://example.com/vmlinux.bin",
    rootfs_url="https://example.com/rootfs.ext4",
)

# This will raise an error
try:
    image.name = "modified"
except Exception as e:
    print(f"Error: {type(e).__name__}")  # Error: ValidationError

# Linux/macOS
sha256sum vmlinux.bin
sha256sum rootfs.ext4

# Or using Python
python3 -c "import hashlib; print(hashlib.sha256(open('vmlinux.bin', 'rb').read()).hexdigest())"

from smolvm.images import BUILTIN_IMAGES

for name, source in BUILTIN_IMAGES.items():
    print(f"{name}:")
    print(f"  Kernel: {source.kernel_url}")
    print(f"  Rootfs: {source.rootfs_url}")
    print(f"  Verified: {source.kernel_sha256 is not None}")