mirror of
				https://github.com/superseriousbusiness/gotosocial.git
				synced 2025-11-04 05:02:25 -06:00 
			
		
		
		
	
		
			
				
	
	
		
			354 lines
		
	
	
	
		
			8.2 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			354 lines
		
	
	
	
		
			8.2 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// Copyright 2015 go-swagger maintainers
 | 
						|
//
 | 
						|
// Licensed under the Apache License, Version 2.0 (the "License");
 | 
						|
// you may not use this file except in compliance with the License.
 | 
						|
// You may obtain a copy of the License at
 | 
						|
//
 | 
						|
//    http://www.apache.org/licenses/LICENSE-2.0
 | 
						|
//
 | 
						|
// Unless required by applicable law or agreed to in writing, software
 | 
						|
// distributed under the License is distributed on an "AS IS" BASIS,
 | 
						|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
						|
// See the License for the specific language governing permissions and
 | 
						|
// limitations under the License.
 | 
						|
 | 
						|
package validate
 | 
						|
 | 
						|
import (
 | 
						|
	"encoding/json"
 | 
						|
	"reflect"
 | 
						|
 | 
						|
	"github.com/go-openapi/errors"
 | 
						|
	"github.com/go-openapi/spec"
 | 
						|
	"github.com/go-openapi/strfmt"
 | 
						|
	"github.com/go-openapi/swag"
 | 
						|
)
 | 
						|
 | 
						|
// SchemaValidator validates data against a JSON schema
 | 
						|
type SchemaValidator struct {
 | 
						|
	Path         string
 | 
						|
	in           string
 | 
						|
	Schema       *spec.Schema
 | 
						|
	validators   [8]valueValidator
 | 
						|
	Root         interface{}
 | 
						|
	KnownFormats strfmt.Registry
 | 
						|
	Options      *SchemaValidatorOptions
 | 
						|
}
 | 
						|
 | 
						|
// AgainstSchema validates the specified data against the provided schema, using a registry of supported formats.
 | 
						|
//
 | 
						|
// When no pre-parsed *spec.Schema structure is provided, it uses a JSON schema as default. See example.
 | 
						|
func AgainstSchema(schema *spec.Schema, data interface{}, formats strfmt.Registry, options ...Option) error {
 | 
						|
	res := NewSchemaValidator(schema, nil, "", formats,
 | 
						|
		append(options, WithRecycleValidators(true), withRecycleResults(true))...,
 | 
						|
	).Validate(data)
 | 
						|
	defer func() {
 | 
						|
		pools.poolOfResults.RedeemResult(res)
 | 
						|
	}()
 | 
						|
 | 
						|
	if res.HasErrors() {
 | 
						|
		return errors.CompositeValidationError(res.Errors...)
 | 
						|
	}
 | 
						|
 | 
						|
	return nil
 | 
						|
}
 | 
						|
 | 
						|
// NewSchemaValidator creates a new schema validator.
 | 
						|
//
 | 
						|
// Panics if the provided schema is invalid.
 | 
						|
func NewSchemaValidator(schema *spec.Schema, rootSchema interface{}, root string, formats strfmt.Registry, options ...Option) *SchemaValidator {
 | 
						|
	opts := new(SchemaValidatorOptions)
 | 
						|
	for _, o := range options {
 | 
						|
		o(opts)
 | 
						|
	}
 | 
						|
 | 
						|
	return newSchemaValidator(schema, rootSchema, root, formats, opts)
 | 
						|
}
 | 
						|
 | 
						|
