Method Transform

transforms.go:157–236

Transform is called for every response to add the `$schema` field and/or the Link header pointing to the JSON Schema.

(ctx Context, _ string, v any)

// Transform is called for every response to add the `$schema` field and/or
// the Link header pointing to the JSON Schema.
func (t *SchemaLinkTransformer) Transform(ctx Context, _ string, v any) (any, error) {
	vv := reflect.ValueOf(v)
	if vv.Kind() == reflect.Pointer && vv.IsNil() {
		return v, nil
	}

	typ := deref(reflect.TypeOf(v))

	if typ.Kind() != reflect.Struct {
		return v, nil
	}

	infos := t.types[typ]
	if len(infos) == 0 {
		return v, nil
	}

	info := infos[0]
	if len(infos) > 1 {
		// More than one schema link for this type, try to find the one that
		// matches the current operation.
		op := ctx.Operation()
		if op != nil {
			opID := op.OperationID
			for _, i := range infos {
				if i.opID == opID {
					info = i
					break
				}
			}
		}
	}

	ctx.AppendHeader("Link", info.header)
	tmp := reflect.New(info.t).Elem()

	// Set the `$schema` field.
	schemaURL := getSchemaHost(ctx)

	if schemaURL == "" || strings.HasPrefix(schemaURL, "localhost") {
		if t.oapi != nil && len(t.oapi.Servers) > 0 && t.oapi.Servers[0].URL != "" {
			serverURL := t.oapi.Servers[0].URL
			if !strings.HasPrefix(serverURL, "/") {
				schemaURL = serverURL
			}
		}
		if schemaURL == "" {
			schemaURL = "localhost"
		}
	}

	buf := bufPool.Get().(*bytes.Buffer)
	if !strings.Contains(schemaURL, "://") {
		if ctx.TLS() == nil && (strings.HasPrefix(schemaURL, "localhost") || strings.HasPrefix(schemaURL, "127.0.0.1")) {
			buf.WriteString("http://")
		} else {
			buf.WriteString("https://")
		}