Merge branch 'main' of giggh:gigalixir/gigalixir-cli

Author: gigatim

gigalixir/__init__.py

@@ -219,7 +219,7 @@ def cli(ctx, env):
 
     if env == "prod":
         stripe.api_key = 'pk_live_45dmSl66k4xLy4X4yfF3RVpd'
-        host = "https://api.gigalixir.com"
+        host = os.environ.get("GIGALIXIR_CLI__HOST", "https://api.gigalixir.com")
     elif env == "dev":
         stripe.api_key = 'pk_test_6tMDkFKTz4N0wIFQZHuzOUyW'
         host = "http://apiserver:3200"
@@ -1162,6 +1162,48 @@ def pg_upgrade(ctx, app_name, database_id, desired_version):
     """
     gigalixir_database.upgrade(ctx.obj['session'], app_name, database_id, desired_version)
 
+@cli.command(name='pg:restart')
+@click.option('-a', '--app_name', envvar="GIGALIXIR_APP")
+@click.option('-d', '--database_id', required=True)
+@click.option('-y', '--yes', is_flag=True)
+@click.pass_context
+@report_errors
+@detect_app_name
+def pg_restart(ctx, app_name, database_id, yes):
+    """
+    Restart database. Find the database id by running `gigalixir pg`
+    """
+    logging.getLogger("gigalixir-cli").info("WARNING: Restarting your database will cause temporary downtime.")
+    if yes or click.confirm('Do you want to restart your database?'):
+        gigalixir_database.restart(ctx.obj['session'], app_name, database_id)
+
+@cli.command(name='pg:stop')
+@click.option('-a', '--app_name', envvar="GIGALIXIR_APP")
+@click.option('-d', '--database_id', required=True)
+@click.option('-y', '--yes', is_flag=True)
+@click.pass_context
+@report_errors
+@detect_app_name
+def pg_stop(ctx, app_name, database_id, yes):
+    """
+    Stop database. Find the database id by running `gigalixir pg`
+    """
+    logging.getLogger("gigalixir-cli").info("WARNING: Stopping your database will make it unavailable until you start it again.")
+    if yes or click.confirm('Do you want to stop your database?'):
+        gigalixir_database.stop(ctx.obj['session'], app_name, database_id)
+
+@cli.command(name='pg:start')
+@click.option('-a', '--app_name', envvar="GIGALIXIR_APP")
+@click.option('-d', '--database_id', required=True)
+@click.pass_context
+@report_errors
+@detect_app_name
+def pg_start(ctx, app_name, database_id):
+    """
+    Start database. Find the database id by running `gigalixir pg`
+    """
+    gigalixir_database.start(ctx.obj['session'], app_name, database_id)
+
 @cli.command(name='stack:set')
 @click.option('-a', '--app_name', envvar="GIGALIXIR_APP")
 @click.option('-s', '--stack', required=True)

gigalixir/database.py

@@ -127,6 +127,28 @@ def restore(session, app_name, database_id, backup_id):
         data = json.loads(r.text)["data"]
         presenter.echo_json(data)
 
+def restart(session, app_name, database_id):
+    r = session.put('/api/apps/%s/databases/%s/restart' % (quote(app_name.encode('utf-8')), quote(database_id.encode('utf-8'))))
+    if r.status_code != 200:
+        if r.status_code == 401:
+            raise auth.AuthException()
+        raise Exception(r.text)
+
+def stop(session, app_name, database_id):
+    r = session.put('/api/apps/%s/databases/%s/stop' % (quote(app_name.encode('utf-8')), quote(database_id.encode('utf-8'))))
+    if r.status_code != 200:
+        if r.status_code == 401:
+            raise auth.AuthException()
+        raise Exception(r.text)
+
+def start(session, app_name, database_id):
+    r = session.put('/api/apps/%s/databases/%s/start' % (quote(app_name.encode('utf-8')), quote(database_id.encode('utf-8'))))
+    if r.status_code != 200:
+        if r.status_code == 401:
+            raise auth.AuthException()
+        raise Exception(r.text)
+    logging.getLogger("gigalixir-cli").info("Starting database. Please allow a few minutes for it to become available.")
+
 def upgrade(session, app_name, database_id, desired_version):
     body = {
         "version": desired_version

test/test_gigalixir.py

@@ -678,6 +678,60 @@ def test_scale_database():
     expect(httpretty.has_request()).to.be.true
     expect(httpretty.last_request().body.decode()).to.equal('{"size": 8.0}')
 
+@httpretty.activate
+def test_restart_database():
+    httpretty.register_uri(httpretty.PUT, 'https://api.gigalixir.com/api/apps/fake-app-name/databases/fake-database-id/restart', body='{}', content_type='application/json')
+    runner = CliRunner()
+    result = runner.invoke(gigalixir.cli, ['pg:restart', '-a', 'fake-app-name', '-d', 'fake-database-id', '-y'])
+    assert result.exit_code == 0
+    expect(httpretty.has_request()).to.be.true
+
+@httpretty.activate
+def test_restart_database_confirm():
+    httpretty.register_uri(httpretty.PUT, 'https://api.gigalixir.com/api/apps/fake-app-name/databases/fake-database-id/restart', body='{}', content_type='application/json')
+    runner = CliRunner()
+    result = runner.invoke(gigalixir.cli, ['pg:restart', '-a', 'fake-app-name', '-d', 'fake-database-id'], input="y\n")
+    assert result.exit_code == 0
+    expect(httpretty.has_request()).to.be.true
+
+@httpretty.activate
+def test_restart_database_deny():
+    runner = CliRunner()
+    result = runner.invoke(gigalixir.cli, ['pg:restart', '-a', 'fake-app-name', '-d', 'fake-database-id'], input="n\n")
+    assert result.exit_code == 0
+    expect(httpretty.has_request()).to.be.false
+
+@httpretty.activate
+def test_stop_database():
+    httpretty.register_uri(httpretty.PUT, 'https://api.gigalixir.com/api/apps/fake-app-name/databases/fake-database-id/stop', body='{}', content_type='application/json')
+    runner = CliRunner()
+    result = runner.invoke(gigalixir.cli, ['pg:stop', '-a', 'fake-app-name', '-d', 'fake-database-id', '-y'])
+    assert result.exit_code == 0
+    expect(httpretty.has_request()).to.be.true
+
+@httpretty.activate
+def test_stop_database_confirm():
+    httpretty.register_uri(httpretty.PUT, 'https://api.gigalixir.com/api/apps/fake-app-name/databases/fake-database-id/stop', body='{}', content_type='application/json')
+    runner = CliRunner()
+    result = runner.invoke(gigalixir.cli, ['pg:stop', '-a', 'fake-app-name', '-d', 'fake-database-id'], input="y\n")
+    assert result.exit_code == 0
+    expect(httpretty.has_request()).to.be.true
+
+@httpretty.activate
+def test_stop_database_deny():
+    runner = CliRunner()
+    result = runner.invoke(gigalixir.cli, ['pg:stop', '-a', 'fake-app-name', '-d', 'fake-database-id'], input="n\n")
+    assert result.exit_code == 0
+    expect(httpretty.has_request()).to.be.false
+
+@httpretty.activate
+def test_start_database():
+    httpretty.register_uri(httpretty.PUT, 'https://api.gigalixir.com/api/apps/fake-app-name/databases/fake-database-id/start', body='{}', content_type='application/json')
+    runner = CliRunner()
+    result = runner.invoke(gigalixir.cli, ['pg:start', '-a', 'fake-app-name', '-d', 'fake-database-id'])
+    assert result.exit_code == 0
+    expect(httpretty.has_request()).to.be.true
+
 @httpretty.activate
 def test_account():
     httpretty.register_uri(httpretty.GET, 'https://api.gigalixir.com/api/users', body='{"data":{}}', content_type='application/json')