httppb/post.go

107 lines
2.5 KiB
Go

// Copyright 1994-2025 WIT.COM Inc Licensed GPL 3.0
package httppb
import (
"bytes"
"io/ioutil"
"net/http"
"net/url"
"os"
"os/user"
"go.wit.com/log"
)
// standard protobuf POST
// returns *HttpRequst protobuf
func DoPost(baseURL string, route string, data []byte) (*HttpRequest, error) {
// if you ever have 'http://www.wit.com//' GO will regect the server recieving it.
// Even though the linux kernel gets the network payload
// also it never gives you an error about that, it just goes away invisably inside GO
tmpURL, _ := url.Parse(baseURL) // "http://forge.grid.wit.com:2520")
finalURL := tmpURL.JoinPath(route) // Correctly produces ...:2520/patches
var err error
var req *http.Request
log.Info("httppb.HttpPost to", finalURL.String())
req, err = http.NewRequest(http.MethodPost, finalURL.String(), bytes.NewBuffer(data))
if req == nil {
return nil, err
}
username := os.Getenv("GIT_AUTHOR_NAME")
if username == "" {
usr, _ := user.Current()
username = usr.Username
}
req.Header.Set("author", username)
hostname, _ := os.Hostname()
req.Header.Set("hostname", hostname)
client := &http.Client{}
resp, err := client.Do(req)
if err != nil {
return nil, err
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return nil, err
}
reqPB, err := ReqToPB(req)
reqPB.URL = finalURL.String()
reqPB.ClientData = data
reqPB.ClientDataLen = int64(len(data))
reqPB.ServerData = body
reqPB.ServerDataLen = int64(len(body))
return reqPB, nil
}
func HttpPost(baseURL string, route string, data []byte) ([]byte, error) {
// Fix using url.JoinPath (Best Practice)
tmpURL, _ := url.Parse(baseURL) // "http://forge.grid.wit.com:2520")
finalURL := tmpURL.JoinPath(route) // Correctly produces ...:2520/patches
var err error
var req *http.Request
log.Info("httppb.HttpPost to", finalURL.String())
req, err = http.NewRequest(http.MethodPost, finalURL.String(), bytes.NewBuffer(data))
if req == nil {
return nil, err
}
username := os.Getenv("GIT_AUTHOR_NAME")
if username == "" {
usr, _ := user.Current()
username = usr.Username
}
req.Header.Set("author", username)
hostname, _ := os.Hostname()
req.Header.Set("hostname", hostname)
return PostReq(req)
}
// Posts a reqest and returns the bytes returned
func PostReq(req *http.Request) ([]byte, error) {
client := &http.Client{}
resp, err := client.Do(req)
if err != nil {
return []byte("client.Do(req) error"), err
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return body, err
}
return body, nil
}