forked from coop-cloud/abra
97 lines
2.2 KiB
Go
97 lines
2.2 KiB
Go
package app
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
|
|
"coopcloud.tech/abra/cli/internal"
|
|
"coopcloud.tech/abra/pkg/client"
|
|
"coopcloud.tech/abra/pkg/config"
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/docker/docker/api/types/filters"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/urfave/cli/v2"
|
|
)
|
|
|
|
var appLogsCommand = &cli.Command{
|
|
Name: "logs",
|
|
Aliases: []string{"l"},
|
|
ArgsUsage: "[<service>]",
|
|
Usage: "Tail app logs",
|
|
Flags: []cli.Flag{
|
|
internal.StderrFlag,
|
|
internal.StdoutFlag,
|
|
internal.HealthcheckFlag,
|
|
},
|
|
Action: func(c *cli.Context) error {
|
|
app := internal.ValidateApp(c)
|
|
|
|
if !internal.Stderr && !internal.Stdout && !internal.Healthcheck {
|
|
internal.Stderr = true
|
|
internal.Stdout = true
|
|
internal.Healthcheck = true
|
|
}
|
|
|
|
logrus.Debugf("flags parsed. --stderr: %t, --stdout: %t, --healthcheck: %t", internal.Stderr, internal.Stdout, internal.Healthcheck)
|
|
cl, err := client.New(app.Server)
|
|
if err != nil {
|
|
logrus.Fatal(err)
|
|
}
|
|
|
|
logOpts := types.ContainerLogsOptions{
|
|
Details: false,
|
|
Follow: true,
|
|
ShowStderr: internal.Stderr,
|
|
ShowStdout: internal.Stdout,
|
|
Tail: "20",
|
|
Timestamps: true,
|
|
}
|
|
|
|
serviceName := c.Args().Get(1)
|
|
if serviceName == "" {
|
|
logrus.Debug("tailing logs for all app services")
|
|
internal.StackLogs(c, app.StackName(), logOpts, cl)
|
|
}
|
|
logrus.Debugf("tailing logs for '%s'", serviceName)
|
|
|
|
service := fmt.Sprintf("%s_%s", app.StackName(), serviceName)
|
|
filters := filters.NewArgs()
|
|
filters.Add("name", service)
|
|
serviceOpts := types.ServiceListOptions{Filters: filters}
|
|
services, err := cl.ServiceList(c.Context, serviceOpts)
|
|
if err != nil {
|
|
logrus.Fatal(err)
|
|
}
|
|
if len(services) != 1 {
|
|
logrus.Fatalf("expected 1 service but got %v", len(services))
|
|
}
|
|
|
|
logs, err := cl.ServiceLogs(c.Context, services[0].ID, logOpts)
|
|
if err != nil {
|
|
logrus.Fatal(err)
|
|
}
|
|
// defer after err check as any err returns a nil io.ReadCloser
|
|
defer logs.Close()
|
|
|
|
_, err = io.Copy(os.Stdout, logs)
|
|
if err != nil && err != io.EOF {
|
|
logrus.Fatal(err)
|
|
}
|
|
|
|
return nil
|
|
},
|
|
BashComplete: func(c *cli.Context) {
|
|
appNames, err := config.GetAppNames()
|
|
if err != nil {
|
|
logrus.Warn(err)
|
|
}
|
|
if c.NArg() > 0 {
|
|
return
|
|
}
|
|
for _, a := range appNames {
|
|
fmt.Println(a)
|
|
}
|
|
},
|
|
}
|