Skip to content

WIP clean up orga creation params #8760

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
wants to merge 2 commits into
base: master
Choose a base branch
from
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
75 changes: 29 additions & 46 deletions app/controllers/AuthenticationController.scala
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,24 @@ import java.security.MessageDigest
import javax.inject.Inject
import scala.concurrent.{ExecutionContext, Future}

case class CreateUserInOrganizationParameters(firstName: String,
lastName: String,
email: String,
password: Option[String],
autoActivate: Option[Boolean])

object CreateUserInOrganizationParameters {
implicit val jsonFormat: OFormat[CreateUserInOrganizationParameters] =
Json.format[CreateUserInOrganizationParameters]
}

case class InviteParameters(
recipients: List[String],
autoActivate: Boolean
)

object InviteParameters { implicit val jsonFormat: Format[InviteParameters] = Json.format[InviteParameters] }

class AuthenticationController @Inject()(
actorSystem: ActorSystem,
credentialsProvider: CredentialsProvider,
Expand Down Expand Up @@ -83,9 +101,9 @@ class AuthenticationController @Inject()(
} else {
for {
inviteBox <- inviteService.findInviteByTokenOpt(signUpData.inviteToken).shiftBox
organizationId = Option(signUpData.organization).filter(_.trim.nonEmpty)
organization <- organizationService.findOneByInviteByIdOrDefault(inviteBox.toOption, organizationId)(
GlobalAccessContext) ?~> Messages("organization.notFound", signUpData.organization)
organization <- organizationService.findOneByInviteByIdOrDefault(inviteBox.toOption,
signUpData.organizationId)(
GlobalAccessContext) ?~> Messages("organization.notFound", signUpData.organizationId)
_ <- organizationService
.assertUsersCanBeAdded(organization._id)(GlobalAccessContext, ec) ?~> "organization.users.userLimitReached"
autoActivate = inviteBox.toOption.map(_.autoActivate).getOrElse(organization.enableAutoVerify)
Expand Down Expand Up @@ -501,9 +519,8 @@ class AuthenticationController @Inject()(
} else {
for {
_ <- initialDataService.insertLocalDataStoreIfEnabled()
organization <- organizationService.createOrganization(
Option(signUpData.organization).filter(_.trim.nonEmpty),
signUpData.organizationName) ?~> "organization.create.failed"
organization <- organizationService
.createOrganization(signUpData.organizationName) ?~> "organization.create.failed"
user <- userService.insert(
organization._id,
email,
Expand Down Expand Up @@ -545,17 +562,6 @@ class AuthenticationController @Inject()(
_ <- organizationService.acceptTermsOfService(user._organization, acceptedVersion)(DBAccessContext(Some(user)), m)
} yield ()

case class CreateUserInOrganizationParameters(firstName: String,
lastName: String,
email: String,
password: Option[String],
autoActivate: Option[Boolean])

object CreateUserInOrganizationParameters {
implicit val jsonFormat: OFormat[CreateUserInOrganizationParameters] =
Json.format[CreateUserInOrganizationParameters]
}

def createUserInOrganization(organizationId: String): Action[CreateUserInOrganizationParameters] =
sil.SecuredAction.async(validateJson[CreateUserInOrganizationParameters]) { implicit request =>
for {
Expand Down Expand Up @@ -615,21 +621,12 @@ class AuthenticationController @Inject()(

}

case class InviteParameters(
recipients: List[String],
autoActivate: Boolean
)

object InviteParameters {
implicit val jsonFormat: Format[InviteParameters] = Json.format[InviteParameters]
}

trait AuthForms {

private val passwordMinLength = 8

// Sign up
case class SignUpData(organization: String,
case class SignUpData(organizationId: Option[String], // None for createOrganizationWithAdmin route
organizationName: String,
email: String,
firstName: String,
Expand All @@ -638,32 +635,18 @@ trait AuthForms {
inviteToken: Option[String],
acceptedTermsOfService: Option[Int])

def signUpForm(implicit messages: Messages): Form[SignUpData] =
def signUpForm: Form[SignUpData] =
Form(
mapping(
"organization" -> text,
"organizationName" -> text,
"organizationId" -> optional(nonEmptyText),
"organizationName" -> nonEmptyText,
"email" -> email,
"password" -> tuple(
"password1" -> nonEmptyText.verifying(minLength(passwordMinLength)),
"password2" -> nonEmptyText
).verifying(Messages("error.passwordsDontMatch"), password => password._1 == password._2),
"password" -> nonEmptyText.verifying(minLength(passwordMinLength)),
"firstName" -> nonEmptyText,
"lastName" -> nonEmptyText,
"inviteToken" -> optional(nonEmptyText),
"acceptedTermsOfService" -> optional(number)
)((organization, organizationName, email, password, firstName, lastName, inviteToken, acceptTos) =>
SignUpData(organization, organizationName, email, firstName, lastName, password._1, inviteToken, acceptTos))(
signUpData =>
Some(
(signUpData.organization,
signUpData.organizationName,
signUpData.email,
("", ""),
signUpData.firstName,
signUpData.lastName,
signUpData.inviteToken,
signUpData.acceptedTermsOfService))))
)(SignUpData.apply)(SignUpData.unapply))

// Sign in
case class SignInData(email: String, password: String)
Expand Down
4 changes: 2 additions & 2 deletions app/controllers/OrganizationController.scala
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ class OrganizationController @Inject()(
} yield Ok(Json.toJson(js))
}

case class OrganizationCreationParameters(organization: Option[String], organizationName: String, ownerEmail: String)
case class OrganizationCreationParameters(organizationName: String, ownerEmail: String)
object OrganizationCreationParameters {
implicit val jsonFormat: OFormat[OrganizationCreationParameters] = Json.format[OrganizationCreationParameters]
}
Expand All @@ -75,7 +75,7 @@ class OrganizationController @Inject()(
for {
_ <- userService.assertIsSuperUser(request.identity._multiUser) ?~> "notAllowed" ~> FORBIDDEN
owner <- multiUserDAO.findOneByEmail(request.body.ownerEmail) ?~> "user.notFound"
org <- organizationService.createOrganization(request.body.organization, request.body.organizationName)
org <- organizationService.createOrganization(request.body.organizationName)
user <- userDAO.findFirstByMultiUser(owner._id)
_ <- userService.joinOrganization(user,
org._id,
Expand Down
48 changes: 23 additions & 25 deletions app/models/organization/OrganizationService.scala
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ package models.organization
import com.scalableminds.util.accesscontext.{DBAccessContext, GlobalAccessContext}
import com.scalableminds.util.objectid.ObjectId
import com.scalableminds.util.time.Instant
import com.scalableminds.util.tools.{Fox, FoxImplicits, TextUtils}
import com.scalableminds.util.tools.{Fox, FoxImplicits}
import com.scalableminds.webknossos.datastore.rpc.RPC
import com.typesafe.scalalogging.LazyLogging

Expand All @@ -14,6 +14,7 @@ import models.team.{PricingPlan, Team, TeamDAO}
import models.user.{Invite, MultiUserDAO, User, UserDAO, UserService}
import play.api.i18n.{Messages, MessagesProvider}
import play.api.libs.json.{JsArray, JsObject, Json}
import security.RandomIDGenerator
import utils.WkConf

import scala.concurrent.{ExecutionContext, Future}
Expand Down Expand Up @@ -109,35 +110,32 @@ class OrganizationService @Inject()(organizationDAO: OrganizationDAO,
_ <- Fox.fromBool(organizations.isEmpty) ?~> "organizationsNotEmpty"
} yield ()

def createOrganization(organizationIdOpt: Option[String], organizationName: String): Fox[Organization] =
for {
normalizedName <- TextUtils.normalizeStrong(organizationName).toFox ?~> "organization.id.invalid"
organizationId = organizationIdOpt
.flatMap(TextUtils.normalizeStrong)
.getOrElse(normalizedName)
.replaceAll(" ", "_")
existingOrganization <- organizationDAO.findOne(organizationId)(GlobalAccessContext).shiftBox
_ <- Fox.fromBool(existingOrganization.isEmpty) ?~> "organization.id.alreadyInUse"
initialPricingParameters = if (conf.Features.isWkorgInstance) (PricingPlan.Basic, Some(3), Some(50000000000L))
private def generateOrganizationId: String = RandomIDGenerator.generateBlocking(24)

def createOrganization(organizationName: String): Fox[Organization] = {
val initialPricingParameters =
if (conf.Features.isWkorgInstance) (PricingPlan.Basic, Some(3), Some(50000000000L))
else (PricingPlan.Custom, None, None)
organizationRootFolder = Folder(ObjectId.generate, folderService.defaultRootName, JsArray.empty)

organization = Organization(
organizationId,
"",
"",
organizationName,
initialPricingParameters._1,
None,
initialPricingParameters._2,
initialPricingParameters._3,
organizationRootFolder._id
)
organizationTeam = Team(ObjectId.generate, organization._id, "Default", isOrganizationTeam = true)

val organizationRootFolder = Folder(ObjectId.generate, folderService.defaultRootName, JsArray.empty)
val organization = Organization(
generateOrganizationId,
"",
"",
organizationName,
initialPricingParameters._1,
None,
initialPricingParameters._2,
initialPricingParameters._3,
organizationRootFolder._id
)
val organizationTeam = Team(ObjectId.generate, organization._id, "Default", isOrganizationTeam = true)
for {
_ <- folderDAO.insertAsRoot(organizationRootFolder)
_ <- organizationDAO.insertOne(organization)
_ <- teamDAO.insertOne(organizationTeam)
} yield organization
}

def createOrganizationDirectory(organizationId: String, dataStoreToken: String): Fox[Unit] = {
def sendRPCToDataStore(dataStore: DataStore) =
Expand Down
2 changes: 1 addition & 1 deletion conf/application.conf
Original file line number Diff line number Diff line change
Expand Up @@ -146,7 +146,7 @@ features {
discussionBoard = "https://forum.image.sc/tag/webknossos"
discussionBoardRequiresAdmin = false
hideNavbarLogin = false
isWkorgInstance = false
isWkorgInstance = true
recommendWkorgInstance = true
taskReopenAllowedInSeconds = 30
allowDeleteDatasets = true
Expand Down
Loading