Merge 3f52f121db into 47162eb0b8

state/mock_tracker.go

@@ -48,6 +48,16 @@ func (_mr *_MockTrackerRecorder) GetNick(arg0 interface{}) *gomock.Call {
 	return _mr.mock.ctrl.RecordCall(_mr.mock, "GetNick", arg0)
 }
 
+func (_m *MockTracker) GetNickByHostmask(host string) *Nick {
+	ret := _m.ctrl.Call(_m, "GetNickByHostmask", host)
+	ret0, _ := ret[0].(*Nick)
+	return ret0
+}
+
+func (_mr *_MockTrackerRecorder) GetNickByHostmask(arg0 interface{}) *gomock.Call {
+	return _mr.mock.ctrl.RecordCall(_mr.mock, "GetNickByHostmask", arg0)
+}
+
 func (_m *MockTracker) ReNick(old string, neu string) *Nick {
 	ret := _m.ctrl.Call(_m, "ReNick", old, neu)
 	ret0, _ := ret[0].(*Nick)

state/tracker.go

@@ -1,6 +1,8 @@
 package state
 
 import (
+	"fmt"
+
 	"github.com/fluffle/goirc/logging"
 
 	"sync"
@@ -11,6 +13,7 @@ type Tracker interface {
 	// Nick methods
 	NewNick(nick string) *Nick
 	GetNick(nick string) *Nick
+	GetNickByHostmask(host string) *Nick
 	ReNick(old, neu string) *Nick
 	DelNick(nick string) *Nick
 	NickInfo(nick, ident, host, name string) *Nick
@@ -100,6 +103,18 @@ func (st *stateTracker) GetNick(n string) *Nick {
 	return nil
 }
 
+// Returns a nick for the hostmask h, if we're tracking it.
+func (st *stateTracker) GetNickByHostmask(h string) *Nick {
+	st.mu.Lock()
+	defer st.mu.Unlock()
+	for n := range st.nicks {
+		if fmt.Sprintf("%s@%s", st.nicks[n].ident, st.nicks[n].host) == h {
+			return st.nicks[n].Nick()
+		}
+	}
+	return nil
+}
+
 // Signals to the tracker that a nick should be tracked
 // under a "neu" nick rather than the old one.
 func (st *stateTracker) ReNick(old, neu string) *Nick {

state/tracker_test.go

@@ -58,6 +58,22 @@ func TestSTGetNick(t *testing.T) {
 	}
 }
 
+func TestSTGetNickByHostmask(t *testing.T) {
+	st := NewTracker("mynick")
+	test1 := st.NewNick("test1")
+	test1 = st.NickInfo("test1", "test", "test", "test")
+
+	if n := st.GetNickByHostmask("test@test"); !test1.Equals(n) {
+		t.Errorf("Incorrect nick returned by GetNickByHostmask.")
+	}
+	if n := st.GetNickByHostmask("test2@test"); n != nil {
+		t.Errorf("Nick unexpectedly returned by GetNickByHostmask.")
+	}
+	if len(st.nicks) != 2 {
+		t.Errorf("Nick list changed size during GetNickByHostmask.")
+	}
+}
+
 func TestSTReNick(t *testing.T) {
 	st := NewTracker("mynick")
 	test1 := st.NewNick("test1")