335cd89ebe
1. **Refactoring of Database Interaction Methods** - Refactored the `store_services` method in the `DatabaseManager` class to handle an array of services instead of a hash. This change was made to simplify the interaction with the database and to avoid unnecessary complexity in the data structure. - The `store_services` method now iterates over an array of services and inserts each service into the database with a default status of true. This design choice was made to ensure that all services are active by default. 2. **Modification of Database Schema** - Modified the `create_services_table` method in the `DatabaseManager` class to create a table with only two columns: `id` and `services`. This change was made to align the database schema with the new data structure used in the `store_services` method. - The `status` column was removed from the `services` table because the status of all services is now assumed to be true by default. 3. **Error Handling and Debugging** - Encountered a `Sequel::DatabaseError` with the message "Operand should contain 1 column(s)" during the execution of the `store_services` method. This error was caused by an attempt to insert a hash into a single database column. - The error was resolved by refactoring the `store_services` method to handle an array of services instead of a hash. 4. **Unorthodox Design Choices** - The decision to use an array of services instead of a hash and to assume that the status of all services is true by default may seem unorthodox. However, these design choices were made to simplify the interaction with the database and to avoid unnecessary complexity in the data structure. - These design choices also helped to resolve the `Sequel::DatabaseError` that was encountered during the execution of the `store_services` method. This commit represents a significant refactoring of the database interaction methods in the NETRAVE project. The changes made in this commit have simplified the interaction with the database and have resolved a `Sequel::DatabaseError` that was encountered during the execution of the `store_services` method.
109 lines
2.9 KiB
Ruby
109 lines
2.9 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'curses'
|
|
require 'dynamic_curses_input'
|
|
require 'dotenv'
|
|
require_relative 'database_manager'
|
|
require_relative 'system_information_gather'
|
|
require_relative 'utilities'
|
|
|
|
# first run class
|
|
class FirstRunInit
|
|
include Utilities
|
|
include Curses
|
|
|
|
def initialize(db_manager = nil)
|
|
@db_manager = db_manager || DatabaseManager.new
|
|
@info_gatherer = SystemInformationGather.new(@db_manager)
|
|
end
|
|
|
|
def run
|
|
first_run_setup
|
|
end
|
|
|
|
def first_run_setup # rubocop:disable Metrics/MethodLength
|
|
db_details = ask_for_db_details
|
|
|
|
until @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 = ask_for_db_details
|
|
end
|
|
|
|
@db_manager.create_system_info_table
|
|
@db_manager.create_services_table
|
|
|
|
uplink_speed = @info_gatherer.ask_for_uplink_speed
|
|
downlink_speed = @info_gatherer.ask_for_downlink_speed
|
|
total_bandwidth = calculate_total_bandwidth(uplink_speed, downlink_speed)
|
|
services = @info_gatherer.ask_for_services
|
|
|
|
system_info = {
|
|
uplink_speed:,
|
|
downlink_speed:,
|
|
total_bandwidth:
|
|
}
|
|
|
|
@db_manager.store_system_info(system_info)
|
|
@db_manager.store_services(services)
|
|
end
|
|
|
|
def ask_for_db_details # rubocop:disable Metrics/MethodLength, Metrics/AbcSize
|
|
Curses.clear
|
|
|
|
Curses.setpos(1, 0)
|
|
Curses.addstr('Please enter your database username: ')
|
|
Curses.refresh
|
|
username = DCI.catch_input(true)
|
|
|
|
Curses.setpos(2, 0)
|
|
Curses.addstr('Please enter your database password: ')
|
|
Curses.refresh
|
|
Curses.noecho
|
|
password = DCI.catch_input(false)
|
|
Curses.echo
|
|
|
|
Curses.setpos(3, 0)
|
|
Curses.addstr('Please enter your database name: ')
|
|
Curses.refresh
|
|
database = DCI.catch_input(true)
|
|
|
|
# Generate a secret key
|
|
key = generate_key
|
|
|
|
# Encrypt the password
|
|
encrypted_password = encrypt_string_chacha20(password, key)
|
|
|
|
{ username:, password: encrypted_password, key:, database: }
|
|
end
|
|
|
|
def write_db_details_to_config_file(db_details)
|
|
# Write the database details to the .env file
|
|
File.open('.env', 'w') do |file|
|
|
file.puts "DB_USERNAME=#{db_details[:username]}"
|
|
file.puts "DB_PASSWORD=#{db_details[:password]}"
|
|
file.puts "DB_SECRET_KEY=#{db_details[:key]}"
|
|
file.puts "DB_DATABASE=#{db_details[:database]}"
|
|
end
|
|
end
|
|
|
|
def ask_for_default_mode
|
|
loop do
|
|
Curses.setpos(8, 0)
|
|
Curses.addstr('Please enter the default mode (TUI, GUI, or WebApp): ')
|
|
Curses.refresh
|
|
mode = Curses.getstr.strip.downcase
|
|
return mode if valid_mode?(mode)
|
|
|
|
Curses.setpos(9, 0)
|
|
Curses.addstr("Whoops! That didn't appear to be a valid mode. Please try again!")
|
|
Curses.refresh
|
|
end
|
|
end
|
|
|
|
def valid_mode?(mode)
|
|
%w[tui gui webapp].include?(mode)
|
|
end
|
|
end
|