Merge pull request #145 from iopred/develop

Clean up voice state updates, make them goroutine safe.

state.go

@@ -460,41 +460,33 @@ func (s *State) MessageRemove(message *Message) error {
 	return errors.New("Message not found.")
 }
 
-func (s *State) VoiceStateUpdate(update *VoiceStateUpdate) error {
+func (s *State) voiceStateUpdate(update *VoiceStateUpdate) error {
-	var exists bool
+	guild, err := s.Guild(update.GuildID)
-	var guild *Guild
+	if err != nil {
-
+		return err
-	for _, guild = range s.Guilds {
-		if guild.ID == update.GuildID {
-			exists = true
-			break
-		}
 	}
 
-	if !exists {
+	s.Lock()
-		return nil
+	defer s.Unlock()
-	}
 
 	// Handle Leaving Channel
 	if update.ChannelID == "" {
 		for i, state := range guild.VoiceStates {
 			if state.UserID == update.UserID {
 				guild.VoiceStates = append(guild.VoiceStates[:i], guild.VoiceStates[i+1:]...)
+				return nil
 			}
 		}
 	} else {
-		exists := false
+		for i, state := range guild.VoiceStates {
-		for _, state := range guild.VoiceStates {
 			if state.UserID == update.UserID {
-				state.ChannelID = update.ChannelID
+				guild.VoiceStates[i] = update.VoiceState
-				exists = true
+				return nil
 			}
 		}
 
-		if !exists {
 		guild.VoiceStates = append(guild.VoiceStates, update.VoiceState)
 	}
-	}
 
 	return nil
 }
@@ -561,7 +553,7 @@ func (s *State) onInterface(se *Session, i interface{}) (err error) {
 	case *MessageDelete:
 		err = s.MessageRemove(t.Message)
 	case *VoiceStateUpdate:
-		err = s.VoiceStateUpdate(t)
+		err = s.voiceStateUpdate(t)
 	}
 
 	return