Function WithEntityScaleSetFilter

database/watcher/filters.go:106–138

WithEntityScaleSetFilter returns true if the change payload is a scale set that belongs to the supplied Github entity.

(ghEntity params.ForgeEntity)

// WithEntityScaleSetFilter returns true if the change payload is a scale set that belongs to the
// supplied Github entity.
func WithEntityScaleSetFilter(ghEntity params.ForgeEntity) dbCommon.PayloadFilterFunc {
	return func(payload dbCommon.ChangePayload) bool {
		forgeType, err := ghEntity.GetForgeType()
		if err != nil {
			return false
		}

		// Gitea does not have scale sets.
		if forgeType == params.GiteaEndpointType {
			return false
		}

		switch payload.EntityType {
		case dbCommon.ScaleSetEntityType:
			scaleSet, ok := payload.Payload.(params.ScaleSet)
			if !ok {
				return false
			}
			switch ghEntity.EntityType {
			case params.ForgeEntityTypeRepository:
				return scaleSet.RepoID == ghEntity.ID
			case params.ForgeEntityTypeOrganization:
				return scaleSet.OrgID == ghEntity.ID
			case params.ForgeEntityTypeEnterprise:
				return scaleSet.EnterpriseID == ghEntity.ID
			default:
				return false
			}
		default:
			return false
		}
	}
}

// WithEntityFilter returns a filter function that filters payloads by entity.
// Change payloads that match the entity type and ID will return true.