package client

import (
	"bytes"
	"context"
	"fmt"
	"net/http"
)

func (cli *Client) PutManifest(ctx context.Context, rawref string, man []byte, mime string) error {
	var err error

	ref, err := NewReference(rawref)
	if err != nil {
		return err
	}
	uri := ref.Manifest()
        user, pass := ref.Userinfo()

	buffer := bytes.NewBuffer(man)
	req, err := http.NewRequestWithContext(ctx, http.MethodPut, uri, buffer)
	if err != nil {
		return err
	}
	req.Header.Set("User-Agent", cli.userAgent)
	req.Header.Set("Docker-Content-Digest", SHA256Digest(man))
	req.Header.Set("Content-Type", mime)
        if cli.authenticator != nil {
                authHeader, authKey, err := cli.authenticator.MakeHeader(user, pass)
                if err != nil {
                        return err
                }
                req.Header.Set(authHeader, authKey)
        }
	resp, err := cli.httpClient.Do(req)
	if err != nil {
		return err
	}
	resp.Body.Close()
        if resp.StatusCode != http.StatusAccepted {
		err = fmt.Errorf("Manifest not accepted, code %d", resp.StatusCode)
		return err
	}
	loc := resp.Header.Get("Location")
	if loc == "" {
		err := fmt.Errorf("Empty manifest location declaration")
		return err
	}
	return err
}