ws/internal/http/upgrade/request/public.go

package request

import (
	"fmt"
	"io"

	"git.xdrm.io/go/ws/internal/http/reader"
	"git.xdrm.io/go/ws/internal/http/upgrade/response"
)

// Parse builds an upgrade HTTP request
// from a reader (typically bufio.NewRead of the socket)
func Parse(r io.Reader) (request *T, err error) {

	req := new(T)
	req.code = 500

	/* (1) Parse request
	---------------------------------------------------------*/
	/* (1) Get chunk reader */
	cr := reader.NewReader(r)
	if err != nil {
		return req, fmt.Errorf("Error while creating chunk reader: %s", err)
	}

	/* (2) Parse header line by line */
	for {

		line, err := cr.Read()
		if err == io.EOF {
			break
		}

		if err != nil {
			return req, err
		}

		err = req.parseHeader(line)

		if err != nil {
			return req, err
		}

	}

	/* (3) Check completion */
	err = req.isComplete()
	if err != nil {
		req.code = response.BadRequest
		return req, err
	}

	req.code = response.SwitchingProtocols
	return req, nil

}

// StatusCode returns the status current
func (r T) StatusCode() response.StatusCode {
	return r.code
}

// BuildResponse builds a response.T from the request
func (r *T) BuildResponse() *response.T {

	inst := new(response.T)

	/* (1) Copy code */
	inst.SetStatusCode(r.code)

	/* (2) Set Protocol */
	if len(r.protocols) > 0 {
		inst.SetProtocol(r.protocols[0])
	}

	/* (4) Process key */
	inst.ProcessKey(r.key)

	return inst
}

// GetURI returns the actual URI
func (r T) GetURI() string {
	return r.request.GetURI()
}