litenode/core/LiteNode/methods/extendResponse.js

Lightweight and modular web framework

import { STE } from "../../STE/ste.js"
import { getContentType } from "../../utils/getContentType.js"
import { readFileSync } from "node:fs"
import { extendResponseWithCookies } from "../../utils/cookieManager.js"

export function extendResponse(nativeRes, viewsDir) {
    // Extend response with cookie management methods
    extendResponseWithCookies(nativeRes)

    nativeRes.redirect = (location, statusCode = 302) => {
        nativeRes.writeHead(statusCode, { Location: location })
        nativeRes.end()
    }

    nativeRes.txt = (text) => {
        nativeRes.setHeader("Content-Type", "text/plain")
        nativeRes.end(text)
    }

    nativeRes.json = (data) => {
        nativeRes.setHeader("Content-Type", "application/json")
        nativeRes.end(JSON.stringify(data))
    }

    nativeRes.sendFile = (filePath) => {
        try {
            const fileContents = readFileSync(filePath)
            const contentType = getContentType(filePath)
            nativeRes.setHeader("Content-Type", contentType)
            nativeRes.end(fileContents)
        } catch (err) {
            nativeRes.writeHead(500)
            nativeRes.end("Internal Server Error")
        }
    }

    nativeRes.render = async (template, data) => {
        try {
            const templateEngine = new STE(viewsDir)
            const html = await templateEngine.render(template, data)
            nativeRes.setHeader("Content-Type", "text/html")
            nativeRes.end(html)
        } catch (error) {
            nativeRes.writeHead(500)
            nativeRes.end(`Error rendering template: ${error.message}`)
        }
    }

    nativeRes.status = (code) => {
        nativeRes.statusCode = code
        return nativeRes
    }

    nativeRes.html = (html, statusCode = 200) => {
        try {
            nativeRes.setHeader("Content-Type", "text/html")
            nativeRes.statusCode = statusCode
            nativeRes.end(html)
        } catch (error) {
            console.error("Error sending HTML response:", error)
            nativeRes.statusCode = 500
            nativeRes.setHeader("Content-Type", "text/plain")
            nativeRes.end("Internal Server Error")
        }
    }

    nativeRes.xml = (xmlContent, statusCode = 200) => {
        try {
            nativeRes.setHeader("Content-Type", "application/xml")
            nativeRes.statusCode = statusCode
            nativeRes.end(xmlContent)
        } catch (error) {
            console.error("Error sending XML response:", error)
            nativeRes.statusCode = 500
            nativeRes.setHeader("Content-Type", "text/plain")
            nativeRes.end("Internal Server Error")
        }
    }
}