@page "/clusters/{ClusterId}/drivers/new/s7"
@attribute [Microsoft.AspNetCore.Authorization.Authorize]
@rendermode RenderMode.InteractiveServer
@using Microsoft.AspNetCore.Components.Forms
@using Microsoft.EntityFrameworkCore
@using ZB.MOM.WW.OtOpcUa.AdminUI.Clients
@using ZB.MOM.WW.OtOpcUa.AdminUI.Components.Shared.Drivers
@using ZB.MOM.WW.OtOpcUa.AdminUI.Components.Shared.Drivers.Pickers
@using ZB.MOM.WW.OtOpcUa.Configuration
@using ZB.MOM.WW.OtOpcUa.Configuration.Entities
@using ZB.MOM.WW.OtOpcUa.Driver.S7
@inject IDbContextFactory<OtOpcUaConfigDbContext> DbFactory
@inject NavigationManager Nav

<div class="d-flex justify-content-between align-items-center mb-3">
    <h4 class="mb-0">@(IsNew ? "New Siemens S7 driver" : "Edit Siemens S7 driver") &middot; <span class="mono">@ClusterId</span></h4>
    <a href="/clusters/@ClusterId/drivers" class="btn btn-outline-secondary btn-sm">Cancel</a>
</div>
<ClusterNav ClusterId="@ClusterId" ActiveTab="drivers" />

