| | 1 | | using System; |
| | 2 | | using MUNity.Database.Context; |
| | 3 | | using MUNity.Database.Models.Organization; |
| | 4 | |
|
| | 5 | | namespace MUNity.Database.FluentAPI; |
| | 6 | |
|
| | 7 | | public class OrganizationOptionsBuilder |
| | 8 | | { |
| | 9 | | private MunityContext _context; |
| | 10 | |
|
| 25 | 11 | | public Organization Organization { get; } |
| | 12 | |
|
| | 13 | | public OrganizationOptionsBuilder WithName(string name) |
| 5 | 14 | | { |
| 5 | 15 | | Organization.OrganizationName = name; |
| 5 | 16 | | return this; |
| 5 | 17 | | } |
| | 18 | |
|
| | 19 | | public OrganizationOptionsBuilder WithShort(string shortName) |
| 5 | 20 | | { |
| 5 | 21 | | Organization.OrganizationShort = shortName; |
| 5 | 22 | | return this; |
| 5 | 23 | | } |
| | 24 | |
|
| | 25 | | public OrganizationOptionsBuilder WithAdminRole() |
| 1 | 26 | | { |
| 1 | 27 | | var adminRole = new OrganizationRole(); |
| 1 | 28 | | adminRole.Organization = this.Organization; |
| 1 | 29 | | this.Organization.Roles.Add(adminRole); |
| 1 | 30 | | adminRole.RoleName = "Admin"; |
| 1 | 31 | | adminRole.CanCreateProject = true; |
| 1 | 32 | | adminRole.CanManageMembers = true; |
| 1 | 33 | | return this; |
| 1 | 34 | | } |
| | 35 | |
|
| | 36 | | public OrganizationOptionsBuilder WithMemberRole(string roleName = "Member") |
| 1 | 37 | | { |
| 1 | 38 | | var memberRole = new OrganizationRole() |
| 1 | 39 | | { |
| 1 | 40 | | CanCreateProject = false, |
| 1 | 41 | | CanCreateRoles = false, |
| 1 | 42 | | CanManageMembers = false, |
| 1 | 43 | | Organization = this.Organization, |
| 1 | 44 | | RoleName = roleName |
| 1 | 45 | | }; |
| 1 | 46 | | this.Organization.Roles.Add(memberRole); |
| 1 | 47 | | return this; |
| 1 | 48 | | } |
| | 49 | |
|
| 5 | 50 | | public OrganizationOptionsBuilder(MunityContext context) |
| 5 | 51 | | { |
| 5 | 52 | | this._context = context; |
| 5 | 53 | | this.Organization = new Organization(); |
| 5 | 54 | | } |
| | 55 | |
|
| | 56 | | public OrganizationOptionsBuilder WithProject(Action<ProjectOptionsBuilder> options) |
| 3 | 57 | | { |
| 3 | 58 | | var builder = new ProjectOptionsBuilder(this._context, this.Organization); |
| 3 | 59 | | options(builder); |
| 3 | 60 | | this.Organization.Projects.Add(builder.Project); |
| 3 | 61 | | return this; |
| 3 | 62 | | } |
| | 63 | |
|
| | 64 | | } |