// -*- Mode: Go; indent-tabs-mode: t -*-

/*
 * Copyright (C) 2019 Canonical Ltd
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 3 as
 * published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

package client_test

import (
	"encoding/json"
	"errors"
	"io"
	"net/http"
	"os"
	"path/filepath"
	"regexp"
	"strings"

	"golang.org/x/xerrors"
	. "gopkg.in/check.v1"

	"github.com/snapcore/snapd/asserts"
	"github.com/snapcore/snapd/client"
	"github.com/snapcore/snapd/dirs"
)

const happyModelAssertionResponse = `type: model
authority-id: mememe
series: 16
brand-id: mememe
model: test-model
architecture: amd64
base: core18
gadget: pc=18
kernel: pc-kernel=18
required-snaps:
  - core
  - hello-world
timestamp: 2017-07-27T00:00:00.0Z
sign-key-sha3-384: 8B3Wmemeu3H6i4dEV4Q85Q4gIUCHIBCNMHq49e085QeLGHi7v27l3Cqmemer4__t

AcLBcwQAAQoAHRYhBMbX+t6MbKGH5C3nnLZW7+q0g6ELBQJdTdwTAAoJELZW7+q0g6ELEvgQAI3j
jXTqR6kKOqvw94pArwdMDUaZ++tebASAZgso8ejrW2DQGWSc0Q7SQICIR8bvHxqS1GtupQswOzwS
U8hjDTv7WEchH1jylyTj/1W1GernmitTKycecRlEkSOE+EpuqBFgTtj6PdA1Fj3CiCRi1rLMhgF2
luCOitBLaP+E8P3fuATsLqqDLYzt1VY4Y14MU75hMn+CxAQdnOZTI+NzGMasPsldmOYCPNaN/b3N
6/fDLU47RtNlMJ3K0Tz8kj0bqRbegKlD0RdNbAgo9iZwNmrr5E9WCu9f/0rUor/NIxO77H2ExIll
zhmsZ7E6qlxvAgBmzKgAXrn68gGrBkIb0eXKiCaKy/i2ApvjVZ9HkOzA6Ldd+SwNJv/iA8rdiMsq
p2BfKV5f3ju5b6+WktHxAakJ8iqQmj9Yh7piHjsOAUf1PEJd2s2nqQ+pEEn1F0B23gVCY/Fa9YRQ
iKtWVeL3rBw4dSAaK9rpTMqlNcr+yrdXfTK5YzkCC6RU4yzc5MW0hKeseeSiEDSaRYxvftjFfVNa
ZaVXKg8Lu+cHtCJDeYXEkPIDQzXswdBO1M8Mb9D0mYxQwHxwvsWv1DByB+Otq08EYgPh4kyHo7ag
85yK2e/NQ/fxSwQJMhBF74jM1z9arq6RMiE/KOleFAOraKn2hcROKnEeinABW+sOn6vNuMVv
`

// note: this serial assertion was generated by adding print statements to the
// test in api_model_test.go that generate a fake serial assertion
const happySerialAssertionResponse = `type: serial
authority-id: my-brand
brand-id: my-brand
model: my-old-model
serial: serialserial
device-key:
    AcZrBFaFwYABAvCgEOrrLA6FKcreHxCcOoTgBUZ+IRG7Nb8tzmEAklaQPGpv7skapUjwD1luE2go
    mTcoTssVHrfLpBoSDV1aBs44rg3NK40ZKPJP7d2zkds1GxUo1Ea5vfet3SJ4h3aRABEBAAE=
device-key-sha3-384: iqLo9doLzK8De9925UrdUyuvPbBad72OTWVE9YJXqd6nz9dKvwJ_lHP5bVxrl3VO
timestamp: 2019-08-26T16:34:21-05:00
sign-key-sha3-384: anCEGC2NYq7DzDEi6y7OafQCVeVLS90XlLt9PNjrRl9sim5rmRHDDNFNO7ODcWQW

AcJwBAABCgAGBQJdZFBdAADCLALwR6Sy24wm9PffwbvUhOEXneyY3BnxKC0+NgdHu1gU8go9vEP1
i+Flh5uoS70+MBIO+nmF8T+9JWIx2QWFDDxvcuFosnIhvUajCEQohauys5FMz/H/WvB0vrbTBpvK
eg==`

const noModelAssertionYetResponse = `
{
	"type": "error",
	"status-code": 404,
	"status": "Not Found",
	"result": {
	  "message": "no model assertion yet",
	  "kind": "assertion-not-found",
	  "value": "model"
	}
}`

const noSerialAssertionYetResponse = `
{
	"type": "error",
	"status-code": 404,
	"status": "Not Found",
	"result": {
	  "message": "no serial assertion yet",
	  "kind": "assertion-not-found",
	  "value": "serial"
	}
}`

func (cs *clientSuite) TestClientRemodelEndpoint(c *C) {
	cs.cli.Remodel([]byte(`{"new-model": "some-model"}`), client.RemodelOpts{})
	c.Check(cs.req.Method, Equals, "POST")
	c.Check(cs.req.URL.Path, Equals, "/v2/model")
}

func (cs *clientSuite) TestClientRemodel(c *C) {
	cs.status = 202
	cs.rsp = `{
		"type": "async",
		"status-code": 202,
                "result": {},
		"change": "d728"
	}`
	remodelJsonData := []byte(`{"new-model": "some-model"}`)
	id, err := cs.cli.Remodel(remodelJsonData, client.RemodelOpts{})
	c.Assert(err, IsNil)
	c.Check(id, Equals, "d728")
	c.Assert(cs.req.Header.Get("Content-Type"), Equals, "application/json")

	body, err := io.ReadAll(cs.req.Body)
	c.Assert(err, IsNil)
	jsonBody := make(map[string]any)
	err = json.Unmarshal(body, &jsonBody)
	c.Assert(err, IsNil)
	c.Check(jsonBody, HasLen, 1)
	c.Check(jsonBody["new-model"], Equals, string(remodelJsonData))
	c.Check(jsonBody["offline"], IsNil)
}

func (cs *clientSuite) TestClientRemodelOffline(c *C) {
	cs.status = 202
	cs.rsp = `{
        "type": "async",
        "status-code": 202,
                "result": {},
        "change": "d728"
    }`
	remodelJsonData := []byte(`{"new-model": "some-model"}`)
	id, err := cs.cli.Remodel(remodelJsonData, client.RemodelOpts{Offline: true})
	c.Assert(err, IsNil)
	c.Check(id, Equals, "d728")
	c.Assert(cs.req.Header.Get("Content-Type"), Equals, "application/json")

	body, err := io.ReadAll(cs.req.Body)
	c.Assert(err, IsNil)
	jsonBody := make(map[string]any)
	err = json.Unmarshal(body, &jsonBody)
	c.Assert(err, IsNil)
	c.Check(jsonBody, HasLen, 2)
	c.Check(jsonBody["new-model"], Equals, string(remodelJsonData))
	c.Check(jsonBody["offline"], Equals, true)
}

func (cs *clientSuite) TestClientGetModelHappy(c *C) {
	cs.status = 200
	cs.rsp = happyModelAssertionResponse
	modelAssertion, err := cs.cli.CurrentModelAssertion()
	c.Assert(err, IsNil)
	expectedAssert, err := asserts.Decode([]byte(happyModelAssertionResponse))
	c.Assert(err, IsNil)
	c.Assert(modelAssertion, DeepEquals, expectedAssert)
}

func (cs *clientSuite) TestClientGetModelNoModel(c *C) {
	cs.status = 404
	cs.rsp = noModelAssertionYetResponse
	cs.header = http.Header{}
	cs.header.Add("Content-Type", "application/json")
	_, err := cs.cli.CurrentModelAssertion()
	c.Assert(err, ErrorMatches, "no model assertion yet")
}

func (cs *clientSuite) TestClientGetModelNoSerial(c *C) {
	cs.status = 404
	cs.rsp = noSerialAssertionYetResponse
	cs.header = http.Header{}
	cs.header.Add("Content-Type", "application/json")
	_, err := cs.cli.CurrentSerialAssertion()
	c.Assert(err, ErrorMatches, "no serial assertion yet")
}

func (cs *clientSuite) TestClientGetSerialHappy(c *C) {
	cs.status = 200
	cs.rsp = happySerialAssertionResponse
	serialAssertion, err := cs.cli.CurrentSerialAssertion()
	c.Assert(err, IsNil)
	expectedAssert, err := asserts.Decode([]byte(happySerialAssertionResponse))
	c.Assert(err, IsNil)
	c.Assert(serialAssertion, DeepEquals, expectedAssert)
}

func (cs *clientSuite) TestClientCurrentModelAssertionErrIsWrapped(c *C) {
	cs.err = errors.New("boom")
	_, err := cs.cli.CurrentModelAssertion()
	var e xerrors.Wrapper
	c.Assert(err, Implements, &e)
}

func (cs *clientSuite) TestClientOfflineRemodel(c *C) {
	cs.status = 202
	cs.rsp = `{
		"type": "async",
		"status-code": 202,
                "result": {},
		"change": "d728"
	}`
	rawModel := []byte(`some-model`)

	var err error
	snapPaths := []string{filepath.Join(dirs.GlobalRootDir, "snap1.snap")}
	err = os.WriteFile(snapPaths[0], []byte("snap1"), 0644)
	c.Assert(err, IsNil)
	assertsPaths := []string{filepath.Join(dirs.GlobalRootDir, "f1.asserts")}
	err = os.WriteFile(assertsPaths[0], []byte("asserts1"), 0644)
	c.Assert(err, IsNil)

	id, err := cs.cli.RemodelWithLocalSnaps(rawModel, snapPaths, assertsPaths)
	c.Assert(err, IsNil)
	c.Check(id, Equals, "d728")
	contentTypeReStr := "^multipart/form-data; boundary=([A-Za-z0-9]*)$"
	contentType := cs.req.Header.Get("Content-Type")
	c.Assert(contentType, Matches, contentTypeReStr)
	contentTypeRe := regexp.MustCompile(contentTypeReStr)
	matches := contentTypeRe.FindStringSubmatch(contentType)
	c.Assert(len(matches), Equals, 2)
	boundary := "--" + matches[1]

	body, err := io.ReadAll(cs.req.Body)
	c.Assert(err, IsNil)
	expected := boundary + `
Content-Disposition: form-data; name="new-model"
Content-Type: application/x.ubuntu.assertion

some-model
` + boundary + `
Content-Disposition: form-data; name="assertion"
Content-Type: application/x.ubuntu.assertion

asserts1
` + boundary + `
Content-Disposition: form-data; name="snap"; filename="snap1.snap"
Content-Type: application/octet-stream

snap1
` + boundary + `--
`
	expected = strings.Replace(expected, "\n", "\r\n", -1)
	c.Assert(string(body), Equals, expected)
}

func (cs *clientSuite) TestClientOfflineRemodelServerError(c *C) {
	cs.status = 404
	cs.rsp = noSerialAssertionYetResponse
	rawModel := []byte(`some-model`)

	var err error
	snapPaths := []string{filepath.Join(dirs.GlobalRootDir, "snap1.snap")}
	err = os.WriteFile(snapPaths[0], []byte("snap1"), 0644)
	c.Assert(err, IsNil)
	assertsPaths := []string{filepath.Join(dirs.GlobalRootDir, "f1.asserts")}
	err = os.WriteFile(assertsPaths[0], []byte("asserts1"), 0644)
	c.Assert(err, IsNil)

	id, err := cs.cli.RemodelWithLocalSnaps(rawModel, snapPaths, assertsPaths)
	c.Assert(err.Error(), Equals, "no serial assertion yet")
	c.Check(id, Equals, "")
}

func (cs *clientSuite) TestClientOfflineRemodelNoFile(c *C) {
	rawModel := []byte(`some-model`)

	paths := []string{filepath.Join(dirs.GlobalRootDir, "snap1.snap")}

	// No snap file
	id, err := cs.cli.RemodelWithLocalSnaps(rawModel, paths, nil)
	c.Assert(err, ErrorMatches, `cannot open .*: no such file or directory`)
	c.Assert(id, Equals, "")

	// No assertions file
	id, err = cs.cli.RemodelWithLocalSnaps(rawModel, nil, paths)
	c.Assert(err, ErrorMatches, `cannot open .*: no such file or directory`)
	c.Assert(id, Equals, "")
}