package shttp

import "fmt"

// Accept sets the Accept header for a request.
func (c *Client) Accept(contentTypes string) {
	c.http.Header.Set("Accept", contentTypes)
}

// CacheControl sets the client-side Cache-Control directive for a request.
func (c *Client) CacheControl(directive string) {
	c.http.Header.Set("Cache-Control", directive)
}

// ContentType sets the Content-Type header on a request.
func (c *Client) ContentType(ct string) {
	c.http.Header.Set("Content-Type", ct)
}

// UserAgent sets the User-Agent header.
func (c *Client) UserAgent(s string) {
	c.http.Header.Set("User-Agent", s)
}

// 设置头部
func (c *Client) Headers(h map[string]string) error {
	if c.http == nil {
		return fmt.Errorf("设置失败")
	}
	for key, v := range h {
		c.http.Header.Set(key, v)
	}
	return nil
}

// 设置头部
func (c *Client) Set(name, value string) {
	c.http.Header.Set(name, value)
}

// 获取返回的数据大小
func (c *Client) GetContentLength() int64 {
	return c.content_length
}

// 获取返回的数据类型
func (c *Client) GetContentType() string {
	return c.content_type
}