func newSchemaValidator(schema *spec.Schema, rootSchema interface{}, root string, formats strfmt.Registry, opts *SchemaValidatorOptions) *SchemaValidator {
 | 
						|
	if schema == nil {
 | 
						|
		return nil
 | 
						|
	}
 | 
						|
 | 
						|
	if rootSchema == nil {
 | 
						|
		rootSchema = schema
 | 
						|
	}
 | 
						|
 | 
						|
	if schema.ID != "" || schema.Ref.String() != "" || schema.Ref.IsRoot() {
 | 
						|
		err := spec.ExpandSchema(schema, rootSchema, nil)
 | 
						|
		if err != nil {
 | 
						|
			msg := invalidSchemaProvidedMsg(err).Error()
 | 
						|
			panic(msg)
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	if opts == nil {
 | 
						|
		opts = new(SchemaValidatorOptions)
 | 
						|
	}
 | 
						|
 | 
						|
	var s *SchemaValidator
 | 
						|
	if opts.recycleValidators {
 | 
						|
		s = pools.poolOfSchemaValidators.BorrowValidator()
 | 
						|
	} else {
 | 
						|
		s = new(SchemaValidator)
 | 
						|
	}
 | 
						|
 | 
						|
	s.Path = root
 | 
						|
	s.in = "body"
 | 
						|
	s.Schema = schema
 | 
						|
	s.Root = rootSchema
 | 
						|
	s.Options = opts
 | 
						|
	s.KnownFormats = formats
 | 
						|
 | 
						|
	s.validators = [8]valueValidator{
 | 
						|
		s.typeValidator(),
 | 
						|
		s.schemaPropsValidator(),
 | 
						|
		s.stringValidator(),
 | 
						|
		s.formatValidator(),
 | 
						|
		s.numberValidator(),
 | 
						|
		s.sliceValidator(),
 | 
						|
		s.commonValidator(),
 | 
						|
		s.objectValidator(),
 | 
						|
	}
 | 
						|
 | 
						|
	return s
 | 
						|
}
 | 
						|
 | 
						|
// SetPath sets the path for this schema valdiator
 | 
						|
func (s *SchemaValidator) SetPath(path string) {
 | 
						|
	s.Path = path
 | 
						|
}
 | 
						|
 | 
						|
// Applies returns true when this schema validator applies
 | 
						|
func (s *SchemaValidator) Applies(source interface{}, _ reflect.Kind) bool {
 | 
						|
	_, ok := source.(*spec.Schema)
 | 
						|
	return ok
 | 
						|
}
 | 
						|
 | 
						|
// Validate validates the data against the schema
 | 
						|
func (s *SchemaValidator) Validate(data interface{}) *Result {
 | 
						|
	if s == nil {
 | 
						|
		return emptyResult
 | 
						|
	}
 | 
						|
 | 
						|
	if s.Options.recycleValidators {
 | 
						|
		defer func() {
 | 
						|
			s.redeemChildren()
 | 
						|
			s.redeem() // one-time use validator
 | 
						|
		}()
 | 
						|
	}
 | 
						|
 | 
						|
	var result *Result
 | 
						|
	if s.Options.recycleResult {
 | 
						|
		result = pools.poolOfResults.BorrowResult()
 | 
						|
		result.data = data
 | 
						|
	} else {
 | 
						|
		result = &Result{data: data}
 | 
						|
	}
 | 
						|
 | 
						|
	if s.Schema != nil && !s.Options.skipSchemataResult {
 | 
						|
		result.addRootObjectSchemata(s.Schema)
 | 
						|
	}
 | 
						|
 | 
						|
	if data == nil {
 | 
						|
		// early exit with minimal validation
 | 
						|
		result.Merge(s.validators[0].Validate(data)) // type validator
 | 
						|
		result.Merge(s.validators[6].Validate(data)) // common validator
 | 
						|
 | 
						|
		if s.Options.recycleValidators {
 | 
						|
			s.validators[0] = nil
 | 
						|
			s.validators[6] = nil
 | 
						|
		}
 | 
						|
 | 
						|
		return result
 | 
						|
	}
 | 
						|
 | 
						|
	tpe := reflect.TypeOf(data)
 | 
						|
	kind := tpe.Kind()
 | 
						|
	for kind == reflect.Ptr {
 | 
						|
		tpe = tpe.Elem()
 | 
						|
		kind = tpe.Kind()
 | 
						|
	}
 | 
						|
	d := data
 | 
						|
 | 
						|
	if kind == reflect.Struct {
 | 
						|
		// NOTE: since reflect retrieves the true nature of types
 | 
						|
		// this means that all strfmt types passed here (e.g. strfmt.Datetime, etc..)
 | 
						|
		// are converted here to strings, and structs are systematically converted
 | 
						|
		// to map[string]interface{}.
 | 
						|
		d = swag.ToDynamicJSON(data)
 | 
						|
	}
 | 
						|
 | 
						|
	// TODO: this part should be handed over to type validator
 | 
						|
	// Handle special case of json.Number data (number marshalled as string)
 | 
						|
	isnumber := s.Schema.Type.Contains(numberType) || s.Schema.Type.Contains(integerType)
 | 
						|
	if num, ok := data.(json.Number); ok && isnumber {
 | 
						|
		if s.Schema.Type.Contains(integerType) { // avoid lossy conversion
 | 
						|
			in, erri := num.Int64()
 | 
						|
			if erri != nil {
 | 
						|
				result.AddErrors(invalidTypeConversionMsg(s.Path, erri))
 | 
						|
				result.Inc()
 | 
						|
 | 
						|
				return result
 | 
						|
			}
 | 
						|
			d = in
 | 
						|
		} else {
 | 
						|
			nf, errf := num.Float64()
 | 
						|
			if errf != nil {
 | 
						|
				result.AddErrors(invalidTypeConversionMsg(s.Path, errf))
 | 
						|
				result.Inc()
 | 
						|
 | 
						|
				return result
 | 
						|
			}
 | 
						|
			d = nf
 | 
						|
		}
 | 
						|
 | 
						|
		tpe = reflect.TypeOf(d)
 | 
						|
		kind = tpe.Kind()
 | 
						|
	}
 | 
						|
 | 
						|
	for idx, v := range s.validators {
 | 
						|
		if !v.Applies(s.Schema, kind) {
 | 
						|
			if s.Options.recycleValidators {
 | 
						|
				// Validate won't be called, so relinquish this validator
 | 
						|
				if redeemableChildren, ok := v.(interface{ redeemChildren() }); ok {
 | 
						|
					redeemableChildren.redeemChildren()
 | 
						|
				}
 | 
						|
				if redeemable, ok := v.(interface{ redeem() }); ok {
 | 
						|
					redeemable.redeem()
 | 
						|
				}
 | 
						|
				s.validators[idx] = nil // prevents further (unsafe) usage
 | 
						|
			}
 | 
						|
 | 
						|
			continue
 | 
						|
		}
 | 
						|
 | 
						|
		result.Merge(v.Validate(d))
 | 
						|
		if s.Options.recycleValidators {
 | 
						|
			s.validators[idx] = nil // prevents further (unsafe) usage
 | 
						|
		}
 | 
						|
		result.Inc()
 | 
						|
	}
 | 
						|
	result.Inc()
 | 
						|
 | 
						|
	return result
 | 
						|
}
 | 
						|
 | 
						|
func (s *SchemaValidator) typeValidator() valueValidator {
 | 
						|
	return newTypeValidator(
 | 
						|
		s.Path,
 | 
						|
		s.in,
 | 
						|
		s.Schema.Type,
 | 
						|
		s.Schema.Nullable,
 | 
						|
		s.Schema.Format,
 | 
						|
		s.Options,
 | 
						|
	)
 | 
						|
}
 | 
						|
 | 
						|
func (s *SchemaValidator) commonValidator() valueValidator {
 | 
						|
	return newBasicCommonValidator(
 | 
						|
		s.Path,
 | 
						|
		s.in,
 | 
						|
		s.Schema.Default,
 | 
						|
		s.Schema.Enum,
 | 
						|
		s.Options,
 | 
						|
	)
 | 
						|
}
 | 
						|
 | 
						|
func (s *SchemaValidator) sliceValidator() valueValidator {
 | 
						|
	return newSliceValidator(
 | 
						|
		s.Path,
 | 
						|
		s.in,
 | 
						|
		s.Schema.MaxItems,
 | 
						|
		s.Schema.MinItems,
 | 
						|
		s.Schema.UniqueItems,
 | 
						|
		s.Schema.AdditionalItems,
 | 
						|
		s.Schema.Items,
 | 
						|
		s.Root,
 | 
						|
		s.KnownFormats,
 | 
						|
		s.Options,
 | 
						|
	)
 | 
						|
}
 | 
						|
 | 
						|
func (s *SchemaValidator) numberValidator() valueValidator {
 | 
						|
	return newNumberValidator(
 | 
						|
		s.Path,
 | 
						|
		s.in,
 | 
						|
		s.Schema.Default,
 | 
						|
		s.Schema.MultipleOf,
 | 
						|
		s.Schema.Maximum,
 | 
						|
		s.Schema.ExclusiveMaximum,
 | 
						|
		s.Schema.Minimum,
 | 
						|
		s.Schema.ExclusiveMinimum,
 | 
						|
		"",
 | 
						|
		"",
 | 
						|
		s.Options,
 | 
						|
	)
 | 
						|
}
 | 
						|
 | 
						|
func (s *SchemaValidator) stringValidator() valueValidator {
 | 
						|
	return newStringValidator(
 | 
						|
		s.Path,
 | 
						|
		s.in,
 | 
						|
		nil,
 | 
						|
		false,
 | 
						|
		false,
 | 
						|
		s.Schema.MaxLength,
 | 
						|
		s.Schema.MinLength,
 | 
						|
		s.Schema.Pattern,
 | 
						|
		s.Options,
 | 
						|
	)
 | 
						|
}
 | 
						|
 | 
						|
func (s *SchemaValidator) formatValidator() valueValidator {
 | 
						|
	return newFormatValidator(
 | 
						|
		s.Path,
 | 
						|
		s.in,
 | 
						|
		s.Schema.Format,
 | 
						|
		s.KnownFormats,
 | 
						|
		s.Options,
 | 
						|
	)
 | 
						|
}
 | 
						|
 | 
						|
func (s *SchemaValidator) schemaPropsValidator() valueValidator {
 | 
						|
	sch := s.Schema
 | 
						|
	return newSchemaPropsValidator(
 | 
						|
		s.Path, s.in, sch.AllOf, sch.OneOf, sch.AnyOf, sch.Not, sch.Dependencies, s.Root, s.KnownFormats,
 | 
						|
		s.Options,
 | 
						|
	)
 | 
						|
}
 | 
						|
 | 
						|
func (s *SchemaValidator) objectValidator() valueValidator {
 | 
						|
	return newObjectValidator(
 | 
						|
		s.Path,
 | 
						|
		s.in,
 | 
						|
		s.Schema.MaxProperties,
 | 
						|
		s.Schema.MinProperties,
 | 
						|
		s.Schema.Required,
 | 
						|
		s.Schema.Properties,
 | 
						|
		s.Schema.AdditionalProperties,
 | 
						|
		s.Schema.PatternProperties,
 | 
						|
		s.Root,
 | 
						|
		s.KnownFormats,
 | 
						|
		s.Options,
 | 
						|
	)
 | 
						|
}
 | 
						|
 | 
						|
func (s *SchemaValidator) redeem() {
 | 
						|
	pools.poolOfSchemaValidators.RedeemValidator(s)
 | 
						|
}
 | 
						|
 | 
						|
func (s *SchemaValidator) redeemChildren() {
 | 
						|
	for i, validator := range s.validators {
 | 
						|
		if validator == nil {
 | 
						|
			continue
 | 
						|
		}
 | 
						|
		if redeemableChildren, ok := validator.(interface{ redeemChildren() }); ok {
 | 
						|
			redeemableChildren.redeemChildren()
 | 
						|
		}
 | 
						|
		if redeemable, ok := validator.(interface{ redeem() }); ok {
 | 
						|
			redeemable.redeem()
 | 
						|
		}
 | 
						|
		s.validators[i] = nil // free up allocated children if not in pool
 | 
						|
	}
 | 
						|
}
 |