coder/enterprise/coderd/portsharing/portsharing.go

package portsharing

import (
	"golang.org/x/xerrors"

	"github.com/coder/coder/v2/coderd/database"
	"github.com/coder/coder/v2/codersdk"
)

type EnterprisePortSharer struct{}

func NewEnterprisePortSharer() *EnterprisePortSharer {
	return &EnterprisePortSharer{}
}

func (EnterprisePortSharer) AuthorizedLevel(template database.Template, level codersdk.WorkspaceAgentPortShareLevel) error {
	maxLevel := codersdk.WorkspaceAgentPortShareLevel(template.MaxPortSharingLevel)
	switch level {
	case codersdk.WorkspaceAgentPortShareLevelPublic:
		if maxLevel != codersdk.WorkspaceAgentPortShareLevelPublic {
			return xerrors.Errorf("port sharing level not allowed. Max level is '%s'", maxLevel)
		}
	case codersdk.WorkspaceAgentPortShareLevelAuthenticated:
		if maxLevel == codersdk.WorkspaceAgentPortShareLevelOwner {
			return xerrors.Errorf("port sharing level not allowed. Max level is '%s'", maxLevel)
		}
	default:
		return xerrors.New("port sharing level is invalid.")
	}

	return nil
}

func (EnterprisePortSharer) ValidateTemplateMaxLevel(level codersdk.WorkspaceAgentPortShareLevel) error {
	if !level.ValidMaxLevel() {
		return xerrors.New("invalid max port sharing level, value must be 'authenticated' or 'public'.")
	}

	return nil
}

func (EnterprisePortSharer) ConvertMaxLevel(level database.AppSharingLevel) codersdk.WorkspaceAgentPortShareLevel {
	return codersdk.WorkspaceAgentPortShareLevel(level)
}