class Kitsune::Kit::Commands::InstallDockerEngine

def create
  filled_options = Kitsune::Kit::OptionsBuilder.build(
    options,
    required: [:server_ip],
    defaults: Kitsune::Kit::Defaults.ssh
  )
  with_ssh_connection(filled_options) do |ssh|
    perform_setup(ssh)
  end
end

def perform_rollback(ssh)
  output = ssh.exec! <<~EOH
    set -e
    sudo mkdir -p /usr/local/backups
    sudo chown deploy:deploy /usr/local/backups
    BACKUP_DIR="/usr/local/backups"
    SCRIPT_ID="install_docker_engine"
    BEFORE_FILE="${BACKUP_DIR}/${SCRIPT_ID}.before"
    AFTER_FILE="${BACKUP_DIR}/${SCRIPT_ID}.after"
    TARGET_PKGS=(docker-ce docker-ce-cli containerd.io)
    echo "✍🏻 TARGET_PKGS=(\${TARGET_PKGS[*]})"
    if [ -f "$AFTER_FILE" ]; then
      to_remove=()
      for pkg in "\${TARGET_PKGS[@]}"; do
        if dpkg -l "\$pkg" &>/dev/null && ! grep -Fxq "\$pkg" "$BEFORE_FILE"; then
          to_remove+=("\$pkg")
        fi
      done
      if [ \${#to_remove[@]} -gt 0 ]; then
        echo "🔁 Removing Docker packages..."
        sudo apt-get remove -y "\${to_remove[@]}" && echo "   - removed: \${to_remove[*]}"
      else
        echo "   - no Docker packages to remove"
      fi
      sudo rm -f "$BEFORE_FILE" "$AFTER_FILE" && echo "   - cleanup markers"
    else
      echo "   - no marker for $SCRIPT_ID, skipping"
    fi
    echo "✅ Rollback done"
  EOH
  say output
  say "✅ Docker Engine rollback completed", :green
end

def perform_setup(ssh)
  output = ssh.exec! <<~EOH
    set -e
    sudo mkdir -p /usr/local/backups
    sudo chown deploy:deploy /usr/local/backups
    BACKUP_DIR="/usr/local/backups"
    SCRIPT_ID="install_docker_engine"
    BEFORE_FILE="${BACKUP_DIR}/${SCRIPT_ID}.before"
    AFTER_FILE="${BACKUP_DIR}/${SCRIPT_ID}.after"
    TARGET_PKGS=(docker-ce docker-ce-cli containerd.io)
    echo "✍🏻 TARGET_PKGS=(\${TARGET_PKGS[*]})"
    if [ ! -f "$AFTER_FILE" ]; then
      for pkg in "\${TARGET_PKGS[@]}"; do
        if dpkg -l "\$pkg" &>/dev/null; then
          echo "\$pkg" >> "$BEFORE_FILE"
        fi
      done
      if [ ! -f /usr/share/keyrings/docker-archive-keyring.gpg ]; then
        curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo gpg --dearmor -o /usr/share/keyrings/docker-archive-keyring.gpg
      fi
      if [ ! -f /etc/apt/sources.list.d/docker.list ]; then
        echo "deb [arch=$(dpkg --print-architecture) signed-by=/usr/share/keyrings/docker-archive-keyring.gpg] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable" | sudo tee /etc/apt/sources.list.d/docker.list > /dev/null
      fi
      echo "✍🏻 Installing Docker Engine..."
      sudo apt-get update -y
      sudo apt-get install -y "\${TARGET_PKGS[@]}"
      sudo touch "$AFTER_FILE" && echo "   - marker at $AFTER_FILE"
      echo "✅ Docker installed"
    else
      echo "🔄 Docker already set up, ensuring latest..."
      sudo apt-get update -y
      sudo apt-get install -y "\${TARGET_PKGS[@]}"
      echo "✅ Docker packages are current"
    fi
  EOH
  say output
  say "✅ Docker Engine setup completed", :green
end

def rollback
  filled_options = Kitsune::Kit::OptionsBuilder.build(
    options,
    required: [:server_ip],
    defaults: Kitsune::Kit::Defaults.ssh
  )
  with_ssh_connection(filled_options) do |ssh|
    perform_rollback(ssh)
  end
end

def with_ssh_connection(filled_options)
  server = filled_options[:server_ip]
  port   = filled_options[:ssh_port]
  key    = File.expand_path(filled_options[:ssh_key_path])
  say "🔑 Connecting as deploy@#{server}:#{port}", :green
  Net::SSH.start(server, "deploy", port: port, keys: [key], non_interactive: true, timeout: 5) do |ssh|
    yield ssh
  end
end