package mws

import (
	"net/http"
	"time"

	mdata "github.com/go-gost/gost/v3/pkg/metadata"
)

const (
	defaultPath = "/ws"
)

type metadata struct {
	host string
	path string

	handshakeTimeout  time.Duration
	readHeaderTimeout time.Duration
	readBufferSize    int
	writeBufferSize   int
	enableCompression bool

	muxKeepAliveDisabled bool
	muxKeepAliveInterval time.Duration
	muxKeepAliveTimeout  time.Duration
	muxMaxFrameSize      int
	muxMaxReceiveBuffer  int
	muxMaxStreamBuffer   int

	header    http.Header
	keepAlive time.Duration
}

func (d *mwsDialer) parseMetadata(md mdata.Metadata) (err error) {
	const (
		host = "host"
		path = "path"

		handshakeTimeout  = "handshakeTimeout"
		readHeaderTimeout = "readHeaderTimeout"
		readBufferSize    = "readBufferSize"
		writeBufferSize   = "writeBufferSize"
		enableCompression = "enableCompression"

		header    = "header"
		keepAlive = "keepAlive"

		muxKeepAliveDisabled = "muxKeepAliveDisabled"
		muxKeepAliveInterval = "muxKeepAliveInterval"
		muxKeepAliveTimeout  = "muxKeepAliveTimeout"
		muxMaxFrameSize      = "muxMaxFrameSize"
		muxMaxReceiveBuffer  = "muxMaxReceiveBuffer"
		muxMaxStreamBuffer   = "muxMaxStreamBuffer"
	)

	d.md.host = mdata.GetString(md, host)

	d.md.path = mdata.GetString(md, path)
	if d.md.path == "" {
		d.md.path = defaultPath
	}

	d.md.muxKeepAliveDisabled = mdata.GetBool(md, muxKeepAliveDisabled)
	d.md.muxKeepAliveInterval = mdata.GetDuration(md, muxKeepAliveInterval)
	d.md.muxKeepAliveTimeout = mdata.GetDuration(md, muxKeepAliveTimeout)
	d.md.muxMaxFrameSize = mdata.GetInt(md, muxMaxFrameSize)
	d.md.muxMaxReceiveBuffer = mdata.GetInt(md, muxMaxReceiveBuffer)
	d.md.muxMaxStreamBuffer = mdata.GetInt(md, muxMaxStreamBuffer)

	d.md.handshakeTimeout = mdata.GetDuration(md, handshakeTimeout)
	d.md.readHeaderTimeout = mdata.GetDuration(md, readHeaderTimeout)
	d.md.readBufferSize = mdata.GetInt(md, readBufferSize)
	d.md.writeBufferSize = mdata.GetInt(md, writeBufferSize)
	d.md.enableCompression = mdata.GetBool(md, enableCompression)

	if m := mdata.GetStringMapString(md, header); len(m) > 0 {
		h := http.Header{}
		for k, v := range m {
			h.Add(k, v)
		}
		d.md.header = h
	}
	d.md.keepAlive = mdata.GetDuration(md, keepAlive)

	return
}