restart.rb

module Brainlet
  module Service
    module_function

    def restart(config)
      cmd    = config[:cmd]
      prompt = config[:prompt]

      name = ""

      printer_profiles = []
      Dir.chdir("/home/#{ENV['SUDO_USER']}/configs") do
        printer_profiles = Dir.glob("*").select { |f| File.directory? f }
      end

      loop do
        name = prompt.select("Which printer would you like to restart?", printer_profiles)
        break if prompt.yes?("Are you sure you would like to restart #{name}?")
      end

      cmd.run  "systemctl restart #{name}-klipper.service"
      cmd.run  "systemctl restart #{name}-moonraker.service"
      cmd.run! "systemctl restart #{name}-webcam.service"
    end

    def restart_all(config)
      cmd    = config[:cmd]
      prompt = config[:prompt]

      return if prompt.yes?("Are you sure you would like to restart all services?")

      printer_profiles = []
      Dir.chdir("/home/#{ENV['SUDO_USER']}/configs") do
        printer_profiles = Dir.glob("*").select { |f| File.directory? f }
      end

      printer_profiles.each do |name|
        cmd.run  "systemctl restart #{name}-klipper.service"
        cmd.run  "systemctl restart #{name}-moonraker.service"
        cmd.run! "systemctl restart #{name}-webcam.service"
      end
    end
  end
end