zrok/controller/limits/shareWarningAction.go

package limits

import (
	"github.com/jmoiron/sqlx"
	"github.com/openziti/edge-api/rest_management_api_client"
	"github.com/openziti/zrok/controller/emailUi"
	"github.com/openziti/zrok/controller/store"
	"github.com/openziti/zrok/util"
	"github.com/pkg/errors"
	"github.com/sirupsen/logrus"
)

type shareWarningAction struct {
	str  *store.Store
	edge *rest_management_api_client.ZitiEdgeManagement
	cfg  *emailUi.Config
}

func newShareWarningAction(cfg *emailUi.Config, str *store.Store, edge *rest_management_api_client.ZitiEdgeManagement) *shareWarningAction {
	return &shareWarningAction{str, edge, cfg}
}

func (a *shareWarningAction) HandleShare(shr *store.Share, rxBytes, txBytes int64, limit *BandwidthPerPeriod, trx *sqlx.Tx) error {
	logrus.Infof("warning '%v'", shr.Token)

	if a.cfg != nil {
		env, err := a.str.GetEnvironment(shr.EnvironmentId, trx)
		if err != nil {
			return err
		}

		if env.AccountId != nil {
			acct, err := a.str.GetAccount(*env.AccountId, trx)
			if err != nil {
				return err
			}

			rxLimit := "unlimited bytes"
			if limit.Limit.Rx != Unlimited {
				rxLimit = util.BytesToSize(limit.Limit.Rx)
			}
			txLimit := "unlimited bytes"
			if limit.Limit.Tx != Unlimited {
				txLimit = util.BytesToSize(limit.Limit.Tx)
			}
			totalLimit := "unlimited bytes"
			if limit.Limit.Total != Unlimited {
				totalLimit = util.BytesToSize(limit.Limit.Total)
			}

			detail := newDetailMessage()
			detail = detail.append("Your share '%v' has received %v and sent %v (for a total of %v), which has triggered a transfer limit warning.", shr.Token, util.BytesToSize(rxBytes), util.BytesToSize(txBytes), util.BytesToSize(rxBytes+txBytes))
			detail = detail.append("This zrok instance only allows a share to receive %v, send %v, totalling not more than %v for each %v.", rxLimit, txLimit, totalLimit, limit.Period)
			detail = detail.append("If you exceed the transfer limit, access to your shares will be temporarily disabled (until the last %v falls below the transfer limit).", limit.Period)

			if err := sendLimitWarningEmail(a.cfg, acct.Email, detail); err != nil {
				return errors.Wrapf(err, "error sending limit warning email to '%v'", acct.Email)
			}
		}
	} else {
		logrus.Warnf("skipping warning email for share limit; no email configuration specified")
	}

	return nil
}