Cloudpods/backend/vendor/yunion.io/x/sqlchemy/ascii.go

// Copyright 2019 Yunion
//
// 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 sqlchemy

import (
	"yunion.io/x/pkg/gotypes"
	"yunion.io/x/pkg/util/reflectutils"
)

func (c *STableField) IsAscii() bool {
	return c.spec.IsAscii()
}

func (c *STableField) IsText() bool {
	return c.spec.IsText()
}

func (c *STableField) IsSearchable() bool {
	return c.spec.IsSearchable()
}

func (c *STableField) GetWidth() int {
	return c.spec.GetWidth()
}

func getTableField(f IQueryField) *STableField {
	if gotypes.IsNil(f) {
		return nil
	}
	switch tf := f.(type) {
	case *STableField:
		return tf
	case *sQueryField:
		return getTableField(tf.from.Field(f.Name()))
	case *SSubQueryField:
		return getTableField(tf.field)
	default:
		return nil
	}
}

func IsLongFieldText(f IQueryField) bool {
	tf := getTableField(f)
	if tf != nil {
		return tf.IsText() && tf.IsSearchable() && tf.GetWidth() == 0
	}
	return false
}

func isFieldRequireAscii(f IQueryField) bool {
	tf := getTableField(f)
	if tf != nil {
		return tf.IsAscii()
	}
	return false
}

func isVariableAscii(v interface{}) bool {
	if gotypes.IsNil(v) {
		return true
	}
	switch v.(type) {
	case IQueryField, *SQuery, *SSubQuery:
		return true
	default:
		vals := reflectutils.ExpandInterface(v)
		for _, val := range vals {
			if strVal, ok := val.(string); ok && len(strVal) > 0 {
				if !isPrintableAsciiString(strVal) {
					return false
				}
			} else if strVal, ok := val.(*string); ok && !gotypes.IsNil(strVal) && len(*strVal) > 0 {
				if !isPrintableAsciiString(*strVal) {
					return false
				}
			}
		}
	}
	return true
}

func isPrintableAscii(b byte) bool {
	if b >= 32 && b <= 126 {
		return true
	}
	return false
}

func isPrintableAsciiString(str string) bool {
	for _, b := range []byte(str) {
		if !isPrintableAscii(b) {
			return false
		}
	}
	return true
}