Added BasicAuth support.

parser/schema_parser.go

@@ -1,7 +1,6 @@
 package parser
 
 import (
-	"fmt"
 	"sort"
 	"strings"
 
@@ -205,8 +204,6 @@ func (p *SchemaParser) createSecuritySchemes(
 		result[index] = scheme
 		index++
 	}
-	fmt.Printf("%v", result)
-
 	return result
 }
 

templates/basic_auth.tmpl

@@ -0,0 +1,59 @@
+type Authentication struct {
+	*BasicAuth
+}
+
+type BasicAuth struct {
+	Username string
+	Password string
+}
+
+func EncodeBasicAuth(username string, password string) string {
+	escapedUsername := url.QueryEscape(username)
+	escapedPassword := url.QueryEscape(password)
+	data := base64.StdEncoding.EncodeToString([]byte(
+		fmt.Sprintf("%v:%v", escapedUsername, escapedPassword)))
+	return "Basic " + data
+}
+
+func (a BasicAuth) Encode() string {
+	return EncodeBasicAuth(a.Username, a.Password)
+}
+
+func DecodeAuth(header string) Authentication {
+	authentication := Authentication{}
+	if strings.HasPrefix(header, "Basic ") {
+		basicAuth, err := DecodeBasicAuth(header)
+		if err == nil {
+			authentication.BasicAuth = basicAuth
+		}
+	}
+	return authentication
+}
+
+func DecodeBasicAuth(header string) (*BasicAuth, error) {
+	trimmed := strings.Trim(header, "Basic ")
+	data, err := base64.StdEncoding.DecodeString(trimmed)
+	if err != nil {
+		return nil, err
+	}
+
+	parts := strings.Split(string(data), ":")
+	if len(parts) < 2 {
+		return nil, errors.New("unable to parse")
+	}
+
+	username, err := url.QueryUnescape(parts[0])
+	if err != nil {
+		return nil, err
+	}
+
+	password, err := url.QueryUnescape(parts[1])
+	if err != nil {
+		return nil, err
+	}
+
+	return &BasicAuth{
+		Username: username,
+		Password: password,
+	}, nil
+}

templates/main.tmpl

@@ -1,14 +1,21 @@
 package {{ .Package }}
 
 import (
+	"encoding/base64"
+	"errors"
+	"fmt"
     "net/http"
+	"net/url"
     "strconv"
+	"strings"
 
     "github.com/gin-gonic/gin"
 )
 
 {{ template "util.tmpl" . }}
 
+{{ template "basic_auth.tmpl" . }}
+
 {{ range .Structs }}
     {{ template "component.tmpl" . }}
 {{ end }}

templates/route.tmpl

@@ -34,11 +34,18 @@ func Use{{ .Name }}(router *gin.Engine, service {{ .Name }}Service) {
             return
         }
 
+        {{ if gt (len .Security) 0 }}
+            authentication := DecodeAuth(header.Authorization)
+        {{ else }}
+            authentication := DecodeAuth("")
+        {{ end }}
+
         request := {{ .Name }}Request{
             params,
             query,
             header,
             body,
+            authentication,
         }
 
         response, err := service.{{ .Name }}(ctx, request)

templates/service.tmpl

@@ -25,6 +25,7 @@ type {{ .Name }}Request struct {
     {{ .Name }}Query
     {{ .Name }}Header
     {{ .Name }}Body
+    Authentication
 }
 
 type {{ .Name }}Response struct {

writer/schema_writer.go

@@ -3,10 +3,10 @@ package writer
 import (
 	"bytes"
 	"go/format"
-	"html/template"
 	"io/fs"
 	"os"
 	"path/filepath"
+	"text/template"
 
 	"github.com/cardboardrobots/go-openapi/entity"
 )