CSI-ES-2324/Projs/PD2/internal/gateway/forwarder.go

package gateway

import (
	"PD2/internal/protocol"
	"bytes"
	"crypto/tls"
	"encoding/json"
	"fmt"
	"io"
	"net/http"
	"net/url"
)

const baseURL = "https://127.0.0.1:8080"

func getHTTPClient(tlsConfig *tls.Config) *http.Client{
	transport := &http.Transport{TLSClientConfig: tlsConfig}
	return &http.Client{Transport: transport}
}

func forwardStoreUserCert(tlsConfig *tls.Config,uid string,storeUserCertificate protocol.StoreUserCert) (int,[]byte,error) {
    client := getHTTPClient(tlsConfig)

    // Parse the base URL
    parsedURL, err := url.Parse(baseURL)
    if err != nil {
        return 0,nil,fmt.Errorf("error parsing URL: %v", err)
    }
    parsedURL = parsedURL.JoinPath("certs")
    parsedURL = parsedURL.JoinPath(uid)

	jsonData, err := json.Marshal(storeUserCertificate)
	if err != nil {
		return 0,nil,fmt.Errorf("error marshaling JSON: %v", err)
	}

    req, err := http.NewRequest("POST", parsedURL.String(), bytes.NewBuffer(jsonData))
	if err != nil {
		return 0,nil,fmt.Errorf("error creating request: %v", err)
	}

    resp, err := client.Do(req)
    if err != nil {
		return 0,nil,fmt.Errorf("error making request: %v", err)
	}
	defer resp.Body.Close()

    // Read response
	body, err := io.ReadAll(resp.Body)
	if err != nil {
		return 0,nil,fmt.Errorf("error reading response body: %v", err)
	}

    return resp.StatusCode,body,nil
}

func forwardGetMessage(tlsConfig *tls.Config,uid string,num string) (int,[]byte,error) {
    client := getHTTPClient(tlsConfig)

    // Parse the base URL
    parsedURL, err := url.Parse(baseURL)
    if err != nil {
        return 0,nil,fmt.Errorf("error parsing URL: %v", err)
    }
    parsedURL = parsedURL.JoinPath("message")
    parsedURL = parsedURL.JoinPath(uid)
    parsedURL = parsedURL.JoinPath(num)

    
    req, err := http.NewRequest("GET", parsedURL.String(), nil)
	if err != nil {
		return 0,nil,fmt.Errorf("error creating request: %v", err)
	}

    resp, err := client.Do(req)
    if err != nil {
		return 0,nil,fmt.Errorf("error making request: %v", err)
	}
	defer resp.Body.Close()

    // Read response
	body, err := io.ReadAll(resp.Body)
	if err != nil {
		return 0,nil,fmt.Errorf("error reading response body: %v", err)
	}

    return resp.StatusCode,body,nil
}

func forwardGetUnreadMsgsInfo(tlsConfig *tls.Config,uid string,page string,pagesize string) (int,[]byte,error) {
    client := getHTTPClient(tlsConfig)

    // Parse the base URL
    parsedURL, err := url.Parse(baseURL)
    if err != nil {
        return 0,nil,fmt.Errorf("error parsing URL: %v", err)
    }
    parsedURL = parsedURL.JoinPath("queue")
    parsedURL = parsedURL.JoinPath(uid)

    query := parsedURL.Query()
    query.Set("page", page)
    query.Set("pagesize", pagesize)
    parsedURL.RawQuery = query.Encode()

    req, err := http.NewRequest("GET", parsedURL.String(), nil)
	if err != nil {
		return 0,nil,fmt.Errorf("error creating request: %v", err)
	}

    resp, err := client.Do(req)
    if err != nil {
		return 0,nil,fmt.Errorf("error making request: %v", err)
	}
	defer resp.Body.Close()

    // Read response
	body, err := io.ReadAll(resp.Body)
	if err != nil {
		return 0,nil,fmt.Errorf("error reading response body: %v", err)
	}
    
    return resp.StatusCode,body,nil
}

func forwardGetUserCert(tlsConfig *tls.Config,uid string) (int,[]byte,error) {
    client := getHTTPClient(tlsConfig)

    // Parse the base URL
    parsedURL, err := url.Parse(baseURL)
    if err != nil {
        return 0,nil,fmt.Errorf("error parsing URL: %v", err)
    }
    parsedURL = parsedURL.JoinPath("cert")
    parsedURL = parsedURL.JoinPath(uid)
    
    req, err := http.NewRequest("GET", parsedURL.String(), nil)
	if err != nil {
		return 0,nil,fmt.Errorf("error creating request: %v", err)
	}

    resp, err := client.Do(req)
    if err != nil {
		return 0,nil,fmt.Errorf("error making request: %v", err)
	}
	defer resp.Body.Close()

    // Read response
	body, err := io.ReadAll(resp.Body)
	if err != nil {
		return 0,nil,fmt.Errorf("error reading response body: %v", err)
	}

    return resp.StatusCode,body,nil
}

func forwardSendMessage(tlsConfig *tls.Config,uid string,sendMsg protocol.SendMsg) (int,[]byte,error) {
    client := getHTTPClient(tlsConfig)

    // Parse the base URL
    parsedURL, err := url.Parse(baseURL)
    if err != nil {
        return 0,nil,fmt.Errorf("error parsing URL: %v", err)
    }
    parsedURL = parsedURL.JoinPath("message")
    parsedURL = parsedURL.JoinPath(uid)

	jsonData, err := json.Marshal(sendMsg)
	if err != nil {
		return 0,nil,fmt.Errorf("error marshaling JSON: %v", err)
	}

    req, err := http.NewRequest("POST", parsedURL.String(), bytes.NewBuffer(jsonData))
	if err != nil {
		return 0,nil,fmt.Errorf("error creating request: %v", err)
	}

    resp, err := client.Do(req)
    if err != nil {
		return 0,nil,fmt.Errorf("error making request: %v", err)
	}
	defer resp.Body.Close()

    // Read response
	body, err := io.ReadAll(resp.Body)
	if err != nil {
		return 0,nil,fmt.Errorf("error reading response body: %v", err)
	}

    return resp.StatusCode,body,nil
}