@if (!_loaded)
{
    <p>Loading…</p>
}
else if (!IsNew && _existing is null)
{
    <section class="panel notice rise" style="animation-delay:.02s">
        Driver instance <span class="mono">@DriverInstanceId</span> was not found in cluster <span class="mono">@ClusterId</span>.
    </section>
}
else
{
    <EditForm Model="_form" OnValidSubmit="SubmitAsync" FormName="s7DriverEdit">
        <DataAnnotationsValidator />
        <DriverFormShell IsNew="IsNew" Busy="_busy" Error="_error"
                         CancelHref="@($"/clusters/{ClusterId}/drivers")"
                         OnDelete="@(IsNew ? null : (EventCallback?)EventCallback.Factory.Create(this, DeleteAsync))">

            <DriverIdentitySection Model="_identityModel" Namespaces="_namespaces" IsNew="IsNew" ShowDriverType="false" />

            @if (!IsNew && !string.IsNullOrEmpty(DriverInstanceId))
            {
                <DriverStatusPanel DriverInstanceId="@DriverInstanceId" Enabled="@_identityModel.Enabled" />
            }

            <div class="mt-2 mb-3">
                <DriverTestConnectButton DriverType="@DriverTypeKey"
                                         GetConfigJson="@SerializeCurrentConfig"
                                         TimeoutSeconds="@_form.AdminProbeTimeoutSeconds" />
                <button type="button" class="btn btn-sm btn-outline-secondary mt-2"
                        @onclick="@(() => _showPicker = true)">
                    Pick address
                </button>
            </div>

            <DriverTagPicker @bind-Visible="_showPicker"
                             Title="S7 address"
                             CurrentAddress="@_pickedAddress"
                             OnPickAddress="@OnAddressPicked">
                <S7AddressPickerBody CurrentAddress="@_pickedAddress"
                                     CurrentAddressChanged="@((s) => _pickedAddress = s)" />
            </DriverTagPicker>

            @* Connection *@
            <section class="panel rise mt-3" style="animation-delay:.05s">
                <div class="panel-head">Connection</div>
                <div style="padding:1rem">
                    <div class="row">
                        <div class="col-md-6 mb-3">
                            <label class="form-label" for="s7Host">Host</label>
                            <InputText id="s7Host" @bind-Value="_form.Host"
                                       class="form-control form-control-sm mono"
                                       placeholder="192.168.0.1" />
                            <div class="form-text">PLC IP address or hostname.</div>
                        </div>
                        <div class="col-md-3 mb-3">
                            <label class="form-label" for="s7Port">Port</label>
                            <InputNumber id="s7Port" @bind-Value="_form.Port"
                                         class="form-control form-control-sm" />
                            <div class="form-text">ISO-on-TCP; usually 102.</div>
                        </div>
                        <div class="col-md-3 mb-3">
                            <label class="form-label" for="s7TimeoutSec">Timeout (seconds)</label>
                            <InputNumber id="s7TimeoutSec" @bind-Value="_form.TimeoutSeconds"
                                         class="form-control form-control-sm" />
                        </div>
                    </div>
                    <div class="row">
                        <div class="col-md-4 mb-3">
                            <label class="form-label" for="s7CpuType">CPU type</label>
                            <InputSelect id="s7CpuType" @bind-Value="_form.CpuType"
                                         class="form-select form-select-sm">
                                @foreach (var v in Enum.GetValues<S7CpuType>())
                                {
                                    <option value="@v">@v</option>
                                }
                            </InputSelect>
                            <div class="form-text">Controls ISO-TSAP slot byte during handshake.</div>
                        </div>
                        <div class="col-md-2 mb-3">
                            <label class="form-label" for="s7Rack">Rack</label>
                            <InputNumber id="s7Rack" @bind-Value="_form.Rack"
                                         class="form-control form-control-sm" />
                            <div class="form-text">Almost always 0.</div>
                        </div>
                        <div class="col-md-2 mb-3">
                            <label class="form-label" for="s7Slot">Slot</label>
                            <InputNumber id="s7Slot" @bind-Value="_form.Slot"
                                         class="form-control form-control-sm" />
                            <div class="form-text">S7-300/400 = 2; S7-1200/1500 = 0.</div>
                        </div>
                    </div>
                </div>
            </section>

            @* Probe *@
            <section class="panel rise mt-3" style="animation-delay:.08s">
                <div class="panel-head">Connectivity probe</div>
                <div style="padding:1rem">
                    <div class="row">
                        <div class="col-md-3 mb-3">
                            <div class="form-check form-switch mt-2">
                                <InputCheckbox id="s7ProbeEnabled" @bind-Value="_form.ProbeEnabled"
                                               class="form-check-input" />
                                <label class="form-check-label" for="s7ProbeEnabled">Probe enabled</label>
                            </div>
                        </div>
                        <div class="col-md-3 mb-3">
                            <label class="form-label" for="s7ProbeIntervalSec">Probe interval (s)</label>
                            <InputNumber id="s7ProbeIntervalSec" @bind-Value="_form.ProbeIntervalSeconds"
                                         class="form-control form-control-sm" />
                        </div>
                        <div class="col-md-3 mb-3">
                            <label class="form-label" for="s7ProbeTimeoutSec">Probe timeout (s)</label>
                            <InputNumber id="s7ProbeTimeoutSec" @bind-Value="_form.ProbeTimeoutSeconds"
                                         class="form-control form-control-sm" />
                        </div>
                        <div class="col-md-3 mb-3">
                            <label class="form-label" for="s7AdminProbeTimeout">Admin probe timeout (s)</label>
                            <InputNumber id="s7AdminProbeTimeout" @bind-Value="_form.AdminProbeTimeoutSeconds"
                                         class="form-control form-control-sm" />
                            <div class="form-text">Test Connect timeout (1–60 s).</div>
                        </div>
                    </div>
                </div>
            </section>

            @* Tags — read-only JSON view *@
            <section class="panel rise mt-3" style="animation-delay:.11s">
                <div class="panel-head">Tags</div>
                <div style="padding:1rem">
                    <div class="form-text mb-2">
                        Tag list editor coming in a follow-up phase. To add/remove tags, edit the JSON directly in the raw driver config via the generic editor, or deploy via the import tooling.
                    </div>
                    @if (_form.TagsJson is not null)
                    {
                        <pre class="form-control form-control-sm mono" style="min-height:4rem;max-height:12rem;overflow:auto;white-space:pre-wrap">@_form.TagsJson</pre>
                    }
                    else
                    {
                        <p class="text-muted"><em>No tags configured.</em></p>
                    }
                </div>
            </section>

            <DriverResilienceSection @bind-ResilienceConfig="_form.ResilienceConfig" />
        </DriverFormShell>
    </EditForm>
}

