/*
Copyright 2013 The Perkeep Authors

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 integration

import (
	"bytes"
	"os"
	"os/exec"
	"path/filepath"
	"runtime"
	"strings"
	"testing"

	"perkeep.org/internal/osutil"
	"perkeep.org/pkg/test"
)

// mkTmpFIFO makes a fifo in a temporary directory and returns the path to it.
func mkTmpFIFO(t *testing.T) (path string) {
	tdir := t.TempDir()

	path = filepath.Join(tdir, "fifo")
	err := osutil.Mkfifo(path, 0660)
	if err != nil {
		t.Fatalf("osutil.mkfifo(): %v", err)
	}

	return
}

// Test that `pk-put' can upload fifos correctly.
func TestCamputFIFO(t *testing.T) {
	if runtime.GOOS == "windows" {
		t.SkipNow()
	}

	fifo := mkTmpFIFO(t)

	// Can we successfully upload a fifo?
	w := test.GetWorld(t)
	out := test.MustRunCmd(t, w.Cmd("pk-put", "file", fifo))

	br := strings.Split(out, "\n")[0]
	out = test.MustRunCmd(t, w.Cmd("pk-get", br))
	t.Logf("Retrieved stored fifo schema: %s", out)
}

// mkTmpSocket makes a socket in a temporary directory and returns the path to
// it.
func mkTmpSocket(t *testing.T) (path string) {
	tdir := t.TempDir()

	path = filepath.Join(tdir, "socket")
	err := osutil.Mksocket(path)
	if err != nil {
		t.Fatalf("osutil.Mksocket(): %v", err)
	}

	return
}

// Test that `pk-put' can upload sockets correctly.
func TestCamputSocket(t *testing.T) {
	if runtime.GOOS == "windows" {
		t.SkipNow()
	}

	socket := mkTmpSocket(t)

	// Can we successfully upload a socket?
	w := test.GetWorld(t)
	out := test.MustRunCmd(t, w.Cmd("pk-put", "file", socket))

	br := strings.Split(out, "\n")[0]
	out = test.MustRunCmd(t, w.Cmd("pk-get", br))
	t.Logf("Retrieved stored socket schema: %s", out)
}

// Test that pk-put twice on the same file only uploads once.
func TestCamputUploadOnce(t *testing.T) {
	w := test.GetWorld(t)

	pkputCmd := func() *exec.Cmd {
		// Use --contents_only because if test is run from devcam,
		// server-config.json is going to be the one from within the fake gopath,
		// hence with a different cTime and with a different blobRef everytime.
		// Also, CAMLI_DEBUG is needed for --contents_only flag.
		return w.CmdWithEnv("pk-put", append(os.Environ(), "CAMLI_DEBUG=1"), "file", "--contents_only=true", filepath.FromSlash("../testdata/server-config.json"))
	}
	wantBlobRef := "sha224-114bc2bd5027736ae9b967668627268262dd659480f125d69ce3c096"
	cmd := pkputCmd()
	out := test.MustRunCmd(t, cmd)
	out = strings.TrimSpace(out)
	if out != wantBlobRef {
		t.Fatalf("wrong pk-put output; wanted %v, got %v", wantBlobRef, out)
	}

	cmd = pkputCmd()
	var stderr bytes.Buffer
	cmd.Stderr = &stderr
	output, err := cmd.Output()
	if err != nil {
		t.Fatalf("second pk-put failed: %v, stdout: %v, stderr: %v", err, output, stderr.String())
	}
	out = strings.TrimSpace(string(output))
	if out != wantBlobRef {
		t.Fatalf("wrong 2nd pk-put output; wanted %v, got %v", wantBlobRef, out)
	}
	wantStats := `[uploadRequests=[blobs=0 bytes=0] uploads=[blobs=0 bytes=0]]`
	if !strings.Contains(stderr.String(), wantStats) {
		t.Fatalf("Wrong stats for 2nd pk-put upload; wanted %v, got %v", wantStats, out)
	}
}