class DropletKit::Droplet

def network_for(type, publicity)
  networks = case type
             when :v4 then self.networks.v4
             when :v6 then self.networks.v6
             end
  networks.find do |network|
    network.type == publicity
  end
end

def private_ip
  network = network_for(:v4, 'private')
  network&.ip_address
end

def public_ip
  network = network_for(:v4, 'public')
  network&.ip_address
end