@code {
    [Parameter] public string ClusterId { get; set; } = "";
    [Parameter] public string? DriverInstanceId { get; set; }

    private const string DriverTypeKey = "S7";

    private bool IsNew => string.IsNullOrEmpty(DriverInstanceId);

    private static readonly System.Text.Json.JsonSerializerOptions _jsonOpts = new()
    {
        PropertyNamingPolicy = System.Text.Json.JsonNamingPolicy.CamelCase,
        UnmappedMemberHandling = System.Text.Json.Serialization.JsonUnmappedMemberHandling.Skip,
        WriteIndented = false,
    };

    private FormModel _form = new();
    private DriverIdentitySection.DriverIdentityModel _identityModel = new() { DriverType = DriverTypeKey };
    private DriverInstance? _existing;
    private List<Namespace> _namespaces = new();
    private bool _loaded, _busy;
    private string? _error;

    // Address picker state
    private bool _showPicker;
    private string _pickedAddress = "";

    private void OnAddressPicked(string address) => _pickedAddress = address;

    protected override async Task OnInitializedAsync()
    {
        await using var db = await DbFactory.CreateDbContextAsync();
        _namespaces = await db.Namespaces.AsNoTracking()
            .Where(n => n.ClusterId == ClusterId)
            .OrderBy(n => n.NamespaceId).ToListAsync();

        if (IsNew)
        {
            _identityModel = new() { DriverType = DriverTypeKey, NamespaceId = _namespaces.FirstOrDefault()?.NamespaceId ?? "", Enabled = true };
            _form = FormModel.FromOptions(new S7DriverOptions());
        }
        else
        {
            _existing = await db.DriverInstances.AsNoTracking()
                .FirstOrDefaultAsync(d => d.ClusterId == ClusterId && d.DriverInstanceId == DriverInstanceId);
            if (_existing is not null)
            {
                _identityModel = new()
                {
                    DriverInstanceId = _existing.DriverInstanceId,
                    Name = _existing.Name,
                    DriverType = _existing.DriverType,
                    NamespaceId = _existing.NamespaceId,
                    Enabled = _existing.Enabled,
                };
                var opts = TryDeserialize(_existing.DriverConfig) ?? new S7DriverOptions();
                _form = FormModel.FromOptions(opts);
                _form.ResilienceConfig = _existing.ResilienceConfig;
                _form.RowVersion = _existing.RowVersion;
            }
        }
        _loaded = true;
    }

    private async Task SubmitAsync()
    {
        _busy = true; _error = null;
        try
        {
            var opts = _form.ToOptions();
            var configJson = System.Text.Json.JsonSerializer.Serialize(opts, _jsonOpts);
            await using var db = await DbFactory.CreateDbContextAsync();
            if (IsNew)
            {
                if (await db.DriverInstances.AnyAsync(d => d.DriverInstanceId == _identityModel.DriverInstanceId))
                {
                    _error = $"Driver instance '{_identityModel.DriverInstanceId}' already exists."; return;
                }
                db.DriverInstances.Add(new DriverInstance
                {
                    DriverInstanceId = _identityModel.DriverInstanceId,
                    ClusterId = ClusterId,
                    NamespaceId = _identityModel.NamespaceId,
                    Name = _identityModel.Name,
                    DriverType = DriverTypeKey,
                    Enabled = _identityModel.Enabled,
                    DriverConfig = configJson,
                    ResilienceConfig = string.IsNullOrWhiteSpace(_form.ResilienceConfig) ? null : _form.ResilienceConfig,
                });
            }
            else
            {
                var entity = await db.DriverInstances.FirstOrDefaultAsync(
                    d => d.ClusterId == ClusterId && d.DriverInstanceId == DriverInstanceId);
                if (entity is null) { _error = "Row no longer exists."; return; }
                db.Entry(entity).Property(e => e.RowVersion).OriginalValue = _form.RowVersion;
                entity.NamespaceId = _identityModel.NamespaceId;
                entity.Name = _identityModel.Name;
                entity.Enabled = _identityModel.Enabled;
                entity.DriverConfig = configJson;
                entity.ResilienceConfig = string.IsNullOrWhiteSpace(_form.ResilienceConfig) ? null : _form.ResilienceConfig;
            }
            await db.SaveChangesAsync();
            Nav.NavigateTo($"/clusters/{ClusterId}/drivers");
        }
        catch (DbUpdateConcurrencyException)
        {
            _error = "Another user changed this driver instance while you were editing. Reload to see the latest values, then re-apply your changes.";
        }
        catch (Exception ex) { _error = ex.Message; }
        finally { _busy = false; }
    }

    private async Task DeleteAsync()
    {
        if (IsNew) return;
        _busy = true; _error = null;
        try
        {
            await using var db = await DbFactory.CreateDbContextAsync();
            var entity = await db.DriverInstances.FirstOrDefaultAsync(
                d => d.ClusterId == ClusterId && d.DriverInstanceId == DriverInstanceId);
            if (entity is null) { Nav.NavigateTo($"/clusters/{ClusterId}/drivers"); return; }
            db.Entry(entity).Property(e => e.RowVersion).OriginalValue = _form.RowVersion;
            db.DriverInstances.Remove(entity);
            await db.SaveChangesAsync();
            Nav.NavigateTo($"/clusters/{ClusterId}/drivers");
        }
        catch (DbUpdateConcurrencyException)
        {
            _error = "Another user changed this driver instance while you were viewing it. Reload before deleting.";
        }
        catch (Exception ex)
        {
            _error = $"Delete failed: {ex.Message}. (Likely because equipment/tags still reference this driver — remove them first.)";
        }
        finally { _busy = false; }
    }

    private string SerializeCurrentConfig()
        => System.Text.Json.JsonSerializer.Serialize(_form.ToOptions(), _jsonOpts);

    private static S7DriverOptions? TryDeserialize(string json)
    {
        try { return System.Text.Json.JsonSerializer.Deserialize<S7DriverOptions>(json, _jsonOpts); }
        catch { return null; }
    }

    public sealed class FormModel
    {
        // Connection
        public string Host { get; set; } = "127.0.0.1";
        public int Port { get; set; } = 102;
        public S7CpuType CpuType { get; set; } = S7CpuType.S71500;
        public short Rack { get; set; } = 0;
        public short Slot { get; set; } = 0;
        public int TimeoutSeconds { get; set; } = 5;

        // Probe
        public bool ProbeEnabled { get; set; } = true;
        public int ProbeIntervalSeconds { get; set; } = 5;
        public int ProbeTimeoutSeconds { get; set; } = 2;
        public int AdminProbeTimeoutSeconds { get; set; } = 5;

        // Tags JSON view (read-only)
        public string? TagsJson { get; set; }

        // Preserved originals (round-tripped unchanged from original options)
        private IReadOnlyList<S7TagDefinition> _tags = [];

        // Common
        public string? ResilienceConfig { get; set; }
        public byte[] RowVersion { get; set; } = [];

        public static FormModel FromOptions(S7DriverOptions o)
        {
            string? tagsJson = o.Tags.Count == 0 ? null
                : System.Text.Json.JsonSerializer.Serialize(o.Tags,
                    new System.Text.Json.JsonSerializerOptions
                    {
                        PropertyNamingPolicy = System.Text.Json.JsonNamingPolicy.CamelCase,
                        WriteIndented = true,
                    });
            return new FormModel
            {
                Host = o.Host,
                Port = o.Port,
                CpuType = o.CpuType,
                Rack = o.Rack,
                Slot = o.Slot,
                TimeoutSeconds = (int)o.Timeout.TotalSeconds,
                ProbeEnabled = o.Probe.Enabled,
                ProbeIntervalSeconds = (int)o.Probe.Interval.TotalSeconds,
                ProbeTimeoutSeconds = (int)o.Probe.Timeout.TotalSeconds,
                AdminProbeTimeoutSeconds = o.ProbeTimeoutSeconds,
                TagsJson = tagsJson,
                _tags = o.Tags,
            };
        }

        public S7DriverOptions ToOptions() => new()
        {
            Host = Host,
            Port = Port,
            CpuType = CpuType,
            Rack = Rack,
            Slot = Slot,
            Timeout = TimeSpan.FromSeconds(TimeoutSeconds),
            Probe = new S7ProbeOptions
            {
                Enabled = ProbeEnabled,
                Interval = TimeSpan.FromSeconds(ProbeIntervalSeconds),
                Timeout = TimeSpan.FromSeconds(ProbeTimeoutSeconds),
            },
            ProbeTimeoutSeconds = AdminProbeTimeoutSeconds,
            Tags = _tags,
        };
    }
}