146 lines
4.5 KiB
Go
146 lines
4.5 KiB
Go
package swaggerset
|
||
|
||
import (
|
||
"fmt"
|
||
"reflect"
|
||
"strings"
|
||
|
||
"github.com/getkin/kin-openapi/openapi3"
|
||
dynamicstruct "github.com/ompluscator/dynamic-struct"
|
||
)
|
||
|
||
type Message struct {
|
||
Type string
|
||
Request httpRequest
|
||
Response interface{}
|
||
}
|
||
|
||
type httpRequest struct {
|
||
Header interface{}
|
||
Body interface{}
|
||
}
|
||
|
||
func handleOperation(method string, operation *openapi3.Operation) (reqParam, reqBody, rsp interface{}) {
|
||
fmt.Printf(" Method: %s\n", method)
|
||
|
||
// Обработка параметров (GET)
|
||
if len(operation.Parameters) > 0 {
|
||
paramsStruct := dynamicstruct.NewStruct()
|
||
for _, paramRef := range operation.Parameters {
|
||
param := paramRef.Value
|
||
fieldName := capitalize(param.Name)
|
||
jsonName := strings.ToLower(param.Name[:1]) + param.Name[1:]
|
||
goType := getGoType(param.Schema)
|
||
|
||
// В зависимости от того, где параметр находится (header, query, path, etc.), добавляем соответствующий тег
|
||
switch param.In {
|
||
case "query":
|
||
paramsStruct = paramsStruct.AddField(fieldName, goType, fmt.Sprintf(`json:"%s" query:"%s"`, jsonName, jsonName))
|
||
case "header":
|
||
paramsStruct = paramsStruct.AddField(fieldName, goType, fmt.Sprintf(`json:"%s" header:"%s"`, jsonName, jsonName))
|
||
default:
|
||
paramsStruct = paramsStruct.AddField(fieldName, goType, fmt.Sprintf(`json:"%s"`, jsonName))
|
||
}
|
||
}
|
||
// Получили структуру запроса для методов, где есть параметры в header, query, path, etc., добавили теги
|
||
reqParam = paramsStruct.Build().New()
|
||
}
|
||
|
||
// Обработка тела запроса (POST)
|
||
if operation.RequestBody != nil {
|
||
for _, content := range operation.RequestBody.Value.Content {
|
||
bodyFields := buildDynamicStruct(content.Schema)
|
||
bodyStruct := reflect.StructOf(bodyFields)
|
||
bodyInstance := reflect.New(bodyStruct).Interface()
|
||
// Получили тело запроса
|
||
reqBody = bodyInstance
|
||
}
|
||
}
|
||
|
||
// Обработка ответов
|
||
for status, response := range operation.Responses.Map() {
|
||
fmt.Printf(" Response Code: %s\n", status)
|
||
for _, content := range response.Value.Content {
|
||
responseFields := buildDynamicStruct(content.Schema)
|
||
responseStruct := reflect.StructOf(responseFields)
|
||
responseInstance := reflect.New(responseStruct).Interface()
|
||
// Получили структуру ответа
|
||
rsp = responseInstance
|
||
}
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
// Рекурсивное создание структуры из схемы с учетом $ref
|
||
func buildDynamicStruct(schema *openapi3.SchemaRef) []reflect.StructField {
|
||
var sfields []reflect.StructField
|
||
|
||
if len(schema.Value.Properties) == 0 {
|
||
return sfields
|
||
}
|
||
|
||
for name, prop := range schema.Value.Properties {
|
||
fieldName := capitalize(name)
|
||
if prop.Ref != "" || prop.Value.Type.Is("object") {
|
||
subBuilder := buildDynamicStruct(prop)
|
||
|
||
sfield := reflect.StructField{
|
||
Name: fieldName,
|
||
Type: reflect.StructOf(subBuilder),
|
||
Tag: reflect.StructTag(fmt.Sprintf(`json:"%s"`, name)),
|
||
}
|
||
|
||
sfields = append(sfields, sfield)
|
||
} else {
|
||
sfield := reflect.StructField{
|
||
Name: fieldName,
|
||
Type: reflect.TypeOf(getGoType(prop)),
|
||
Tag: reflect.StructTag(fmt.Sprintf(`json:"%s"`, name)),
|
||
}
|
||
|
||
sfields = append(sfields, sfield)
|
||
}
|
||
}
|
||
|
||
return sfields
|
||
}
|
||
|
||
// Преобразование типа OpenAPI в тип Go
|
||
func getGoType(schema *openapi3.SchemaRef) interface{} {
|
||
switch {
|
||
case schema.Value.Type.Is("string"):
|
||
return ""
|
||
case schema.Value.Type.Is("integer"):
|
||
return 0
|
||
case schema.Value.Type.Is("boolean"):
|
||
return false
|
||
case schema.Value.Type.Is("array"):
|
||
return []interface{}{}
|
||
case schema.Value.Type.Is("object"):
|
||
return buildDynamicStruct(schema)
|
||
default:
|
||
return nil
|
||
}
|
||
}
|
||
|
||
func capitalize(fieldName string) string {
|
||
// Заменяем точки на подчеркивания для унификации
|
||
fieldName = strings.ReplaceAll(fieldName, ".", "_")
|
||
|
||
// Разделяем строку по подчеркиваниям
|
||
parts := strings.Split(fieldName, "_")
|
||
if len(parts) == 1 {
|
||
return strings.ToUpper(fieldName[:1]) + fieldName[1:]
|
||
}
|
||
|
||
// Обрабатываем каждый фрагмент
|
||
for i := 0; i < len(parts); i++ {
|
||
// Капитализируем первые буквы всех частей, кроме первой
|
||
parts[i] = strings.Title(parts[i]) //cases.Title(language.English).String(parts[i])
|
||
}
|
||
|
||
// Собираем строку обратно, соединяя части без подчеркиваний
|
||
return strings.Join(parts, "")
|
||
}
|