logs.go

// Copyright (c) 2016-present Mattermost, Inc. All Rights Reserved.
// See License.txt for license information.

package commands

import (
	"io"
	"os"

	"github.com/mattermost/mattermost-server/mlog/human"
	"github.com/spf13/cobra"
)

var LogsCmd = &cobra.Command{
	Use:   "logs",
	Short: "Display logs in a human-readable format",
	RunE:  logsCmdF,
}

func init() {
	LogsCmd.Flags().Bool("logrus", false, "Use logrus for formatting.")
	RootCmd.AddCommand(LogsCmd)
}

func logsCmdF(command *cobra.Command, args []string) error {
	// check stdin to see if we have a pipe
	fi, err := os.Stdin.Stat()
	if err != nil {
		return err
	}

	var input io.Reader
	if fi.Size() == 0 && fi.Mode()&os.ModeNamedPipe == 0 {
		file, err := os.Open("mattermost.log")
		if err != nil {
			return err
		}
		defer file.Close()
		input = file
	} else {
		input = os.Stdin
	}
	var writer human.LogWriter

	if flag, _ := command.Flags().GetBool("logrus"); flag {
		writer = human.NewLogrusWriter(os.Stdout)
	} else {
		writer = human.NewSimpleWriter(os.Stdout)
	}
	human.ProcessLogs(input, writer)

	return nil
}