core/response.go

// Copyright 2021 Harran Ali <harran.m@gmail.com>. All rights reserved.
// Copyright (c) 2024 Zeni Kim <zenik@smarteching.com>
// Use of this source code is governed by MIT-style
// license that can be found in the LICENSE file.

package core

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

type Response struct {
	headers             []header
	body                []byte
	statusCode          int
	contentType         string
	overrideContentType string
	isTerminated        bool
	redirectTo          string
	HttpResponseWriter  http.ResponseWriter
}

type header struct {
	key string
	val string
}

// TODO add doc
func (rs *Response) Any(body any) *Response {
	if rs.isTerminated == false {
		rs.contentType = CONTENT_TYPE_HTML
		rs.body = []byte(rs.castBasicVarsToString(body))
	}

	return rs
}

// TODO add doc
func (rs *Response) Json(body string) *Response {
	if rs.isTerminated == false {
		rs.contentType = CONTENT_TYPE_JSON
		rs.body = []byte(body)
	}
	return rs
}

// TODO add doc
func (rs *Response) Text(body string) *Response {
	if rs.isTerminated == false {
		rs.contentType = CONTENT_TYPE_TEXT
		rs.body = []byte(body)
	}
	return rs
}

// TODO add doc
func (rs *Response) HTML(body string) *Response {
	if rs.isTerminated == false {
		rs.contentType = CONTENT_TYPE_HTML
		rs.body = []byte(body)
	}
	return rs
}

// TODO add doc
func (rs *Response) Template(name string, data interface{}) *Response {
	var buffer bytes.Buffer
	if rs.isTerminated == false {
		err := tmpl.ExecuteTemplate(&buffer, name, data)
		if err != nil {
			panic(fmt.Sprintf("error executing template: %v", err))
		}
		rs.contentType = CONTENT_TYPE_HTML
		buffer.WriteTo(rs.HttpResponseWriter)
	}
	return rs
}

// TODO add doc
func (rs *Response) SetStatusCode(code int) *Response {
	if rs.isTerminated == false {
		rs.statusCode = code
	}

	return rs
}

// TODO add doc
func (rs *Response) SetContentType(c string) *Response {
	if rs.isTerminated == false {
		rs.overrideContentType = c
	}

	return rs
}

// TODO add doc
func (rs *Response) SetHeader(key string, val string) *Response {
	if rs.isTerminated == false {
		h := header{
			key: key,
			val: val,
		}
		rs.headers = append(rs.headers, h)
	}
	return rs
}

func (rs *Response) ForceSendResponse() {
	rs.isTerminated = true
}

func (rs *Response) Redirect(url string) *Response {
	validator := resolveValidator()
	v := validator.Validate(map[string]interface{}{
		"url": url,
	}, map[string]interface{}{
		"url": "url",
	})
	if v.Failed() {
		if url[0:1] != "/" {
			rs.redirectTo = "/" + url
		} else {
			rs.redirectTo = url
		}
		return rs
	}
	rs.redirectTo = url
	return rs
}

func (rs *Response) castBasicVarsToString(data interface{}) string {
	switch dataType := data.(type) {
	case string:
		return fmt.Sprintf("%v", data)
	case []byte:
		d := data.(string)
		return fmt.Sprintf("%v", d)
	case int:
		intVar, _ := data.(int)
		return fmt.Sprintf("%v", intVar)
	case int8:
		int8Var := data.(int8)
		return fmt.Sprintf("%v", int8Var)
	case int16:
		int16Var := data.(int16)
		return fmt.Sprintf("%v", int16Var)
	case int32:
		int32Var := data.(int32)
		return fmt.Sprintf("%v", int32Var)
	case int64:
		int64Var := data.(int64)
		return fmt.Sprintf("%v", int64Var)
	case uint:
		uintVar, _ := data.(uint)
		return fmt.Sprintf("%v", uintVar)
	case uint8:
		uint8Var := data.(uint8)
		return fmt.Sprintf("%v", uint8Var)
	case uint16:
		uint16Var := data.(uint16)
		return fmt.Sprintf("%v", uint16Var)
	case uint32:
		uint32Var := data.(uint32)
		return fmt.Sprintf("%v", uint32Var)
	case uint64:
		uint64Var := data.(uint64)
		return fmt.Sprintf("%v", uint64Var)
	case float32:
		float32Var := data.(float32)
		return fmt.Sprintf("%v", float32Var)
	case float64:
		float64Var := data.(float64)
		return fmt.Sprintf("%v", float64Var)
	case complex64:
		complex64Var := data.(complex64)
		return fmt.Sprintf("%v", complex64Var)
	case complex128:
		complex128Var := data.(complex128)
		return fmt.Sprintf("%v", complex128Var)
	case bool:
		boolVar := data.(bool)
		return fmt.Sprintf("%v", boolVar)
	default:
		panic(fmt.Sprintf("unsupported response data type %v!", dataType))
	}
}

func (rs *Response) reset() {
	rs.body = nil
	rs.statusCode = http.StatusOK
	rs.contentType = CONTENT_TYPE_HTML
	rs.overrideContentType = ""
	rs.isTerminated = false
	rs.redirectTo = ""
}