package registry // import "github.com/gbmor/getwtxt/registry" import ( "sort" "strings" ) // QueryUser checks the user index for nicknames that contain the // nickname provided as an argument. Entries are returned sorted // by the date they were added to the index. func (index UserIndex) QueryUser(name string) []string { timekey := NewTimeMap() keys := make(TimeSlice, 0) var users []string imutex.RLock() for k, v := range index { if strings.Contains(v.Nick, name) { timekey[v.Date] = v.Nick + "\t" + k + "\t" + string(v.APIdate) keys = append(keys, v.Date) } } imutex.RUnlock() sort.Sort(keys) for _, e := range keys { users = append(users, timekey[e]) } return users } // QueryTag returns all the known statuses that // contain the provided tag. func (index UserIndex) QueryTag(tag string) []string { statusmap := NewTimeMapSlice() i := 0 imutex.RLock() for _, v := range index { statusmap[i] = v.FindTag(tag) i++ } imutex.RUnlock() return statusmap.SortByTime() } // FindTag takes a user's tweets and looks for a given tag. // Returns the tweets with the tag as a []string. func (userdata *Data) FindTag(tag string) TimeMap { statuses := NewTimeMap() for k, e := range userdata.Status { parts := strings.Split(e, "\t") statusslice := strings.Split(parts[3], " ") for _, v := range statusslice { if v[1:] == tag { statuses[k] = e break } } } return statuses } // SortByTime returns a string slice of the query results // sorted by time.Time func (tm TimeMapSlice) SortByTime() []string { var unionmap = NewTimeMap() var times = make(TimeSlice, 0) var data []string for _, e := range tm { for k, v := range e { unionmap[k] = v } } for k := range unionmap { times = append(times, k) } sort.Sort(times) for _, e := range times { data = append(data, unionmap[e]) } return data }