gtsocial-umbx

Unnamed repository; edit this file 'description' to name the repository.
Log | Files | Refs | README | LICENSE

sorter.go (2641B)

      1 package yaml
      2 
      3 import (
      4 	"reflect"
      5 	"unicode"
      6 )
      7 
      8 type keyList []reflect.Value
      9 
     10 func (l keyList) Len() int      { return len(l) }
     11 func (l keyList) Swap(i, j int) { l[i], l[j] = l[j], l[i] }
     12 func (l keyList) Less(i, j int) bool {
     13 	a := l[i]
     14 	b := l[j]
     15 	ak := a.Kind()
     16 	bk := b.Kind()
     17 	for (ak == reflect.Interface || ak == reflect.Ptr) && !a.IsNil() {
     18 		a = a.Elem()
     19 		ak = a.Kind()
     20 	}
     21 	for (bk == reflect.Interface || bk == reflect.Ptr) && !b.IsNil() {
     22 		b = b.Elem()
     23 		bk = b.Kind()
     24 	}
     25 	af, aok := keyFloat(a)
     26 	bf, bok := keyFloat(b)
     27 	if aok && bok {
     28 		if af != bf {
     29 			return af < bf
     30 		}
     31 		if ak != bk {
     32 			return ak < bk
     33 		}
     34 		return numLess(a, b)
     35 	}
     36 	if ak != reflect.String || bk != reflect.String {
     37 		return ak < bk
     38 	}
     39 	ar, br := []rune(a.String()), []rune(b.String())
     40 	for i := 0; i < len(ar) && i < len(br); i++ {
     41 		if ar[i] == br[i] {
     42 			continue
     43 		}
     44 		al := unicode.IsLetter(ar[i])
     45 		bl := unicode.IsLetter(br[i])
     46 		if al && bl {
     47 			return ar[i] < br[i]
     48 		}
     49 		if al || bl {
     50 			return bl
     51 		}
     52 		var ai, bi int
     53 		var an, bn int64
     54 		if ar[i] == '0' || br[i] == '0' {
     55 			for j := i-1; j >= 0 && unicode.IsDigit(ar[j]); j-- {
     56 				if ar[j] != '0' {
     57 					an = 1
     58 					bn = 1
     59 					break
     60 				}
     61 			}
     62 		}
     63 		for ai = i; ai < len(ar) && unicode.IsDigit(ar[ai]); ai++ {
     64 			an = an*10 + int64(ar[ai]-'0')
     65 		}
     66 		for bi = i; bi < len(br) && unicode.IsDigit(br[bi]); bi++ {
     67 			bn = bn*10 + int64(br[bi]-'0')
     68 		}
     69 		if an != bn {
     70 			return an < bn
     71 		}
     72 		if ai != bi {
     73 			return ai < bi
     74 		}
     75 		return ar[i] < br[i]
     76 	}
     77 	return len(ar) < len(br)
     78 }
     79 
     80 // keyFloat returns a float value for v if it is a number/bool
     81 // and whether it is a number/bool or not.
     82 func keyFloat(v reflect.Value) (f float64, ok bool) {
     83 	switch v.Kind() {
     84 	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
     85 		return float64(v.Int()), true
     86 	case reflect.Float32, reflect.Float64:
     87 		return v.Float(), true
     88 	case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr:
     89 		return float64(v.Uint()), true
     90 	case reflect.Bool:
     91 		if v.Bool() {
     92 			return 1, true
     93 		}
     94 		return 0, true
     95 	}
     96 	return 0, false
     97 }
     98 
     99 // numLess returns whether a < b.
    100 // a and b must necessarily have the same kind.
    101 func numLess(a, b reflect.Value) bool {
    102 	switch a.Kind() {
    103 	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
    104 		return a.Int() < b.Int()
    105 	case reflect.Float32, reflect.Float64:
    106 		return a.Float() < b.Float()
    107 	case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr:
    108 		return a.Uint() < b.Uint()
    109 	case reflect.Bool:
    110 		return !a.Bool() && b.Bool()
    111 	}
    112 	panic("not a number")
    113 }