Cortex local

cli/cluster/delete.go

@@ -0,0 +1,69 @@
+/*
+Copyright 2020 Cortex Labs, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package cluster
+
+import (
+	"fmt"
+
+	"github.com/cortexlabs/cortex/pkg/lib/errors"
+	"github.com/cortexlabs/cortex/pkg/lib/json"
+	"github.com/cortexlabs/cortex/pkg/lib/prompt"
+	s "github.com/cortexlabs/cortex/pkg/lib/strings"
+	"github.com/cortexlabs/cortex/pkg/operator/schema"
+)
+
+func Delete(operatorConfig OperatorConfig, apiName string, keepCache bool, force bool) (schema.DeleteResponse, error) {
+	if !force {
+		readyReplicas := getReadyReplicasOrNil(operatorConfig, apiName)
+		if readyReplicas != nil && *readyReplicas > 2 {
+			prompt.YesOrExit(fmt.Sprintf("are you sure you want to delete %s (which has %d live replicas)?", apiName, *readyReplicas), "", "")
+		}
+	}
+
+	params := map[string]string{
+		"apiName":   apiName,
+		"keepCache": s.Bool(keepCache),
+	}
+
+	httpRes, err := HTTPDelete(operatorConfig, "/delete/"+apiName, params)
+	if err != nil {
+		return schema.DeleteResponse{}, err
+	}
+
+	var deleteRes schema.DeleteResponse
+	err = json.Unmarshal(httpRes, &deleteRes)
+	if err != nil {
+		return schema.DeleteResponse{}, errors.Wrap(err, "/delete", string(httpRes))
+	}
+
+	return deleteRes, nil
+}
+
+func getReadyReplicasOrNil(operatorConfig OperatorConfig, apiName string) *int32 {
+	httpRes, err := HTTPGet(operatorConfig, "/get/"+apiName)
+	if err != nil {
+		return nil
+	}
+
+	var apiRes schema.GetAPIResponse
+	if err = json.Unmarshal(httpRes, &apiRes); err != nil {
+		return nil
+	}
+
+	totalReady := apiRes.Status.Updated.Ready + apiRes.Status.Stale.Ready
+	return &totalReady
+}

cli/cluster/deploy.go

@@ -0,0 +1,46 @@
+/*
+Copyright 2020 Cortex Labs, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package cluster
+
+import (
+	"github.com/cortexlabs/cortex/pkg/lib/errors"
+	"github.com/cortexlabs/cortex/pkg/lib/json"
+	s "github.com/cortexlabs/cortex/pkg/lib/strings"
+	"github.com/cortexlabs/cortex/pkg/operator/schema"
+)
+
+func Deploy(operatorConfig OperatorConfig, configPath string, deploymentBytesMap map[string][]byte, force bool) (schema.DeployResponse, error) {
+	params := map[string]string{
+		"force":      s.Bool(force),
+		"configPath": configPath,
+	}
+	uploadInput := &HTTPUploadInput{
+		Bytes: deploymentBytesMap,
+	}
+
+	response, err := HTTPUpload(operatorConfig, "/deploy", uploadInput, params)
+	if err != nil {
+		return schema.DeployResponse{}, err
+	}
+
+	var deployResponse schema.DeployResponse
+	if err := json.Unmarshal(response, &deployResponse); err != nil {
+		return schema.DeployResponse{}, errors.Wrap(err, "/deploy", string(response))
+	}
+
+	return deployResponse, nil
+}

cli/cluster/errors.go

@@ -0,0 +1,89 @@
+/*
+Copyright 2020 Cortex Labs, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package cluster
+
+import (
+	"fmt"
+	"net/url"
+	"strings"
+
+	"github.com/cortexlabs/cortex/pkg/lib/errors"
+	"github.com/cortexlabs/cortex/pkg/lib/urls"
+)
+
+const (
+	_errStrCantMakeRequest = "unable to make request"
+	_errStrRead            = "unable to read"
+)
+
+func errStrFailedToConnect(u url.URL) string {
+	return "failed to connect to " + urls.TrimQueryParamsURL(u)
+}
+
+const (
+	ErrFailedToConnectOperator       = "cli.failed_to_connect_operator"
+	ErrOperatorSocketRead            = "cli.operator_socket_read"
+	ErrResponseUnknown               = "cli.response_unknown"
+	ErrOperatorResponseUnknown       = "cli.operator_response_unknown"
+	ErrOperatorStreamResponseUnknown = "cli.operator_stream_response_unknown"
+)
+
+func ErrorFailedToConnectOperator(originalError error, envName string, operatorURL string) error {
+	originalErrMsg := ""
+	if originalError != nil {
+		originalErrMsg = urls.TrimQueryParamsStr(errors.Message(originalError)) + "\n"
+	}
+
+	return errors.WithStack(&errors.Error{
+		Kind:    ErrFailedToConnectOperator,
+		Message: fmt.Sprintf("%sfailed to connect to the operator in the %s environment (operator endpoint: %s); run `cortex env configure %s` if you need to update the operator endpoint, `cortex cluster info` to show your operator endpoint, or `cortex cluster up` to create a new cluster", originalErrMsg, envName, operatorURL, envName),
+	})
+}
+
+func ErrorOperatorSocketRead(err error) error {
+	return errors.WithStack(&errors.Error{
+		Kind:    ErrOperatorSocketRead,
+		Message: err.Error(),
+		NoPrint: true,
+	})
+}
+
+func ErrorResponseUnknown(body string, statusCode int) error {
+	msg := body
+	if strings.TrimSpace(body) == "" {
+		msg = fmt.Sprintf("empty response (status code %d)", statusCode)
+	}
+
+	return errors.WithStack(&errors.Error{
+		Kind:    ErrResponseUnknown,
+		Message: msg,
+	})
+}
+
+func ErrorOperatorResponseUnknown(body string, statusCode int) error {
+	return errors.WithStack(&errors.Error{
+		Kind:    ErrOperatorResponseUnknown,
+		Message: fmt.Sprintf("unexpected response from operator (status code %d): %s", statusCode, body),
+	})
+}
+
+func ErrorOperatorStreamResponseUnknown(body string, statusCode int) error {
+	return errors.WithStack(&errors.Error{
+		Kind:    ErrOperatorStreamResponseUnknown,
+		Message: fmt.Sprintf("unexpected response from operator (status code %d): %s", statusCode, body),
+	})
+}

cli/cluster/get.go

@@ -0,0 +1,50 @@
+/*
+Copyright 2020 Cortex Labs, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package cluster
+
+import (
+	"github.com/cortexlabs/cortex/pkg/lib/errors"
+	"github.com/cortexlabs/cortex/pkg/lib/json"
+	"github.com/cortexlabs/cortex/pkg/operator/schema"
+)
+
+func GetAPIs(operatorConfig OperatorConfig) (schema.GetAPIsResponse, error) {
+	httpRes, err := HTTPGet(operatorConfig, "/get")
+	if err != nil {
+		return schema.GetAPIsResponse{}, err
+	}
+
+	var apisRes schema.GetAPIsResponse
+	if err = json.Unmarshal(httpRes, &apisRes); err != nil {
+		return schema.GetAPIsResponse{}, errors.Wrap(err, "/get", string(httpRes))
+	}
+	return apisRes, nil
+}
+
+func GetAPI(operatorConfig OperatorConfig, apiName string) (schema.GetAPIResponse, error) {
+	httpRes, err := HTTPGet(operatorConfig, "/get/"+apiName)
+	if err != nil {
+		return schema.GetAPIResponse{}, err
+	}
+
+	var apiRes schema.GetAPIResponse
+	if err = json.Unmarshal(httpRes, &apiRes); err != nil {
+		return schema.GetAPIResponse{}, errors.Wrap(err, "/get/"+apiName, string(httpRes))
+	}
+
+	return apiRes, nil
+}

cli/cluster/info.go

@@ -0,0 +1,38 @@
+/*
+Copyright 2020 Cortex Labs, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package cluster
+
+import (
+	"github.com/cortexlabs/cortex/pkg/lib/errors"
+	"github.com/cortexlabs/cortex/pkg/lib/json"
+	"github.com/cortexlabs/cortex/pkg/operator/schema"
+)
+
+func Info(operatorConfig OperatorConfig) (*schema.InfoResponse, error) {
+	httpResponse, err := HTTPGet(operatorConfig, "/info")
+	if err != nil {
+		return nil, errors.Wrap(err, "unable to connect to operator", "/info")
+	}
+
+	var infoResponse schema.InfoResponse
+	err = json.Unmarshal(httpResponse, &infoResponse)
+	if err != nil {
+		return nil, errors.Wrap(err, "/info", string(httpResponse))
+	}
+
+	return &infoResponse, nil
+}