Class: WindowsShutdownTimer::ShutdownStarter

Inherits:
Object
  • Object
show all
Defined in:
lib/windows-shutdown-timer.rb

Overview

Shutdown!

Constant Summary collapse

INSOMNIA = 
'Insomnia.exe'
PATH_TO_TEMP = 
Dir.tmpdir + '/windows-shutdown-timer'
PATH_TO_INSOMNIA = 
PATH_TO_TEMP + '/' + INSOMNIA
PATH = 
'insomnia/64-bit'
COMMAND_SHUTDOWN = 
'shutdown /s /t '
COMMAND_CANCEL_SHUTDOWN = 
'shutdown /a'
COMMAND_INSOMNIA = 
"start #{PATH_TO_INSOMNIA}"

Instance Method Summary collapse

Constructor Details

#initialize(arguments) ⇒ ShutdownStarter

Returns a new instance of ShutdownStarter.



18
19
20
21
22
23
24
 
# File 'lib/windows-shutdown-timer.rb', line 18

def initialize(arguments)

  # Get the first arg as the time
  @time = %w(-h --help -c).include?(arguments.first) ? nil : arguments.shift

  create_options_parser(arguments)
end

Instance Method Details

#create_options_parser(args) ⇒ Object 



26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
 
# File 'lib/windows-shutdown-timer.rb', line 26

def create_options_parser(args)
  args.options do |opts|
    opts.banner = 'Usage: windows-shutdown-timer MINUTES_UNTIL_SHUTDOWN [OPTIONS]'
    opts.separator ''
    opts.separator 'Options'
    opts.on('-c', '--cancel', 'Cancel a pending shutdown') do
      @time = 0
    end
    opts.on('-i', '--insomnia', 'Just runs Insomnia') do
      @time = -1
    end
    opts.on('-h', '--help', 'Displays help') do
      puts opts.help
      exit
    end
    opts.parse!
  end
end

#download(base_url, path, file_name) ⇒ Object 



71
72
73
74
75
76
77
78
 
# File 'lib/windows-shutdown-timer.rb', line 71

def download(base_url, path, file_name)
  Net::HTTP.start(base_url) do |http|
    resp = http.get(path)
    open(file_name, 'wb') do |file|
      file.write(resp.body)
    end
  end
end

#start_timer ⇒ Object 



45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
 
# File 'lib/windows-shutdown-timer.rb', line 45

def start_timer
  unless File.file?(PATH_TO_INSOMNIA)
    puts 'Downloading Insomnia.exe'
    download('dlaa.me', '/Samples/Insomnia/Insomnia.zip', 'Insomnia.zip')
    unzip('Insomnia.zip', 'insomnia')
    FileUtils.mkdir(PATH_TO_TEMP)
    FileUtils.mv('insomnia/64-bit/Insomnia.exe', PATH_TO_INSOMNIA)
    # Cleanup
    FileUtils.rm_rf('insomnia')
    FileUtils.rm('Insomnia.zip')
  end
  if @time == nil
    puts 'Please enter the number of minutes until shutdown (0 to cancel):'
    @time = gets
  end
  if @time.to_i == 0
    `#{COMMAND_CANCEL_SHUTDOWN}`
  elsif @time.to_i == -1
    `#{COMMAND_INSOMNIA}`
  else
    time_in_seconds = @time.to_i * 60
    `#{COMMAND_SHUTDOWN} #{time_in_seconds}`
    `#{COMMAND_INSOMNIA}`
  end
end

#unzip(file, destination) ⇒ Object 



80
81
82
83
84
85
86
87
88
 
# File 'lib/windows-shutdown-timer.rb', line 80

def unzip(file, destination)
  Zip::File.open(file) do |zip_file|
    zip_file.each do |f|
      f_path = File.join(destination, f.name)
      FileUtils.mkdir_p(File.dirname(f_path))
      zip_file.extract(f, f_path) unless File.exist?(f_path)
    end
  end
end