Class: DO

Inherits:
Thor
  • Object
show all
Defined in:
lib/dockit/digitalocean.rb

Constant Summary collapse

USERNAME = 
'root'.freeze
REMOTE_CMDS = 
%w[start push create].freeze

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.remote_required?(extra_cmds = []) ⇒ Boolean

Returns:

  • (Boolean) 


9
10
11
12
 
# File 'lib/dockit/digitalocean.rb', line 9

def self.remote_required?(extra_cmds=[])
  ARGV[0] == 'do' && (
    REMOTE_CMDS.include?(ARGV[1]) || extra_cmds.include?(ARGV[1]))
end

Instance Method Details

#create ⇒ Object 



25
26
27
28
29
30
31
32
33
34
35
36
37
38
 
# File 'lib/dockit/digitalocean.rb', line 25

def create
  if find(options.remote)
    say  "Droplet #{options.remote} exists. Please destroy it first.", :red
    exit 1
  end
  say "creating droplet: #{options.remote}"
  d = client.droplets.create(DropletKit::Droplet.new(
                          name: options.remote,
                          region: options.region,
                          size: options[:size],
                          image: 'docker',
                          ssh_keys: client.ssh_keys.all.collect(&:id)))
  say  [d.id, d.status, d.name].join(' ')
end

#destroy ⇒ Object 



51
52
53
54
55
56
57
58
59
 
# File 'lib/dockit/digitalocean.rb', line 51

def destroy
  force = options[:force]
  say "Destroying droplet: #{options.remote}", force ? :red : nil
  if force || yes?("Are you sure?", :red)
    client.droplets.delete(id: find(options.remote).id)
  else
    say "#{options.remote} not destroyed", :red
  end
end

#list ⇒ Object 



41
42
43
44
45
46
47
 
# File 'lib/dockit/digitalocean.rb', line 41

def list
  l = client.droplets.all.collect do |d|
    [d.id, d.name, d.status, d.networks[:v4].first.ip_address]
  end
  l.unshift %w[id name status ip]
  print_table l
end

#push(*args) ⇒ Object 



65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
 
# File 'lib/dockit/digitalocean.rb', line 65

def push(*args)
  args = dockit.services.keys if args.empty?
  say "Pushing to #{options.remote} as #{options.user}", :green
  say "Processing images for #{args}"
  args.each do |k|
    s = service(k)
    unless s.image
      say ". #{k}: No image!", :red
      next
    end
    name    = s.config.get(:build, :t)
    unless name.present?
      say ". #{k}: not a local build", :red
      next
    end
    id      = s.image.id
    msg     =  "#{k}(#{id[0..11]}[#{name}]):"
    if ssh(options.remote, options.user,
           "docker images --no-trunc | grep #{id} > /dev/null")
      say ". #{msg} exists"
    else
      if options.backup
        tag = "#{k}:#{options.tag}"
        say "#{msg} tagging as #{tag}"
        ssh(options.remote, options.user, "docker tag #{name} #{tag}")
      end
      say "#{msg} pushing"
      ssh(options.remote, options.user, 'docker load', "docker save #{name}")
    end
  end
end

#start(name) ⇒ Object 



100
101
102
103
104
105
106
107
108
109
110
111
112
 
# File 'lib/dockit/digitalocean.rb', line 100

def start(name)
  s     = service(name)
  name  = s.name
  links = config(s, :run,    :Links, 'l')
  binds = config(s, :run,    :Binds, 'v')

  env   = config(s, :create, :Env,   'e')
  env   << (options[:vars]||{}).collect { |k,v| ['-e', "#{k}='#{v}'"]}
  env   << ['-e', "ENV='#{options.env}'"]

  cmd = ['docker', 'run', env, links, binds].join(' ')
  ssh(options.remote, options.user, cmd)
end