/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

package containers

import (
	"fmt"
	"io"
	"os"
	"path/filepath"
	"strings"
	"syscall"

	"github.com/talos-systems/talos/pkg/chunker"
	"github.com/talos-systems/talos/pkg/chunker/file"
	"github.com/talos-systems/talos/pkg/chunker/stream"
)

// Container presents information about a container
type Container struct {
	Inspector Inspector

	Display      string // Friendly Name
	Name         string // container name
	ID           string // container sha/id
	Digest       string // Container Digest
	Image        string
	PodName      string
	Sandbox      string
	Status       string // Running state of container
	RestartCount string
	LogPath      string
	Metrics      *ContainerMetrics
	Pid          uint32
	IsPodSandbox bool // real container or just pod sandbox
}

// ContainerMetrics represents container cgroup stats
type ContainerMetrics struct {
	MemoryUsage uint64
	CPUUsage    uint64
}

// GetProcessStderr returns process stderr
func (c *Container) GetProcessStderr() (string, error) {
	return c.Inspector.GetProcessStderr(c.ID)
}

// GetLogFile returns path to log file, k8s-style
func (c *Container) GetLogFile() string {
	if c.LogPath != "" {
		return c.LogPath
	}

	if c.Sandbox == "" || !strings.Contains(c.Display, ":") {
		return ""
	}

	return filepath.Join(c.Sandbox, c.Name, c.RestartCount+".log")
}

// Kill sends signal to container task
func (c *Container) Kill(signal syscall.Signal) error {
	return c.Inspector.Kill(c.ID, c.IsPodSandbox, signal)
}

// GetLogChunker returns chunker for container log file
func (c *Container) GetLogChunker() (chunker.Chunker, io.Closer, error) {
	logFile := c.GetLogFile()
	if logFile != "" {
		f, err := os.OpenFile(logFile, os.O_RDONLY, 0)
		if err != nil {
			return nil, nil, err
		}

		return file.NewChunker(f), f, nil
	}

	filename, err := c.GetProcessStderr()
	if err != nil {
		return nil, nil, err
	}

	if filename == "" {
		return nil, nil, fmt.Errorf("no log available")
	}

	f, err := os.OpenFile(filename, os.O_RDONLY, 0)
	if err != nil {
		return nil, nil, err
	}

	return stream.NewChunker(f), f, nil
}