package wifi

import (
	"time"

	"github.com/bettercap/bettercap/network"
)

func (mod *WiFiModule) onChannel(channel int, cb func()) {
	mod.chanLock.Lock()
	defer mod.chanLock.Unlock()

	prev := mod.stickChan
	mod.stickChan = channel

	if err := network.SetInterfaceChannel(mod.Session.Interface.Name(), channel); err != nil {
		mod.Warning("error while hopping to channel %d: %s", channel, err)
	} else {
		mod.Debug("hopped on channel %d", channel)
	}

	cb()

	mod.stickChan = prev
}

func (mod *WiFiModule) channelHopper() {
	mod.reads.Add(1)
	defer mod.reads.Done()

	mod.Info("channel hopper started.")

	for mod.Running() {
		delay := mod.hopPeriod
		// if we have both 2.4 and 5ghz capabilities, we have
		// more channels, therefore we need to increase the time
		// we hop on each one otherwise me lose information
		if len(mod.frequencies) > 14 {
			delay = delay * 2
		}

		frequencies := mod.frequencies

	loopCurrentChannels:
		for _, frequency := range frequencies {
			channel := network.Dot11Freq2Chan(frequency)
			// stick to the access point channel as long as it's selected
			// or as long as we're deauthing on it
			if mod.stickChan != 0 {
				channel = mod.stickChan
			}

			mod.Debug("hopping on channel %d", channel)

			mod.chanLock.Lock()
			if err := network.SetInterfaceChannel(mod.Session.Interface.Name(), channel); err != nil {
				mod.Warning("error while hopping to channel %d: %s", channel, err)
			}
			mod.chanLock.Unlock()

			select {
			case _ = <-mod.hopChanges:
				mod.Debug("hop changed")
				break loopCurrentChannels
			case <-time.After(delay):
				if !mod.Running() {
					return
				}
			}
		}
	}
}