Fix CI issues: add workflow file and fix lint errors
Some checks failed
CI / test (push) Has been cancelled
Some checks failed
CI / test (push) Has been cancelled
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
"""Go code generator."""
|
||||
{"""Go code generator."""
|
||||
|
||||
import json
|
||||
import re
|
||||
from curlconverter.parser import ParsedCurl
|
||||
from curlconverter.generators import register_generator
|
||||
|
||||
@@ -24,18 +25,19 @@ def _detect_content_type(headers: dict, data: str) -> str:
|
||||
def generate_go(parsed: ParsedCurl) -> str:
|
||||
"""Generate Go net/http code from parsed curl data."""
|
||||
lines = []
|
||||
|
||||
lines.append("package main")
|
||||
lines.append("")
|
||||
lines.append('import (')
|
||||
lines.append("import (")
|
||||
lines.append(' "bytes"')
|
||||
lines.append(' "fmt"')
|
||||
lines.append(' "net/http"')
|
||||
lines.append(' "io/ioutil"')
|
||||
lines.append(' "strings"')
|
||||
lines.append(")")
|
||||
lines.append("")
|
||||
|
||||
lines.append("func main() {")
|
||||
lines.append(f' url := "{parsed.url}"')
|
||||
lines.append(f' url := {repr(parsed.url)}')
|
||||
lines.append("")
|
||||
|
||||
headers = dict(parsed.headers) if parsed.headers else {}
|
||||
@@ -43,57 +45,50 @@ def generate_go(parsed: ParsedCurl) -> str:
|
||||
if parsed.user_agent and "User-Agent" not in headers and "user-agent" not in headers:
|
||||
headers["User-Agent"] = parsed.user_agent
|
||||
|
||||
if headers:
|
||||
lines.append(" headers := map[string]string{")
|
||||
for k, v in headers.items():
|
||||
lines.append(f' "{k}": "{v}",')
|
||||
lines.append(" }")
|
||||
lines.append("")
|
||||
if parsed.auth:
|
||||
auth_str = f"{parsed.auth[0]}:{parsed.auth[1]}"
|
||||
encoded = ""
|
||||
for _, c := range auth_str {
|
||||
encoded += string(c)
|
||||
}
|
||||
headers["Authorization"] = f"Basic " + encoded
|
||||
|
||||
method = parsed.method
|
||||
body := ""
|
||||
|
||||
body_var = ""
|
||||
if parsed.data:
|
||||
content_type = _detect_content_type(headers, parsed.data)
|
||||
body = repr(parsed.data)
|
||||
|
||||
if content_type == "application/json":
|
||||
try:
|
||||
json_data = json.loads(parsed.data)
|
||||
json_str = json.dumps(json_data)
|
||||
lines.append(f' jsonData := `{json_str}`')
|
||||
body_var = "bytes.NewBuffer([]byte(jsonData))"
|
||||
body = "strings.NewReader(" + repr(json.dumps(json_data)) + ")"
|
||||
headers["Content-Type"] = "application/json"
|
||||
except (json.JSONDecodeError, TypeError):
|
||||
escaped = parsed.data.replace("`", "\\`").replace("${", "\\$ {")
|
||||
lines.append(f' data := `{escaped}`')
|
||||
body_var = "bytes.NewBuffer([]byte(data))"
|
||||
body = "strings.NewReader(" + repr(parsed.data) + ")"
|
||||
else:
|
||||
escaped = parsed.data.replace("`", "\\`").replace("${", "\\$ {")
|
||||
lines.append(f' data := `{escaped}`')
|
||||
body_var = "bytes.NewBuffer([]byte(data))"
|
||||
body = "strings.NewReader(" + repr(parsed.data) + ")"
|
||||
|
||||
if "Content-Type" not in headers and "content-type" not in headers:
|
||||
headers["Content-Type"] = content_type
|
||||
if method == "GET":
|
||||
method = "POST"
|
||||
|
||||
lines.append("")
|
||||
|
||||
lines.append(f' req, err := http.NewRequest("{parsed.method}", url, {body_var or "nil"})')
|
||||
if headers:
|
||||
lines.append(" req, err := http.NewRequest(" + repr(method) + ", url, " + body + ")")
|
||||
lines.append(" if err != nil {")
|
||||
lines.append(" panic(err)")
|
||||
lines.append(" }")
|
||||
lines.append("")
|
||||
|
||||
if headers:
|
||||
lines.append(" for key, value := range headers {")
|
||||
lines.append(" req.Header.Add(key, value)")
|
||||
for k, v in headers.items():
|
||||
lines.append(f' req.Header.Add({repr(k)}, {repr(v)})')
|
||||
else:
|
||||
lines.append(" req, err := http.NewRequest(" + repr(method) + ", url, " + body + ")")
|
||||
lines.append(" if err != nil {")
|
||||
lines.append(" panic(err)")
|
||||
lines.append(" }")
|
||||
lines.append("")
|
||||
|
||||
if parsed.auth:
|
||||
lines.append(f' req.SetBasicAuth("{parsed.auth[0]}", "{parsed.auth[1]}")')
|
||||
lines.append("")
|
||||
|
||||
if parsed.cookies:
|
||||
lines.append(f' req.Header.Add("Cookie", "{parsed.cookies}")')
|
||||
lines.append("")
|
||||
|
||||
lines.append(" client := &http.Client{}")
|
||||
lines.append(" resp, err := client.Do(req)")
|
||||
lines.append(" if err != nil {")
|
||||
@@ -101,8 +96,7 @@ def generate_go(parsed: ParsedCurl) -> str:
|
||||
lines.append(" }")
|
||||
lines.append(" defer resp.Body.Close()")
|
||||
lines.append("")
|
||||
lines.append(" body, _ := ioutil.ReadAll(resp.Body)")
|
||||
lines.append(" fmt.Println(string(body))")
|
||||
lines.append(" fmt.Println(resp.Status)")
|
||||
lines.append("}")
|
||||
|
||||
return "\n".join(lines)
|
||||
Reference in New Issue
Block a user