package main

import (
	"archive/tar"
	"bytes"
	"context"
	"encoding/json"
	"io/ioutil"
	"log"
	"net/http"
)

const (
	systemLogDir         = "editionslogs"
	systemContainerLabel = "com.docker.editions.system"
)

// SystemContainerCapturer gets the logs from containers which are run
// specifically for Docker Editions.
type SystemContainerCapturer struct{}

// Capture writes output from a CommandCapturer to a tar archive
func (s SystemContainerCapturer) Capture(parentCtx context.Context, w *tar.Writer) {
	ctx, cancel := context.WithTimeout(context.Background(), defaultCaptureTimeout)
	defer cancel()

	errCh := make(chan error)

	go func() {
		resp, err := dockerHTTPGet(ctx, "/containers/json?all=1&label="+systemContainerLabel)
		if err != nil {
			errCh <- err
			return
		}

		defer resp.Body.Close()

		names := []struct {
			ID    string `json:"id"`
			Names []string
		}{}

		if err := json.NewDecoder(resp.Body).Decode(&names); err != nil {
			errCh <- err
			return
		}

		for _, c := range names {
			transport := &UnixSocketRoundTripper{
				Stream: true,
			}

			client := &http.Client{
				Transport: transport,
			}

			resp, err := dockerHTTPGetWithClient(ctx, "/containers/"+c.ID+"/logs?stderr=1&stdout=1&timestamps=1&tail=all", client)
			if err != nil {
				log.Println("ERROR (get request):", err)
				continue
			}

			defer resp.Body.Close()

			// logs makes streaming request where the original http
			// conn is left open so we must clean up after
			// ourselves when we're done reading
			defer transport.Close()

			logLines, err := ioutil.ReadAll(resp.Body)
			if err != nil {
				log.Println("ERROR (reading response):", err)
				continue
			}

			// Docker API returns a Names array where the names are
			// like this: ["/foobar", "/quux"]
			//
			// Use the first one from it.
			//
			// Additionally, the slash from it helps delimit a path
			// separator in the tar archive.
			//
			// TODO(nathanleclaire): This seems fragile, but I'm
			// not sure what approach would be much better.
			tarWrite(w, bytes.NewBuffer(logLines), systemLogDir+c.Names[0])
		}

		errCh <- nil
	}()

	select {
	case <-ctx.Done():
		log.Println("System container log capture context error", ctx.Err())
	case err := <-errCh:
		if err != nil {
			log.Println("System container log capture error", err)
		}
		log.Println("System container log capture finished successfully")
	}
}