PSProxmoxVE/src/PSProxmoxVE.Core/Services/CloudInitService.cs

using System;
using System.Collections.Generic;
using System.Linq;
using Newtonsoft.Json.Linq;
using PSProxmoxVE.Core.Authentication;
using PSProxmoxVE.Core.Client;
using PSProxmoxVE.Core.Models.Vms;

namespace PSProxmoxVE.Core.Services
{
    /// <summary>
    /// Service for managing Cloud-Init configuration on Proxmox VE QEMU/KVM VMs.
    /// All operations target the /nodes/{node}/qemu/{vmid}/config endpoint.
    /// </summary>
    public class CloudInitService
    {
        private readonly IPveHttpClient? _injectedClient;

        // Cloud-Init field names as used in the PVE API
        private static readonly string[] CloudInitFields =
        {
            "ciuser", "cipassword", "sshkeys",
            "ipconfig0", "ipconfig1", "ipconfig2", "ipconfig3",
            "nameserver", "searchdomain", "cicustom"
        };

        /// <summary>
        /// Initializes a new instance of the <see cref="CloudInitService"/> class.
        /// </summary>
        public CloudInitService() { }

        /// <summary>
        /// Initializes a new instance of the <see cref="CloudInitService"/> class with an injected HTTP client.
        /// </summary>
        /// <param name="client">The HTTP client to use for API calls. The caller owns its lifetime.</param>
        public CloudInitService(IPveHttpClient client)
        {
            _injectedClient = client ?? throw new ArgumentNullException(nameof(client));
        }

        /// <summary>
        /// Retrieves the Cloud-Init specific configuration fields for a VM.
        /// Internally fetches the full VM config and extracts the CI fields.
        /// </summary>
        public PveCloudInitConfig GetCloudInitConfig(PveSession session, string node, int vmid)
        {
            if (session == null) throw new ArgumentNullException(nameof(session));
            if (string.IsNullOrWhiteSpace(node)) throw new ArgumentNullException(nameof(node));

            IPveHttpClient client = _injectedClient ?? new PveHttpClient(session);
            try
            {
                var response = client.GetAsync($"nodes/{Uri.EscapeDataString(node)}/qemu/{vmid}/config")
                    .GetAwaiter().GetResult();
                var data = JObject.Parse(response)["data"];
                if (data == null) return new PveCloudInitConfig();

                // Extract only the Cloud-Init fields into a reduced JObject for deserialization
                var ciObj = new JObject();
                foreach (var field in CloudInitFields)
                {
                    if (data[field] != null)
                        ciObj[field] = data[field];
                }

                return ciObj.ToObject<PveCloudInitConfig>() ?? new PveCloudInitConfig();
            }
            finally
            {
                if (_injectedClient == null) client.Dispose();
            }
        }

        /// <summary>
        /// Updates Cloud-Init configuration fields on a VM. Only fields present in
        /// <paramref name="config"/> are changed; unspecified fields are left as-is.
        /// </summary>
        /// <remarks>
        /// Pass only the Cloud-Init keys you want to change, for example:
        /// <c>new Dictionary&lt;string, object&gt; { ["ciuser"] = "ubuntu", ["ipconfig0"] = "ip=dhcp" }</c>
        /// </remarks>
        public void SetCloudInitConfig(
            PveSession session,
            string node,
            int vmid,
            Dictionary<string, object> config)
        {
            if (session == null) throw new ArgumentNullException(nameof(session));
            if (string.IsNullOrWhiteSpace(node)) throw new ArgumentNullException(nameof(node));
            if (config == null) throw new ArgumentNullException(nameof(config));

            IPveHttpClient client = _injectedClient ?? new PveHttpClient(session);
            try
            {
                var formData = config.ToDictionary(
                    kvp => kvp.Key,
                    kvp => kvp.Value?.ToString() ?? string.Empty);
                client.PutAsync($"nodes/{Uri.EscapeDataString(node)}/qemu/{vmid}/config", formData)
                    .GetAwaiter().GetResult();
            }
            finally
            {
                if (_injectedClient == null) client.Dispose();
            }
        }

        /// <summary>
        /// Regenerates the Cloud-Init ISO drive attached to the VM
        /// (PUT /nodes/{node}/qemu/{vmid}/cloudinit).
        /// </summary>
        /// <returns>The UPID of the regeneration task.</returns>
        public string RegenerateCloudInitImage(PveSession session, string node, int vmid)
        {
            if (session == null) throw new ArgumentNullException(nameof(session));
            if (string.IsNullOrWhiteSpace(node)) throw new ArgumentNullException(nameof(node));

            IPveHttpClient client = _injectedClient ?? new PveHttpClient(session);
            try
            {
                var response = client.PutAsync($"nodes/{Uri.EscapeDataString(node)}/qemu/{vmid}/cloudinit", null)
                    .GetAwaiter().GetResult();
                var data = JObject.Parse(response)["data"];
                return data?.ToString() ?? string.Empty;
            }
            finally
            {
                if (_injectedClient == null) client.Dispose();
            }
        }
    }
}