AutoMapService.cs

using Bindito.Core;
using Calloatti.Config;
using Timberborn.Automation;
using Timberborn.PlayerDataSystem;
using Timberborn.QuickNotificationSystem;
using Timberborn.RelationSystem;
using Timberborn.SelectionSystem;
using Timberborn.SingletonSystem;
using Timberborn.Localization;
using UnityEngine;
using System;
using System.IO;

namespace Calloatti.AutoTools
{
  public enum MapDisplayState
  {
    Hidden,
    Single,
    Global
  }

  public partial class AutoMapService : ILoadableSingleton, IPostLoadableSingleton, IUnloadableSingleton, IDisposable
  {
    private readonly AutomatorRegistry _automatorRegistry;
    private readonly EventBus _eventBus;
    private readonly AutoMapInputService _inputService;
    private readonly QuickNotificationService _notificationService;
    private readonly EntitySelectionService _selectionService;
    private readonly ILoc _loc;

    private MapDisplayState _currentState = MapDisplayState.Hidden;
    private Automator _singleVisualizedAutomator;
    private Material _lineMaterial;
    private bool _isDirty = true;
    private int _lastActivePartitionId = -1;

    public bool IsReady { get; private set; } = false;

    [Inject]
    public AutoMapService(
        AutomatorRegistry automatorRegistry,
        EventBus eventBus,
        AutoMapInputService inputService,
        QuickNotificationService notificationService,
        EntitySelectionService selectionService,
        ILoc loc)
    {
      _automatorRegistry = automatorRegistry;
      _eventBus = eventBus;
      _inputService = inputService;
      _notificationService = notificationService;
      _selectionService = selectionService;
      _loc = loc;
    }

    public void Load()
    {
      InitializeVisuals();

      try
      {
        _currentState = ModStarter.Config.GetEnum<MapDisplayState>("MapDisplayState");

        // Load visual tuning variables with your new defaults
        _connectionHeightFraction = ModStarter.Config.GetFloat("ConnectionHeightFraction");
        _glowWidthMultiplier = ModStarter.Config.GetFloat("GlowWidthMultiplier");
        _glowAlpha = ModStarter.Config.GetFloat("GlowAlpha");
        _offStateBrightnessMultiplier = ModStarter.Config.GetFloat("OffStateBrightnessMultiplier");

      }
      catch (Exception e)
      {
        Debug.LogWarning($"[AutoTools] Load error: {e.Message}");
      }
    }

    public void PostLoad()
    {
      _eventBus.Register(this);
      _inputService.OnToggleAutoMap += ToggleAutoMap;

      foreach (Automator automator in _automatorRegistry.Automators)
      {
        ((IRelationOwner)automator).RelationsChanged += OnRelationsChanged;
      }

      RefreshVisuals(suppressInfoNotification: true);
      IsReady = true;
    }

    public void Unload() => SaveState();

    public void SaveState()
    {
      try
      {
        if (ModStarter.Config != null)
        {
          ModStarter.Config.Set("MapDisplayState", _currentState);
          ModStarter.Config.Save();
        }
      }
      catch (Exception e)
      {
        Debug.LogError($"[AutoTools] Save error: {e.Message}");
      }
    }

    public void Dispose()
    {
      _eventBus.Unregister(this);
      _inputService.OnToggleAutoMap -= ToggleAutoMap;

      foreach (Automator automator in _automatorRegistry.Automators)
      {
        ((IRelationOwner)automator).RelationsChanged -= OnRelationsChanged;
      }

      OnDispose();
    }
  }
}