mc/common.go

/*
 * Mini Object Storage, (C) 2014,2015 Minio, 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 main

import (
	"bytes"
	"os"
	"path"
	"strings"
	"time"

	"net/http"
	"net/url"

	"github.com/cheggaaa/pb"
	"github.com/codegangsta/cli"
	"github.com/minio-io/mc/pkg/s3"
)

// StartBar -- instantiate a progressbar
func startBar(size int64) *pb.ProgressBar {
	bar := pb.New(int(size))
	bar.SetUnits(pb.U_BYTES)
	bar.SetRefreshRate(time.Millisecond * 10)
	bar.NotPrint = true
	bar.ShowSpeed = true
	bar.Callback = func(s string) {
		// Colorize
		infoCallback(s)
	}
	// Feels like wget
	bar.Format("[=> ]")
	return bar
}

// getBashCompletion -
func getBashCompletion() {
	var b bytes.Buffer
	b.WriteString(mcBashCompletion)
	f := getMcBashCompletionFilename()
	fl, err := os.OpenFile(f, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0600)
	defer fl.Close()
	_, err = fl.Write(b.Bytes())
	if err != nil {
		fatal(err.Error())
	}
	msg := "\nConfiguration written to " + f
	msg = msg + "\n\n$ source ${HOME}/.minio/mc/mc.bash_completion\n"
	msg = msg + "$ echo 'source ${HOME}/.minio/mc/mc.bash_completion' >> ${HOME}/.bashrc\n"
	info(msg)
}

// NewClient - get new client
func getNewClient(c *cli.Context) (*s3.Client, error) {
	var client *s3.Client

	config, err := getMcConfig()
	if err != nil {
		return nil, err
	}

	switch c.GlobalBool("debug") {
	case true:
		trace := s3Trace{BodyTraceFlag: false, RequestTransportFlag: true, Writer: nil}
		traceTransport := s3.GetNewTraceTransport(trace, http.DefaultTransport)
		client = s3.GetNewClient(&config.S3.Auth, traceTransport)
	case false:
		client = s3.GetNewClient(&config.S3.Auth, http.DefaultTransport)
	}

	return client, nil
}

// Parse global options
func parseGlobalOptions(c *cli.Context) {
	switch {
	case c.Bool("get-bash-completion") == true:
		getBashCompletion()
	default:
		cli.ShowAppHelp(c)
	}
}

// Parse subcommand options
func parseArgs(c *cli.Context) (args *cmdArgs, err error) {
	args = new(cmdArgs)
	args.quiet = c.GlobalBool("quiet")

	switch len(c.Args()) {
	case 0:
		return args, nil
	case 1:
		if strings.HasPrefix(c.Args().Get(0), "s3://") {
			uri, err := url.Parse(c.Args().Get(0))
			if err != nil {
				return nil, err
			}
			if uri.Scheme != "s3" {
				return nil, errInvalidScheme
			}
			args.source.bucket = uri.Host
			args.source.key = strings.TrimPrefix(uri.Path, "/")
		} else {
			return nil, errInvalidScheme
		}
	case 2:
		switch true {
		case c.Args().Get(0) != "":
			uri, err := url.Parse(c.Args().Get(0))
			if err != nil {
				return nil, err
			}
			switch true {
			case uri.Scheme == "s3":
				if uri.Host == "" {
					if uri.Path == "" {
						return nil, errInvalidScheme
					}
					return nil, errInvalidScheme
				}
				args.source.bucket = uri.Host
				args.source.key = strings.TrimPrefix(uri.Path, "/")
			case uri.Scheme == "":
				if uri.Host != "" {
					return nil, errInvalidScheme
				}
				if uri.Path != c.Args().Get(0) {
					return nil, errInvalidScheme
				}
				if uri.Path == "." {
					return nil, errFskey
				}
				args.source.bucket = uri.Host
				args.source.key = strings.TrimPrefix(uri.Path, "/")
			case uri.Scheme != "s3":
				return nil, errInvalidScheme
			}
			fallthrough
		case c.Args().Get(1) != "":
			uri, err := url.Parse(c.Args().Get(1))
			if err != nil {
				return nil, err
			}
			switch true {
			case uri.Scheme == "s3":
				if uri.Host == "" {
					if uri.Path == "" {
						return nil, errInvalidScheme
					}
					return nil, errInvalidScheme
				}
				args.destination.bucket = uri.Host
				args.destination.key = strings.TrimPrefix(uri.Path, "/")
			case uri.Scheme == "":
				if uri.Host != "" {
					return nil, errInvalidScheme
				}
				if uri.Path == "." {
					args.destination.key = args.source.key
				} else {
					args.destination.key = strings.TrimPrefix(uri.Path, "/")
				}
				args.destination.bucket = uri.Host
			case uri.Scheme != "s3":
				return nil, errInvalidScheme
			}
		}
	default:
		return nil, errInvalidScheme
	}
	return
}

func getMcBashCompletionFilename() string {
	return path.Join(getMcConfigDir(), "mc.bash_completion")
}