Skip to content

Commit

Permalink
fix auto mode, use mumble ping, die properly
Browse files Browse the repository at this point in the history
  • Loading branch information
stryan committed Jan 19, 2021
1 parent daacaaa commit 577c5e7
Showing 1 changed file with 22 additions and 12 deletions.
34 changes: 22 additions & 12 deletions bridge.go
Expand Up @@ -50,6 +50,9 @@ type BridgeState struct {
MumbleUsers map[string]bool
MumbleUsersMutex sync.Mutex

// Total Number of Mumble users
MumbleUserCount int

// Kill the auto connect routine
AutoChanDie chan bool

Expand Down Expand Up @@ -129,14 +132,21 @@ func (b *BridgeState) startBridge() {
ticker := time.NewTicker(500 * time.Millisecond)
for {
<-ticker.C
if b.MumbleClient.State() != 2 {
log.Println("Lost mumble connection " + strconv.Itoa(int(b.MumbleClient.State())))
if b.MumbleClient == nil || b.MumbleClient.State() != 2 {
if b.MumbleClient != nil {
log.Println("Lost mumble connection " + strconv.Itoa(int(b.MumbleClient.State())))
} else {
log.Println("Lost mumble connection due to bridge dieing")
return
}
select {
default:
close(b.BridgeDie)
case <-b.BridgeDie:
//die is already closed

default:
close(b.BridgeDie)
}

}
}
}()
Expand All @@ -150,6 +160,7 @@ func (b *BridgeState) startBridge() {
det.Detach()
close(toDiscord)
close(toMumble)
close(b.BridgeDie)
b.Connected = false
b.DiscordVoice = nil
b.MumbleClient = nil
Expand All @@ -170,17 +181,17 @@ func (b *BridgeState) discordStatusUpdate() {
b.DiscordSession.UpdateListeningStatus("an error pinging mumble")
} else {
b.MumbleUsersMutex.Lock()
userCount := resp.ConnectedUsers
b.MumbleUserCount = resp.ConnectedUsers
if b.Connected {
userCount = userCount - 1
b.MumbleUserCount = b.MumbleUserCount - 1
}
if userCount == 0 {
if b.MumbleUserCount == 0 {
status = "No users in Mumble"
} else {
if len(b.MumbleUsers) > 0 {
status = fmt.Sprintf("%v/%v users in Mumble\n", len(b.MumbleUsers), userCount)
status = fmt.Sprintf("%v/%v users in Mumble\n", len(b.MumbleUsers), b.MumbleUserCount)
} else {
status = fmt.Sprintf("%v users in Mumble\n", userCount)
status = fmt.Sprintf("%v users in Mumble\n", b.MumbleUserCount)
}
}
b.MumbleUsersMutex.Unlock()
Expand All @@ -207,14 +218,13 @@ func (b *BridgeState) AutoBridge() {
b.MumbleUsersMutex.Lock()
b.DiscordUsersMutex.Lock()

if !b.Connected && len(b.MumbleUsers) > 0 && len(b.DiscordUsers) > 0 {
if !b.Connected && b.MumbleUserCount > 0 && len(b.DiscordUsers) > 0 {
log.Println("users detected in mumble and discord, bridging")
go b.startBridge()
}
if b.Connected && len(b.MumbleUsers) == 0 && len(b.DiscordUsers) <= 1 {
if b.Connected && b.MumbleUserCount == 0 && len(b.DiscordUsers) <= 1 {
log.Println("no one online, killing bridge")
b.BridgeDie <- true
b.BridgeDie = nil
}

b.MumbleUsersMutex.Unlock()
Expand Down

0 comments on commit 577c5e7

Please sign in to comment.