2017-08-16 14:37:37 -07:00
|
|
|
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
2016-04-10 14:39:38 -07:00
|
|
|
// See License.txt for license information.
|
|
|
|
|
|
|
|
package model
|
|
|
|
|
|
|
|
import (
|
2016-06-23 11:28:05 -07:00
|
|
|
"regexp"
|
2016-04-10 14:39:38 -07:00
|
|
|
"strings"
|
2018-11-18 09:55:05 -08:00
|
|
|
"time"
|
2016-04-10 14:39:38 -07:00
|
|
|
)
|
|
|
|
|
2016-06-23 11:28:05 -07:00
|
|
|
var searchTermPuncStart = regexp.MustCompile(`^[^\pL\d\s#"]+`)
|
|
|
|
var searchTermPuncEnd = regexp.MustCompile(`[^\pL\d\s*"]+$`)
|
|
|
|
|
2016-04-10 14:39:38 -07:00
|
|
|
type SearchParams struct {
|
2018-11-18 09:55:05 -08:00
|
|
|
Terms string
|
|
|
|
IsHashtag bool
|
|
|
|
InChannels []string
|
|
|
|
FromUsers []string
|
|
|
|
AfterDate string
|
|
|
|
BeforeDate string
|
|
|
|
OnDate string
|
|
|
|
OrTerms bool
|
|
|
|
IncludeDeletedChannels bool
|
|
|
|
TimeZoneOffset int
|
2016-04-10 14:39:38 -07:00
|
|
|
}
|
|
|
|
|
2018-11-18 09:55:05 -08:00
|
|
|
// Returns the epoch timestamp of the start of the day specified by SearchParams.AfterDate
|
|
|
|
func (p *SearchParams) GetAfterDateMillis() int64 {
|
|
|
|
date, err := time.Parse("2006-01-02", PadDateStringZeros(p.AfterDate))
|
2017-08-16 14:37:37 -07:00
|
|
|
if err != nil {
|
2018-11-18 09:55:05 -08:00
|
|
|
date = time.Now()
|
2017-08-16 14:37:37 -07:00
|
|
|
}
|
2018-11-18 09:55:05 -08:00
|
|
|
|
|
|
|
// travel forward 1 day
|
|
|
|
oneDay := time.Hour * 24
|
|
|
|
afterDate := date.Add(oneDay)
|
|
|
|
return GetStartOfDayMillis(afterDate, p.TimeZoneOffset)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Returns the epoch timestamp of the end of the day specified by SearchParams.BeforeDate
|
|
|
|
func (p *SearchParams) GetBeforeDateMillis() int64 {
|
|
|
|
date, err := time.Parse("2006-01-02", PadDateStringZeros(p.BeforeDate))
|
|
|
|
if err != nil {
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
|
|
|
// travel back 1 day
|
|
|
|
oneDay := time.Hour * -24
|
|
|
|
beforeDate := date.Add(oneDay)
|
|
|
|
return GetEndOfDayMillis(beforeDate, p.TimeZoneOffset)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Returns the epoch timestamps of the start and end of the day specified by SearchParams.OnDate
|
|
|
|
func (p *SearchParams) GetOnDateMillis() (int64, int64) {
|
|
|
|
date, err := time.Parse("2006-01-02", PadDateStringZeros(p.OnDate))
|
|
|
|
if err != nil {
|
|
|
|
return 0, 0
|
|
|
|
}
|
|
|
|
|
|
|
|
return GetStartOfDayMillis(date, p.TimeZoneOffset), GetEndOfDayMillis(date, p.TimeZoneOffset)
|
2017-08-16 14:37:37 -07:00
|
|
|
}
|
|
|
|
|
2018-11-18 09:55:05 -08:00
|
|
|
var searchFlags = [...]string{"from", "channel", "in", "before", "after", "on"}
|
2016-04-10 14:39:38 -07:00
|
|
|
|
|
|
|
func splitWords(text string) []string {
|
|
|
|
words := []string{}
|
|
|
|
|
|
|
|
foundQuote := false
|
|
|
|
location := 0
|
|
|
|
for i, char := range text {
|
|
|
|
if char == '"' {
|
|
|
|
if foundQuote {
|
|
|
|
// Grab the quoted section
|
|
|
|
word := text[location : i+1]
|
|
|
|
words = append(words, word)
|
|
|
|
foundQuote = false
|
|
|
|
location = i + 1
|
|
|
|
} else {
|
2018-02-08 15:11:04 -08:00
|
|
|
words = append(words, strings.Fields(text[location:i])...)
|
2016-04-10 14:39:38 -07:00
|
|
|
foundQuote = true
|
|
|
|
location = i
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-02-08 15:11:04 -08:00
|
|
|
words = append(words, strings.Fields(text[location:])...)
|
2016-04-10 14:39:38 -07:00
|
|
|
|
|
|
|
return words
|
|
|
|
}
|
|
|
|
|
|
|
|
func parseSearchFlags(input []string) ([]string, [][2]string) {
|
|
|
|
words := []string{}
|
|
|
|
flags := [][2]string{}
|
|
|
|
|
|
|
|
skipNextWord := false
|
|
|
|
for i, word := range input {
|
|
|
|
if skipNextWord {
|
|
|
|
skipNextWord = false
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
|
|
|
isFlag := false
|
|
|
|
|
|
|
|
if colon := strings.Index(word, ":"); colon != -1 {
|
|
|
|
flag := word[:colon]
|
|
|
|
value := word[colon+1:]
|
|
|
|
|
|
|
|
for _, searchFlag := range searchFlags {
|
|
|
|
// check for case insensitive equality
|
|
|
|
if strings.EqualFold(flag, searchFlag) {
|
|
|
|
if value != "" {
|
|
|
|
flags = append(flags, [2]string{searchFlag, value})
|
|
|
|
isFlag = true
|
|
|
|
} else if i < len(input)-1 {
|
|
|
|
flags = append(flags, [2]string{searchFlag, input[i+1]})
|
|
|
|
skipNextWord = true
|
|
|
|
isFlag = true
|
|
|
|
}
|
|
|
|
|
|
|
|
if isFlag {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if !isFlag {
|
|
|
|
// trim off surrounding punctuation (note that we leave trailing asterisks to allow wildcards)
|
2016-06-23 11:28:05 -07:00
|
|
|
word = searchTermPuncStart.ReplaceAllString(word, "")
|
|
|
|
word = searchTermPuncEnd.ReplaceAllString(word, "")
|
2016-04-10 14:39:38 -07:00
|
|
|
|
|
|
|
// and remove extra pound #s
|
|
|
|
word = hashtagStart.ReplaceAllString(word, "#")
|
|
|
|
|
|
|
|
if len(word) != 0 {
|
|
|
|
words = append(words, word)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return words, flags
|
|
|
|
}
|
|
|
|
|
2018-11-18 09:55:05 -08:00
|
|
|
func ParseSearchParams(text string, timeZoneOffset int) []*SearchParams {
|
2016-04-10 14:39:38 -07:00
|
|
|
words, flags := parseSearchFlags(splitWords(text))
|
|
|
|
|
|
|
|
hashtagTermList := []string{}
|
|
|
|
plainTermList := []string{}
|
|
|
|
|
|
|
|
for _, word := range words {
|
|
|
|
if validHashtag.MatchString(word) {
|
|
|
|
hashtagTermList = append(hashtagTermList, word)
|
|
|
|
} else {
|
|
|
|
plainTermList = append(plainTermList, word)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
hashtagTerms := strings.Join(hashtagTermList, " ")
|
|
|
|
plainTerms := strings.Join(plainTermList, " ")
|
|
|
|
|
|
|
|
inChannels := []string{}
|
|
|
|
fromUsers := []string{}
|
2018-11-18 09:55:05 -08:00
|
|
|
afterDate := ""
|
|
|
|
beforeDate := ""
|
|
|
|
onDate := ""
|
2016-04-10 14:39:38 -07:00
|
|
|
|
|
|
|
for _, flagPair := range flags {
|
|
|
|
flag := flagPair[0]
|
|
|
|
value := flagPair[1]
|
|
|
|
|
|
|
|
if flag == "in" || flag == "channel" {
|
|
|
|
inChannels = append(inChannels, value)
|
|
|
|
} else if flag == "from" {
|
|
|
|
fromUsers = append(fromUsers, value)
|
2018-11-18 09:55:05 -08:00
|
|
|
} else if flag == "after" {
|
|
|
|
afterDate = value
|
|
|
|
} else if flag == "before" {
|
|
|
|
beforeDate = value
|
|
|
|
} else if flag == "on" {
|
|
|
|
onDate = value
|
2016-04-10 14:39:38 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
paramsList := []*SearchParams{}
|
|
|
|
|
|
|
|
if len(plainTerms) > 0 {
|
|
|
|
paramsList = append(paramsList, &SearchParams{
|
2018-11-18 09:55:05 -08:00
|
|
|
Terms: plainTerms,
|
|
|
|
IsHashtag: false,
|
|
|
|
InChannels: inChannels,
|
|
|
|
FromUsers: fromUsers,
|
|
|
|
AfterDate: afterDate,
|
|
|
|
BeforeDate: beforeDate,
|
|
|
|
OnDate: onDate,
|
|
|
|
TimeZoneOffset: timeZoneOffset,
|
2016-04-10 14:39:38 -07:00
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(hashtagTerms) > 0 {
|
|
|
|
paramsList = append(paramsList, &SearchParams{
|
2018-11-18 09:55:05 -08:00
|
|
|
Terms: hashtagTerms,
|
|
|
|
IsHashtag: true,
|
|
|
|
InChannels: inChannels,
|
|
|
|
FromUsers: fromUsers,
|
|
|
|
AfterDate: afterDate,
|
|
|
|
BeforeDate: beforeDate,
|
|
|
|
OnDate: onDate,
|
|
|
|
TimeZoneOffset: timeZoneOffset,
|
2016-04-10 14:39:38 -07:00
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
// special case for when no terms are specified but we still have a filter
|
2018-11-18 09:55:05 -08:00
|
|
|
if len(plainTerms) == 0 && len(hashtagTerms) == 0 && (len(inChannels) != 0 || len(fromUsers) != 0 || len(afterDate) != 0 || len(beforeDate) != 0 || len(onDate) != 0) {
|
2016-04-10 14:39:38 -07:00
|
|
|
paramsList = append(paramsList, &SearchParams{
|
2018-11-18 09:55:05 -08:00
|
|
|
Terms: "",
|
|
|
|
IsHashtag: false,
|
|
|
|
InChannels: inChannels,
|
|
|
|
FromUsers: fromUsers,
|
|
|
|
AfterDate: afterDate,
|
|
|
|
BeforeDate: beforeDate,
|
|
|
|
OnDate: onDate,
|
|
|
|
TimeZoneOffset: timeZoneOffset,
|
2016-04-10 14:39:38 -07:00
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
return paramsList
|
|
|
|
}
|