mc/cat-main.go

/*
 * Minio Client, (C) 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 (
	"bufio"
	"errors"
	"fmt"
	"io"
	"io/ioutil"
	"os"
	"strings"
	"syscall"

	"github.com/minio/cli"
	"github.com/minio/minio-xl/pkg/probe"
)

// Display contents of a file.
var catCmd = cli.Command{
	Name:   "cat",
	Usage:  "Display contents of a file.",
	Action: mainCat,
	CustomHelpTemplate: `NAME:
   mc {{.Name}} - {{.Usage}}

USAGE:
   mc {{.Name}} SOURCE [SOURCE...]

EXAMPLES:
   1. Concantenate an object from Amazon S3 cloud storage to mplayer standard input.
      $ mc {{.Name}} https://s3.amazonaws.com/ferenginar/klingon_opera_aktuh_maylotah.ogg | mplayer -

   2. Concantenate contents of file1.txt and stdin to standard output.
      $ mc {{.Name}} file1.txt - > file.txt

   3. Concantenate multiple files from the local filesystem to standard output.
      $ mc {{.Name}} part.* > complete.img

   4. Concatenate a non english file name from Amazon S3 cloud storage.
      $ mc {{.Name}} s3/andoria/本語 > /tmp/本語

   5. Behave like operating system ‘cat’ tool. Useful for alias cat='mc --mimic cat'.
      $ echo "Hello, World!" | mc --mimic {{.Name}}
      Hello, World!

   6. Concantenate an object with space characters from Amazon S3 cloud storage.
      $ mc {{.Name}} 's3/miniocloud/Readme First.txt' | head -1
`,
}

// checkCatSyntax performs command-line input validation for cat command.
func checkCatSyntax(ctx *cli.Context) {
	if (!ctx.Args().Present() && !globalMimicFlag) || ctx.Args().First() == "help" {
		cli.ShowCommandHelpAndExit(ctx, "cat", 1) // last argument is exit code
	}

	for _, arg := range ctx.Args() {
		if strings.HasPrefix(arg, "-") && len(arg) > 1 {
			fatalIf(probe.NewError(errors.New("")), fmt.Sprintf("Unknown flag ‘%s’ passed.", arg))
		}
	}
}

// catURL displays contents of a URL to stdout.
func catURL(sourceURL string) *probe.Error {
	config := mustGetMcConfig()
	URL := getAliasURL(sourceURL, config.Aliases)

	var reader io.ReadCloser
	switch URL {
	case "-":
		reader = ioutil.NopCloser(bufio.NewReader(os.Stdin))
	default:
		sourceClnt, err := url2Client(URL)
		if err != nil {
			return err.Trace(URL)
		}
		// Ignore size, since os.Stat() would not return proper size all the time for local filesystem for example /proc files.
		reader, _, err = sourceClnt.GetObject(0, 0)
		if err != nil {
			return err.Trace(URL)
		}
	}
	defer reader.Close()

	return catOut(reader).Trace(URL)
}

// catOut reads from reader stream and writes to stdout.
func catOut(r io.Reader) *probe.Error {
	// Do not forget to flush after stdout.
	//	out := bufio.NewWriter(os.Stdout)
	//	defer out.Flush()

	// Read till EOF.
	if _, err := io.Copy(os.Stdout, r); err != nil {
		switch e := err.(type) {
		case *os.PathError:
			if e.Err == syscall.EPIPE {
				// stdout closed by the user. Gracefully exit.
				return nil
			}
			return probe.NewError(err)
		default:
			return probe.NewError(err)
		}
	}
	return nil
}

// mainCat is the main entry point for cat command.
func mainCat(ctx *cli.Context) {
	checkCatSyntax(ctx)

	stdinMode := false
	if globalMimicFlag && !ctx.Args().Present() {
		stdinMode = true
	}

	// handle std input data
	if stdinMode {
		fatalIf(catOut(os.Stdin).Trace(), "Unable to read from standard input.")
		return
	}

	// if Args contain ‘-’, we need to preserve its order specially.
	args := []string(ctx.Args())
	if ctx.Args().First() == "-" {
		for i, arg := range os.Args {
			if arg == "cat" {
				// Overwrite ctx.Args with os.Args.
				args = os.Args[i+1:]
				break
			}
		}
	}
	// Convert arguments to URLs: expand alias, fix format...
	for _, arg := range args {
		fatalIf(catURL(arg).Trace(arg), "Unable to read from ‘"+arg+"’.")
	}
}