package modules

import (
	"github.com/evilsocket/bettercap-ng/network"
)

type WiFiStation struct {
	*network.Endpoint
	IsAP    bool
	Channel int
}

func NewWiFiStation(essid, bssid string, isAp bool, channel int) *WiFiStation {
	return &WiFiStation{
		Endpoint: network.NewEndpointNoResolve(network.MonitorModeAddress, bssid, essid, 0),
		IsAP:     isAp,
		Channel:  channel,
	}
}

func (s WiFiStation) BSSID() string {
	return s.HwAddress
}

func (s *WiFiStation) ESSID() string {
	return s.Hostname
}