Add sql command to the command line ui

Add sql command to the command line ui , useful for executing and processing an sql file against a local or remote database.

cli/cli.go

@@ -32,6 +32,7 @@ var mainCmd = &cobra.Command{
 func init() {
 
 	mainCmd.AddCommand(
+		sqlCmd,
 		startCmd,
 		importCmd,
 		exportCmd,

cli/sql.go

@@ -0,0 +1,149 @@
+// 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("Import 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()
+
+		// Configure the export connection endpoint url
+		// and specify the authentication header using
+		// basic auth for root login.
+
+		url := fmt.Sprintf("http://%s@%s:%s/sql", 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().StringVar(&opts.Auth.Auth, "auth", "root:root", "Master authentication details to use when connecting.")
+	sqlCmd.PersistentFlags().StringVar(&opts.DB.Host, "host", "surreal.io", "Database server host to connect to.")
+	sqlCmd.PersistentFlags().StringVar(&opts.DB.Port, "port", "80", "Database server port to connect to.")
+
+	sqlCmd.PersistentFlags().StringVar(&opts.Format.Type, "format", "text", "The output format for the server response data.")
+
+}

cnf/cnf.go

@@ -112,6 +112,10 @@ type Options struct {
 		Join []string // Slice of cluster peers to join
 	}
 
+	Format struct {
+		Type string // Stores the cli output format
+	}
+
 	Logging struct {
 		Level  string // Stores the configured logging level
 		Output string // Stores the configured logging output