prvng_extensions/providers/upcloud/provider.nu

# UpCloud Provider Adapter
# Implements the standard provider interface for UpCloud

use nulib/upcloud/env.nu
use nulib/upcloud/servers.nu *
use nulib/upcloud/cache.nu *
use nulib/upcloud/prices.nu *

# Provider metadata
export def get-provider-metadata []: nothing -> record {
    {
        name: "upcloud"
        version: "1.0.0"
        description: "UpCloud provider for European cloud infrastructure"
        capabilities: {
            server_management: true
            network_management: true
            storage_management: true
            load_balancer: true
            dns_management: false
            cdn: false
            backup_service: true
            monitoring: true
            logging: false
            auto_scaling: false
            spot_instances: false
            containers: false
            serverless: false
            multi_region: true
            encryption_at_rest: true
            encryption_in_transit: true
            compliance_certifications: ["GDPR", "ISO27001"]
        }
    }
}

# Server query operations
export def query_servers [
    find?: string
    cols?: string
]: nothing -> list {
    let str_find = if $find != null { $find } else { "" }
    let str_cols = if $cols != null { $cols } else { "" }
    upcloud_query_servers $str_find $str_cols
}

# Server information operations
export def server_info [
    server: record
    check: bool
    find?: string
    cols?: string
]: nothing -> record {
    upcloud_server_info $server $check
}

# Server existence and status operations
export def server_exists [
    server: record
    error_exit: bool
]: nothing -> bool {
    upcloud_server_exists $server $error_exit
}

export def server_is_running [
    server: record
    error_exit: bool
]: nothing -> bool {
    upcloud_server_is_running $server $error_exit
}

# Server lifecycle operations
export def check_server_requirements [
    settings: record
    server: record
    check: bool
]: nothing -> bool {
    upcloud_check_server_requirements $settings $server $check
}

export def create_server [
    settings: record
    server: record
    check: bool
    wait: bool
]: nothing -> bool {
    try {
        upcloud_server_state $server "started" true $wait $settings
        true
    } catch {
        false
    }
}

export def delete_server [
    settings: record
    server: record
    keep_storage: bool
    error_exit: bool
]: nothing -> bool {
    upcloud_delete_server $settings $server $keep_storage $error_exit
}

export def delete_server_storage [
    settings: record
    server: record
    error_exit: bool
]: nothing -> bool {
    upcloud_delete_server_storage $settings $server $error_exit
}

export def post_create_server [
    settings: record
    server: record
    check: bool
]: nothing -> bool {
    upcloud_post_create_server $settings $server $check
}

export def modify_server [
    settings: record
    server: record
    new_values: list
    error_exit: bool
]: nothing -> bool {
    upcloud_modify_server $settings $server $new_values $error_exit
}

# Server state management
export def server_state [
    server: record
    new_state: string
    error_exit: bool
    wait: bool
    settings: record
]: nothing -> bool {
    upcloud_server_state $server $new_state $error_exit $wait $settings
}

# Network operations
export def get_ip [
    settings: record
    server: record
    ip_type: string
    error_exit: bool
]: nothing -> string {
    let use_type = match $ip_type {
        "$network_public_ip" => "public"
        "$network_private_ip" => "private"
        _ => $ip_type
    }

    let result = (upcloud_server [ "get_ip", $use_type ] --server $server --settings $settings)
    if ($result == null) or ($result | is-empty) {
        ""
    } else {
        $result | to text | str trim
    }
}

export def servers_ips [
    settings: record
    data: list
    prov?: string
    serverpos?: int
]: nothing -> list {
    mut result = []
    mut index = -1

    for srv in $data {
        $index += 1
        let settings_server = ($settings.data.servers | where {|it| $it.hostname == $srv.hostname})
        if ($settings_server | length) == 0 { continue }
        let provider = ($settings_server | get -o 0 | get -o provider | default "")
        if $prov != null and $provider != "upcloud" { continue }
        if $serverpos != null and $serverpos != $index { continue }

        if $srv.ip_addresses? != null {
            $result = ($result | append ($srv.ip_addresses? |
                each {|it| { hostname: $srv.hostname, ip: $it.address, access: $it.access, family: $it.family }} |
                flatten
            ))
        }
    }

    $result
}

# Infrastructure operations
export def load_infra_servers_info [
    settings: record
    server: record
    error_exit: bool
]: nothing -> record {
    upcloud_load_infra_servers_info $settings $server $error_exit
}

export def load_infra_storages_info [
    settings: record
    server: record
    error_exit: bool
]: nothing -> record {
    upcloud_load_infra_storages_info $settings $server $error_exit
}

export def get_infra_storage [
    server: record
    settings: record
    cloud_data: record
    error_exit: bool
]: nothing -> list {
    # Load data directly instead of relying on passed cloud_data
    # This avoids issues with JSON serialization through wrapper scripts
    let upcloud_data = (upcloud_load_infra_servers_info $settings $server $error_exit)

    upcloud_get_item_for_storage $server $settings $upcloud_data
}

export def get_infra_item [
    server: record
    settings: record
    cloud_data: record
    error_exit: bool
]: nothing -> record {
    # Load data directly instead of relying on passed cloud_data
    # This avoids issues with JSON serialization through wrapper scripts
    let upcloud_data = (upcloud_load_infra_servers_info $settings $server $error_exit)

    upcloud_get_item_for_server $server $settings $upcloud_data
}

export def get_infra_price [
    server: record
    data: record
    key: string
    error_exit: bool
    price_col?: string
]: nothing -> float {
    if ($data | get -o item | is-empty) { return 0.0 }
    try {
        upcloud_get_price $data $key $price_col
    } catch {
        0.0
    }
}

# Phase 2 Optimization: Get all prices at once
export def get_all_infra_prices [
    server: record
    data: record
    error_exit: bool
    price_col?: string
]: nothing -> record {
    if ($data | get -o item | is-empty) {
        return { hour: 0.0, day: 0.0, month: 0.0, unit_info: 1 }
    }
    try {
        upcloud_get_all_infra_prices $data $error_exit $price_col
    } catch {
        { hour: 0.0, day: 0.0, month: 0.0, unit_info: 1 }
    }
}

# Cache operations
export def start_cache_info [
    settings: record
    server: record
]: nothing -> nothing {
    upcloud_start_cache_info $settings $server
}

export def create_cache [
    settings: record
    server: record
    error_exit: bool
]: nothing -> nothing {
    upcloud_create_cache $settings $server $error_exit
}

export def read_cache [
    settings: record
    server: record
    error_exit: bool
]: nothing -> nothing {
    upcloud_read_cache $settings $server $error_exit
}

export def clean_cache [
    settings: record
    server: record
    error_exit: bool
]: nothing -> nothing {
    upcloud_clean_cache $settings $server $error_exit
}

export def ip_from_cache [
    settings: record
    server: record
    error_exit: bool
]: nothing -> nothing {
    upcloud_ip_from_cache $settings $server $error_exit
}

# Provider metadata operations
export def on_prov_server [
    server: record
]: nothing -> string {
    # Return empty string (no special message for upcloud servers)
    ""
}

# Provider validation
export def validate_provider []: nothing -> record {
    {
        provider: "upcloud"
        valid: true
        interface_version: "1.0.0"
        capabilities: (get-provider-metadata).capabilities
        last_validated: (date now)
    }
}