This commit is contained in:
Shivaram Lingamneni 2019-05-21 23:55:04 -04:00
parent eaf0328608
commit 1121e3d320
3 changed files with 8 additions and 1 deletions

View File

@ -1108,6 +1108,10 @@ func (am *AccountManager) VHostReject(account string, reason string) (result VHo
func (am *AccountManager) VHostSetEnabled(client *Client, enabled bool) (result VHostInfo, err error) { func (am *AccountManager) VHostSetEnabled(client *Client, enabled bool) (result VHostInfo, err error) {
munger := func(input VHostInfo) (output VHostInfo, err error) { munger := func(input VHostInfo) (output VHostInfo, err error) {
if input.ApprovedVHost == "" {
err = errNoVhost
return
}
output = input output = input
output.Enabled = enabled output.Enabled = enabled
return return

View File

@ -44,6 +44,7 @@ var (
errFeatureDisabled = errors.New(`That feature is disabled`) errFeatureDisabled = errors.New(`That feature is disabled`)
errBanned = errors.New("IP or nickmask banned") errBanned = errors.New("IP or nickmask banned")
errInvalidParams = utils.ErrInvalidParams errInvalidParams = utils.ErrInvalidParams
errNoVhost = errors.New(`You do not have an approved vhost`)
) )
// Socket Errors // Socket Errors

View File

@ -154,7 +154,9 @@ func hsOnOffHandler(server *Server, client *Client, command string, params []str
} }
_, err := server.accounts.VHostSetEnabled(client, enable) _, err := server.accounts.VHostSetEnabled(client, enable)
if err != nil { if err == errNoVhost {
hsNotice(rb, client.t(err.Error()))
} else if err != nil {
hsNotice(rb, client.t("An error occurred")) hsNotice(rb, client.t("An error occurred"))
} else if enable { } else if enable {
hsNotice(rb, client.t("Successfully enabled your vhost")) hsNotice(rb, client.t("Successfully enabled your vhost"))