require 'curses'
require_relative 'database_manager.rb'
require_relative 'system_information_gather.rb'
require_relative 'utilities.rb'

class FirstRunInit
  include Utilities

  def initialize
    @db_manager = DatabaseManager.new
    @info_gatherer = SystemInformationGather.new
  end

  def first_run_setup
    color = @info_gatherer.ask_for_color
    # TODO: Use the color for something
    db_details = @info_gatherer.ask_for_db_details
    while !@db_manager.test_db_connection(db_details)
      Curses.setpos(4, 0)
      Curses.addstr("Whoops! We couldn't connect to the database with the details you provided. Please try again!")
      Curses.refresh
      db_details = @info_gatherer.ask_for_db_details
    end
    @db_manager.create_system_info_table
    uplink_speed = @info_gatherer.ask_for_uplink_speed
    @db_manager.store_system_info(uplink_speed)
    downlink_speed = @info_gatherer.ask_for_downlink_speed
    @db_manager.store_system_info(downlink_speed)
    total_bandwidth = calculate_total_bandwidth(uplink_speed, downlink_speed)
    @db_manager.store_system_info(total_bandwidth)
    services = @info_gatherer.ask_for_services
    @db_manager.store_system_info(services)
  end

  def ask_for_default_mode
    while true
      Curses.setpos(8, 0)
      Curses.addstr("Please enter the default mode (TUI, GUI, or WebApp): ")
      Curses.refresh
      mode = Curses.getstr.strip.downcase
      if valid_mode?(mode)
        return mode
      else
        Curses.setpos(9, 0)
        Curses.addstr("Whoops! That didn't appear to be a valid mode. Please try again!")
        Curses.refresh
      end
    end
  end

  def valid_mode?(mode)
    ['tui', 'gui', 'webapp'].include?(mode)
  end
end