mirror of
https://github.com/go-gitea/gitea.git
synced 2024-12-05 05:07:58 +01:00
ad2642a8aa
* Implementation for calculating language statistics Impement saving code language statistics to database Implement rendering langauge stats Add primary laguage to show in repository list Implement repository stats indexer queue Add indexer test Refactor to use queue module * Do not timeout for queues
217 lines
4.9 KiB
Go
217 lines
4.9 KiB
Go
package substring
|
|
|
|
import (
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/toqueteos/trie"
|
|
)
|
|
|
|
type StringsMatcher interface {
|
|
Match(s string) bool
|
|
MatchIndex(s string) int
|
|
}
|
|
|
|
// regexp
|
|
type regexpString struct{ re *regexp.Regexp }
|
|
|
|
func Regexp(pat string) *regexpString { return ®expString{regexp.MustCompile(pat)} }
|
|
func (m *regexpString) Match(s string) bool { return m.re.MatchString(s) }
|
|
func (m *regexpString) MatchIndex(s string) int {
|
|
found := m.re.FindStringIndex(s)
|
|
if found != nil {
|
|
return found[1]
|
|
}
|
|
return -1
|
|
}
|
|
|
|
// exact
|
|
type exactString struct{ pat string }
|
|
|
|
func Exact(pat string) *exactString { return &exactString{pat} }
|
|
func (m *exactString) Match(s string) bool { return m.pat == s }
|
|
func (m *exactString) MatchIndex(s string) int {
|
|
if m.pat == s {
|
|
return len(s)
|
|
}
|
|
return -1
|
|
}
|
|
|
|
// any, search `s` in `.Match(pat)`
|
|
type anyString struct{ pat string }
|
|
|
|
func Any(pat string) *anyString { return &anyString{pat} }
|
|
func (m *anyString) Match(s string) bool {
|
|
return strings.Index(m.pat, s) >= 0
|
|
}
|
|
func (m *anyString) MatchIndex(s string) int {
|
|
if idx := strings.Index(m.pat, s); idx >= 0 {
|
|
return idx + len(s)
|
|
}
|
|
return -1
|
|
}
|
|
|
|
// has, search `pat` in `.Match(s)`
|
|
type hasString struct{ pat string }
|
|
|
|
func Has(pat string) *hasString { return &hasString{pat} }
|
|
func (m *hasString) Match(s string) bool {
|
|
return strings.Index(s, m.pat) >= 0
|
|
}
|
|
func (m *hasString) MatchIndex(s string) int {
|
|
if idx := strings.Index(s, m.pat); idx >= 0 {
|
|
return idx + len(m.pat)
|
|
}
|
|
return -1
|
|
}
|
|
|
|
// prefix
|
|
type prefixString struct{ pat string }
|
|
|
|
func Prefix(pat string) *prefixString { return &prefixString{pat} }
|
|
func (m *prefixString) Match(s string) bool { return strings.HasPrefix(s, m.pat) }
|
|
func (m *prefixString) MatchIndex(s string) int {
|
|
if strings.HasPrefix(s, m.pat) {
|
|
return len(m.pat)
|
|
}
|
|
return -1
|
|
}
|
|
|
|
// prefixes
|
|
type prefixesString struct{ t *trie.Trie }
|
|
|
|
func Prefixes(pats ...string) *prefixesString {
|
|
t := trie.New()
|
|
for _, pat := range pats {
|
|
t.Insert([]byte(pat))
|
|
}
|
|
return &prefixesString{t}
|
|
}
|
|
func (m *prefixesString) Match(s string) bool { return m.t.PrefixIndex([]byte(s)) >= 0 }
|
|
func (m *prefixesString) MatchIndex(s string) int {
|
|
if idx := m.t.PrefixIndex([]byte(s)); idx >= 0 {
|
|
return idx
|
|
}
|
|
return -1
|
|
}
|
|
|
|
// suffix
|
|
type suffixString struct{ pat string }
|
|
|
|
func Suffix(pat string) *suffixString { return &suffixString{pat} }
|
|
func (m *suffixString) Match(s string) bool { return strings.HasSuffix(s, m.pat) }
|
|
func (m *suffixString) MatchIndex(s string) int {
|
|
if strings.HasSuffix(s, m.pat) {
|
|
return len(m.pat)
|
|
}
|
|
return -1
|
|
}
|
|
|
|
// suffixes
|
|
type suffixesString struct{ t *trie.Trie }
|
|
|
|
func Suffixes(pats ...string) *suffixesString {
|
|
t := trie.New()
|
|
for _, pat := range pats {
|
|
t.Insert(reverse([]byte(pat)))
|
|
}
|
|
return &suffixesString{t}
|
|
}
|
|
func (m *suffixesString) Match(s string) bool {
|
|
return m.t.PrefixIndex(reverse([]byte(s))) >= 0
|
|
}
|
|
func (m *suffixesString) MatchIndex(s string) int {
|
|
if idx := m.t.PrefixIndex(reverse([]byte(s))); idx >= 0 {
|
|
return idx
|
|
}
|
|
return -1
|
|
}
|
|
|
|
// after
|
|
type afterString struct {
|
|
first string
|
|
matcher StringsMatcher
|
|
}
|
|
|
|
func After(first string, m StringsMatcher) *afterString {
|
|
return &afterString{first, m}
|
|
}
|
|
func (a *afterString) Match(s string) bool {
|
|
if idx := strings.Index(s, a.first); idx >= 0 {
|
|
return a.matcher.Match(s[idx+len(a.first):])
|
|
}
|
|
return false
|
|
}
|
|
func (a *afterString) MatchIndex(s string) int {
|
|
if idx := strings.Index(s, a.first); idx >= 0 {
|
|
return idx + a.matcher.MatchIndex(s[idx+len(a.first):])
|
|
}
|
|
return -1
|
|
}
|
|
|
|
// and, returns true iff all matchers return true
|
|
type andString struct{ matchers []StringsMatcher }
|
|
|
|
func And(m ...StringsMatcher) *andString { return &andString{m} }
|
|
func (a *andString) Match(s string) bool {
|
|
for _, m := range a.matchers {
|
|
if !m.Match(s) {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
func (a *andString) MatchIndex(s string) int {
|
|
longest := 0
|
|
for _, m := range a.matchers {
|
|
if idx := m.MatchIndex(s); idx < 0 {
|
|
return -1
|
|
} else if idx > longest {
|
|
longest = idx
|
|
}
|
|
}
|
|
return longest
|
|
}
|
|
|
|
// or, returns true iff any matcher returns true
|
|
type orString struct{ matchers []StringsMatcher }
|
|
|
|
func Or(m ...StringsMatcher) *orString { return &orString{m} }
|
|
func (o *orString) Match(s string) bool {
|
|
for _, m := range o.matchers {
|
|
if m.Match(s) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
func (o *orString) MatchIndex(s string) int {
|
|
for _, m := range o.matchers {
|
|
if idx := m.MatchIndex(s); idx >= 0 {
|
|
return idx
|
|
}
|
|
}
|
|
return -1
|
|
}
|
|
|
|
type suffixGroupString struct {
|
|
suffix StringsMatcher
|
|
matchers []StringsMatcher
|
|
}
|
|
|
|
func SuffixGroup(s string, m ...StringsMatcher) *suffixGroupString {
|
|
return &suffixGroupString{Suffix(s), m}
|
|
}
|
|
func (sg *suffixGroupString) Match(s string) bool {
|
|
if sg.suffix.Match(s) {
|
|
return Or(sg.matchers...).Match(s)
|
|
}
|
|
return false
|
|
}
|
|
func (sg *suffixGroupString) MatchIndex(s string) int {
|
|
if sg.suffix.MatchIndex(s) >= 0 {
|
|
return Or(sg.matchers...).MatchIndex(s)
|
|
}
|
|
return -1
|
|
}
|