surrealpatch/cli/sql.go

// Copyright © 2016 Abcum Ltd
//
// 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 cli

import (
	"fmt"
	"net/http"
	"os"

	"io/ioutil"

	"github.com/spf13/cobra"

	"github.com/abcum/surreal/log"
)

var sqlCmd = &cobra.Command{
	Use:     "sql [flags] <file>",
	Short:   "Execute a SQL script against an existing database",
	Example: "  surreal sql --auth root:root script.sql",
	RunE: func(cmd *cobra.Command, args []string) (err error) {

		var fle *os.File
		var req *http.Request
		var res *http.Response

		// Ensure that the command has a filepath
		// as the output file argument. If no filepath
		// has been provided then return an error.

		if len(args) != 1 {
			log.Fatalln("No filepath provided.")
			return
		}

		// Attempt to open or create the specified file
		// in write-only mode, and if there is a problem
		// creating the file, then return an error.

		if fle, err = os.OpenFile(args[0], os.O_RDONLY, 0644); err != nil {
			log.Fatalln("SQL failed - please check the filepath and try again.")
			return
		}

		// Ensure that we properly close the file handle
		// when we have finished with the file so that
		// the file descriptor is released.

		defer fle.Close()

		// Check to see if the http request type has
		// been specified as eith 'http' or 'https'
		// as these are the only supported schemes.

		if opts.DB.Type != "http" && opts.DB.Type != "https" {
			log.Fatalln("Connection failed - please specify 'http' or 'https' for the scheme.")
			return
		}

		// Configure the export connection endpoint url
		// and specify the authentication header using
		// basic auth for root login.

		url := fmt.Sprintf("%s://%s@%s:%s/sql", opts.DB.Type, opts.Auth.Auth, opts.DB.Host, opts.DB.Port)

		// Create a new http request object that we
		// can use to connect to the import endpoint
		// using a POST http request type.

		if req, err = http.NewRequest("POST", url, fle); err != nil {
			log.Fatalln("Connection failed - check the connection details and try again.")
			return
		}

		// Specify that the request is an octet stream
		// so that we can stream the file contents to
		// the server without reading the whole file.

		switch opts.Format.Type {
		case "pack":
			req.Header.Set("Content-Type", "application/msgpack")
		case "json":
			req.Header.Set("Content-Type", "application/json")
		case "cork":
			req.Header.Set("Content-Type", "application/cork")
		default:
			req.Header.Set("Content-Type", "text/plain")
		}

		// Attempt to dial the import endpoint and
		// if there is an error then stop execution
		// and return the connection error.

		if res, err = http.DefaultClient.Do(req); err != nil {
			log.Fatalln("Connection failed - check the connection details and try again.")
			return
		}

		// Ensure that we close the body, otherwise
		// if the Body is not closed, the Client can
		// not use the underlying transport again.

		defer res.Body.Close()

		// Ensure that we didn't receive a 401 status
		// code back from the server, otherwise there
		// was a problem with our authentication.

		if res.StatusCode == 401 {
			log.Fatalln("Authentication failed - check the connection details and try again.")
			return
		}

		// Ensure that we received a http 200 status
		// code back from the server, otherwise there
		// was a problem with our request.

		if res.StatusCode != 200 {
			bdy, _ := ioutil.ReadAll(res.Body)
			log.Fatalf("%s", bdy)
			return
		}

		// Otherwise let's output the whole response
		// body to the terminal, using the output
		// format specified in the request.

		if bdy, _ := ioutil.ReadAll(res.Body); len(bdy) > 0 {
			log.SetLevel("INFO")
			log.Infof("%s", bdy)
		}

		return

	},
}

func init() {

	sqlCmd.PersistentFlags().StringVarP(&opts.Auth.Auth, "auth", "a", "root:root", "Master authentication details to use when connecting.")
	sqlCmd.PersistentFlags().StringVar(&opts.DB.Type, "scheme", "https", "HTTP connection scheme to use to connect to the database.")
	sqlCmd.PersistentFlags().StringVar(&opts.DB.Host, "host", "surreal.io", "Database server host to connect to.")
	sqlCmd.PersistentFlags().StringVar(&opts.DB.Port, "port", "443", "Database server port to connect to.")

	sqlCmd.PersistentFlags().StringVar(&opts.Format.Type, "format", "text", "The output format for the server response data.")

}