class SimpleServerController(object):
def __init__(self, ood_instance):
self.state, _ = SimpleServerState.objects.get_or_create(
ood=ood_instance)
self.mcc = Client(ood_instance)
def start(self):
self.mcc.reset_player_info()
return self._send_cmd('start')
def stop(self):
return self._send_cmd('stop')
def running(self):
response = self._send_cmd('running').lower()
return response == 'true'
def _send_cmd(self, cmd):
buf = ''
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect((self.state.ip_address, self.state.port))
s.sendall('%s\n' % cmd)
while '\n' not in buf:
buf += s.recv(1024)
response, nl, buf = buf.partition('\n')
s.close()
return response
class SimpleServerController(object):
def __init__(self, ood_instance):
self.state, _ = SimpleServerState.objects.get_or_create(
ood=ood_instance)
self.mcc = Client(ood_instance)
def start(self):
self.mcc.reset_player_info()
return self._send_cmd('start')
def stop(self):
return self._send_cmd('stop')
def running(self):
response = self._send_cmd('running').lower()
return response == 'true'
def _send_cmd(self, cmd):
buf = ''
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect((self.state.ip_address, self.state.port))
s.sendall('%s\n' % cmd)
while '\n' not in buf:
buf += s.recv(1024)
response, nl, buf = buf.partition('\n')
s.close()
return response
def __init__(self, ood_instance, data_dir=DEFAULT_DATA_DIR):
self.state, _ = DropletState.objects.get_or_create(ood=ood_instance)
self.mcc = Client(ood_instance)
self.data_dir = os.path.expanduser(data_dir)
if not os.path.exists(self.data_dir):
os.mkdir(self.data_dir)
self.api_key_path = os.path.join(self.data_dir, DROPLET_KEY_FILENAME)
self._snapshot_action = None
self._droplet_ip = None
self._refresh_droplet()
def __init__(self, ood_instance, data_dir=DEFAULT_DATA_DIR):
self.state, _ = DropletState.objects.get_or_create(ood=ood_instance)
self.mcc = Client(ood_instance)
self.data_dir = os.path.expanduser(data_dir)
if not os.path.exists(self.data_dir):
os.mkdir(self.data_dir)
self.api_key_path = os.path.join(self.data_dir, DROPLET_KEY_FILENAME)
self._snapshot_action = None
self._droplet_ip = None
self._refresh_droplet()
def __init__(self, ood_instance):
self.state, _ = SimpleServerState.objects.get_or_create(
ood=ood_instance)
self.mcc = Client(ood_instance)
class DropletController(object):
def __init__(self, ood_instance, data_dir=DEFAULT_DATA_DIR):
self.state, _ = DropletState.objects.get_or_create(ood=ood_instance)
self.mcc = Client(ood_instance)
self.data_dir = os.path.expanduser(data_dir)
if not os.path.exists(self.data_dir):
os.mkdir(self.data_dir)
self.api_key_path = os.path.join(self.data_dir, DROPLET_KEY_FILENAME)
self._snapshot_action = None
self._droplet_ip = None
self._refresh_droplet()
def start(self):
"""Starts up a droplet from the most recent snapshot."""
snapshot = self._find_snapshot()
if snapshot is None:
logging.error('No ood snapshot found.')
return
ssh_key = self._find_ssh_key()
if ssh_key is None:
logging.error('No ood ssh key found.')
return
self.droplet = digitalocean.Droplet(token=self.api_key,
name=self.state.name,
region=self.state.region,
image=snapshot.id,
size_slug='2gb',
ssh_keys=[ssh_key])
self.droplet.create()
def stop(self):
"""Starts the stop-snapshot-destroy process."""
self._refresh_droplet()
logging.info('Stopping Minecraft.')
self._exec_ssh_cmd('supervisorctl stop minecraft')
def shutdown(self):
logging.info('Shutting down host.')
self.droplet.shutdown()
# TODO: Keep the action object and use it to verify state (but
# double check that the state is 'off' after).
# TODO: Call self.droplet.poweroff() if shutdown() fails or takes
# too long.
def snapshot(self, shutdown_error=None):
if self.droplet is None:
logging.error('Cannot start snapshot: no droplet.')
return
if self.snapshot_action:
logging.error('Cannot start snapshot: snapshot in progress.')
return
snapshot_name = '%s-%d' % (self.state.name, time.time())
if shutdown_error:
snapshot_name += '-error'
self.state.snapshot_action_id = self.droplet.take_snapshot(
snapshot_name)['action']['id']
self.state.save()
def destroy(self):
self.droplet.destroy()
def clear_snapshot_action(self):
self.state.snapshot_action_id = None
self.state.save()
def running(self):
return self.mcc.port_open()
def prune_snapshots(self):
snapshots = self._get_snapshots()
for old_snapshot in snapshots[MAX_SNAPSHOTS_PER_INSTANCE:]:
logging.info('Pruning old snapshot %s.' % old_snapshot.name)
old_snapshot.destroy()
@property
def api_key(self):
return file(self.api_key_path).read().strip()
@property
def droplet_ip(self):
if self._droplet_ip:
return self._droplet_ip
# TODO: Should maybe refresh droplet here. Dependencies are unclear.
if self.droplet is None:
return None
for network in self.droplet.networks['v4']:
if network['type'] == 'public':
self._droplet_ip = network['ip_address']
return self._droplet_ip
return None
@property
def manager(self):
return digitalocean.Manager(token=self.api_key)
@property
def snapshot_action(self):
if self.state.snapshot_action_id is None:
return None
get_snapshot_action = (
self._snapshot_action is None or
self._snapshot_action.id != self.state.snapshot_action_id
)
if get_snapshot_action:
try:
self._snapshot_action = digitalocean.Action.get_object(
self.api_key, self.state.snapshot_action_id)
except digitalocean.DataReadError:
self.state.snapshot_action_id = None
self.state.save()
self._snapshot_action = None
return self._snapshot_action
def _find_ssh_key(self):
for key in self.manager.get_all_sshkeys():
if key.name == 'ood':
return key
return None
def _get_snapshots(self):
"""Return snapshots in reverse chronological age (i.e. newest first).
"""
return sorted([img for img in self.manager.get_my_images()
if img.name.startswith('%s-' % self.state.name)],
key=attrgetter('name'), reverse=True)
def _find_snapshot(self):
snapshots = self._get_snapshots()
if not snapshots:
return None
return snapshots[0]
def _refresh_droplet(self):
for droplet in self.manager.get_all_droplets():
if droplet.name == self.state.name:
self.droplet = droplet
if self.state.droplet_ip_address != self.droplet_ip:
self.state.droplet_ip_address = self.droplet_ip
self.state.save()
self.mcc.update_ip_address(
self.state.droplet_ip_address,
)
break
else:
self.droplet = None
def _exec_ssh_cmd(self, cmdline):
pkey_buf = StringIO(self.state.pkey)
client = SSHClient()
client.set_missing_host_key_policy(AutoAddPolicy())
client.connect(self.droplet_ip, username='******',
pkey=RSAKey.from_private_key(pkey_buf))
stdin, stdout, stderr = client.exec_command(cmdline)
for line in stdout:
logging.info(line)
for line in stderr:
logging.info(line)
class DropletController(object):
def __init__(self, ood_instance, data_dir=DEFAULT_DATA_DIR):
self.state, _ = DropletState.objects.get_or_create(ood=ood_instance)
self.mcc = Client(ood_instance)
self.data_dir = os.path.expanduser(data_dir)
if not os.path.exists(self.data_dir):
os.mkdir(self.data_dir)
self.api_key_path = os.path.join(self.data_dir, DROPLET_KEY_FILENAME)
self._snapshot_action = None
self._droplet_ip = None
self._refresh_droplet()
def start(self):
"""Starts up a droplet from the most recent snapshot."""
snapshot = self._find_snapshot()
if snapshot is None:
logging.error('No ood snapshot found.')
return
ssh_key = self._find_ssh_key()
if ssh_key is None:
logging.error('No ood ssh key found.')
return
self.droplet = digitalocean.Droplet(token=self.api_key,
name=self.state.name,
region=self.state.region,
image=snapshot.id,
size_slug='2gb',
ssh_keys=[ssh_key])
self.droplet.create()
def stop(self):
"""Starts the stop-snapshot-destroy process."""
self._refresh_droplet()
logging.info('Stopping Minecraft.')
self._exec_ssh_cmd('supervisorctl stop minecraft')
def shutdown(self):
logging.info('Shutting down host.')
self.droplet.shutdown()
# TODO: Keep the action object and use it to verify state (but
# double check that the state is 'off' after).
# TODO: Call self.droplet.poweroff() if shutdown() fails or takes
# too long.
def snapshot(self, shutdown_error=None):
if self.droplet is None:
logging.error('Cannot start snapshot: no droplet.')
return
if self.snapshot_action:
logging.error('Cannot start snapshot: snapshot in progress.')
return
snapshot_name = '%s-%d' % (self.state.name, time.time())
if shutdown_error:
snapshot_name += '-error'
self.state.snapshot_action_id = self.droplet.take_snapshot(
snapshot_name)['action']['id']
self.state.save()
def destroy(self):
self.droplet.destroy()
def clear_snapshot_action(self):
self.state.snapshot_action_id = None
self.state.save()
def running(self):
return self.mcc.port_open()
def prune_snapshots(self):
snapshots = self._get_snapshots()
for old_snapshot in snapshots[MAX_SNAPSHOTS_PER_INSTANCE:]:
logging.info('Pruning old snapshot %s.' % old_snapshot.name)
old_snapshot.destroy()
@property
def api_key(self):
return file(self.api_key_path).read().strip()
@property
def droplet_ip(self):
if self._droplet_ip:
return self._droplet_ip
# TODO: Should maybe refresh droplet here. Dependencies are unclear.
if self.droplet is None:
return None
for network in self.droplet.networks['v4']:
if network['type'] == 'public':
self._droplet_ip = network['ip_address']
return self._droplet_ip
return None
@property
def manager(self):
return digitalocean.Manager(token=self.api_key)
@property
def snapshot_action(self):
if self.state.snapshot_action_id is None:
return None
get_snapshot_action = (
self._snapshot_action is None
or self._snapshot_action.id != self.state.snapshot_action_id)
if get_snapshot_action:
try:
self._snapshot_action = digitalocean.Action.get_object(
self.api_key, self.state.snapshot_action_id)
except digitalocean.DataReadError:
self.state.snapshot_action_id = None
self.state.save()
self._snapshot_action = None
return self._snapshot_action
def _find_ssh_key(self):
for key in self.manager.get_all_sshkeys():
if key.name == 'ood':
return key
return None
def _get_snapshots(self):
"""Return snapshots in reverse chronological age (i.e. newest first).
"""
return sorted([
img for img in self.manager.get_my_images()
if img.name.startswith('%s-' % self.state.name)
],
key=attrgetter('name'),
reverse=True)
def _find_snapshot(self):
snapshots = self._get_snapshots()
if not snapshots:
return None
return snapshots[0]
def _refresh_droplet(self):
for droplet in self.manager.get_all_droplets():
if droplet.name == self.state.name:
self.droplet = droplet
if self.state.droplet_ip_address != self.droplet_ip:
self.state.droplet_ip_address = self.droplet_ip
self.state.save()
self.mcc.update_ip_address(self.state.droplet_ip_address, )
break
else:
self.droplet = None
def _exec_ssh_cmd(self, cmdline):
pkey_buf = StringIO(self.state.pkey)
client = SSHClient()
client.set_missing_host_key_policy(AutoAddPolicy())
client.connect(self.droplet_ip,
username='******',
pkey=RSAKey.from_private_key(pkey_buf))
stdin, stdout, stderr = client.exec_command(cmdline)
for line in stdout:
logging.info(line)
for line in stderr:
logging.info(line)
def __init__(self, ood_instance):
self.state, _ = SimpleServerState.objects.get_or_create(
ood=ood_instance)
self.mcc = Client(ood_instance)
