Add create organization

Signed-off-by: gearnode <bryan@frimin.fr>

pkg/api/console/v1/schema.graphql

@@ -85,6 +85,16 @@ type PageInfo {
   endCursor: CursorKey
 }
 
+type OrganizationConnection {
+  edges: [OrganizationEdge!]!
+  pageInfo: PageInfo!
+}
+
+type OrganizationEdge {
+  cursor: CursorKey!
+  node: Organization!
+}
+
 type Organization implements Node {
   id: ID!
   name: String!
@@ -342,13 +352,17 @@ type EvidenceStateTransition {
   updatedAt: Datetime!
 }
 
-# Authentication types
-type User {
+type User implements Node {
   id: ID!
   fullName: String!
   email: String!
 
-  organizations: [Organization!]! @goField(forceResolver: true)
+  organizations(
+    first: Int
+    after: CursorKey
+    last: Int
+    before: CursorKey
+  ): OrganizationConnection! @goField(forceResolver: true)
 
   createdAt: Datetime!
   updatedAt: Datetime!
@@ -371,6 +385,12 @@ type Mutation {
   createPeople(input: CreatePeopleInput!): CreatePeoplePayload!
   updatePeople(input: UpdatePeopleInput!): People!
   deletePeople(input: DeletePeopleInput!): DeletePeoplePayload!
+  createOrganization(
+    input: CreateOrganizationInput!
+  ): CreateOrganizationPayload!
+  deleteOrganization(
+    input: DeleteOrganizationInput!
+  ): DeleteOrganizationPayload!
 }
 
 input CreateVendorInput {
@@ -474,3 +494,19 @@ type DeleteVendorPayload {
 type DeletePeoplePayload {
   deletedPeopleId: ID!
 }
+
+input CreateOrganizationInput {
+  name: String!
+}
+
+input DeleteOrganizationInput {
+  organizationId: ID!
+}
+
+type CreateOrganizationPayload {
+  organizationEdge: OrganizationEdge!
+}
+
+type DeleteOrganizationPayload {
+  deletedOrganizationId: ID!
+}

pkg/api/console/v1/types/organization.go

@@ -27,3 +27,9 @@ func NewOrganization(o *coredata.Organization) *Organization {
 		UpdatedAt: o.UpdatedAt,
 	}
 }
+
+func NewOrganizationEdge(o *coredata.Organization) *OrganizationEdge {
+	return &OrganizationEdge{
+		Node: NewOrganization(o),
+	}
+}

pkg/api/console/v1/types/types.go

@@ -59,6 +59,14 @@ type ControlStateTransitionEdge struct {
 	Node   *ControlStateTransition `json:"node"`
 }
 
+type CreateOrganizationInput struct {
+	Name string `json:"name"`
+}
+
+type CreateOrganizationPayload struct {
+	OrganizationEdge *OrganizationEdge `json:"organizationEdge"`
+}
+
 type CreatePeopleInput struct {
 	OrganizationID           gid.GID             `json:"organizationId"`
 	FullName                 string              `json:"fullName"`
@@ -88,6 +96,14 @@ type CreateVendorPayload struct {
 	VendorEdge *VendorEdge `json:"vendorEdge"`
 }
 
+type DeleteOrganizationInput struct {
+	OrganizationID gid.GID `json:"organizationId"`
+}
+
+type DeleteOrganizationPayload struct {
+	DeletedOrganizationID gid.GID `json:"deletedOrganizationId"`
+}
+
 type DeletePeopleInput struct {
 	PeopleID gid.GID `json:"peopleId"`
 }
@@ -186,6 +202,16 @@ type Organization struct {
 func (Organization) IsNode()             {}
 func (this Organization) GetID() gid.GID { return this.ID }
 
+type OrganizationConnection struct {
+	Edges    []*OrganizationEdge `json:"edges"`
+	PageInfo *PageInfo           `json:"pageInfo"`
+}
+
+type OrganizationEdge struct {
+	Cursor page.CursorKey `json:"cursor"`
+	Node   *Organization  `json:"node"`
+}
+
 type PageInfo struct {
 	HasNextPage     bool            `json:"hasNextPage"`
 	HasPreviousPage bool            `json:"hasPreviousPage"`
@@ -292,14 +318,17 @@ type UpdateVendorInput struct {
 }
 
 type User struct {
-	ID            gid.GID         `json:"id"`
-	FullName      string          `json:"fullName"`
-	Email         string          `json:"email"`
-	Organizations []*Organization `json:"organizations"`
-	CreatedAt     time.Time       `json:"createdAt"`
-	UpdatedAt     time.Time       `json:"updatedAt"`
+	ID            gid.GID                 `json:"id"`
+	FullName      string                  `json:"fullName"`
+	Email         string                  `json:"email"`
+	Organizations *OrganizationConnection `json:"organizations"`
+	CreatedAt     time.Time               `json:"createdAt"`
+	UpdatedAt     time.Time               `json:"updatedAt"`
 }
 
+func (User) IsNode()             {}
+func (this User) GetID() gid.GID { return this.ID }
+
 type Vendor struct {
 	ID                   gid.GID                     `json:"id"`
 	Name                 string                      `json:"name"`

pkg/api/console/v1/v1_resolver.go

@@ -169,6 +169,30 @@ func (r *mutationResolver) DeletePeople(ctx context.Context, input types.DeleteP
 	}, nil
 }
 
+// CreateOrganization is the resolver for the createOrganization field.
+func (r *mutationResolver) CreateOrganization(ctx context.Context, input types.CreateOrganizationInput) (*types.CreateOrganizationPayload, error) {
+	organization, err := r.proboSvc.CreateOrganization(ctx, probo.CreateOrganizationRequest{
+		Name: input.Name,
+	})
+	if err != nil {
+		return nil, fmt.Errorf("cannot create organization: %w", err)
+	}
+
+	err = r.usrmgrSvc.AddUserToOrganization(ctx, UserFromContext(ctx).ID, organization.ID)
+	if err != nil {
+		return nil, fmt.Errorf("cannot add user to organization: %w", err)
+	}
+
+	return &types.CreateOrganizationPayload{
+		OrganizationEdge: types.NewOrganizationEdge(organization),
+	}, nil
+}
+
+// DeleteOrganization is the resolver for the deleteOrganization field.
+func (r *mutationResolver) DeleteOrganization(ctx context.Context, input types.DeleteOrganizationInput) (*types.DeleteOrganizationPayload, error) {
+	panic(fmt.Errorf("not implemented: DeleteOrganization - deleteOrganization"))
+}
+
 // Frameworks is the resolver for the frameworks field.
 func (r *organizationResolver) Frameworks(ctx context.Context, obj *types.Organization, first *int, after *page.CursorKey, last *int, before *page.CursorKey) (*types.FrameworkConnection, error) {
 	cursor := types.NewCursor(first, after, last, before)
@@ -300,29 +324,35 @@ func (r *taskResolver) Evidences(ctx context.Context, obj *types.Task, first *in
 }
 
 // Organizations is the resolver for the organizations field.
-func (r *userResolver) Organizations(ctx context.Context, obj *types.User) ([]*types.Organization, error) {
+func (r *userResolver) Organizations(ctx context.Context, obj *types.User, first *int, after *page.CursorKey, last *int, before *page.CursorKey) (*types.OrganizationConnection, error) {
 	// Get the user's organization IDs
 	organizationIDs, err := r.usrmgrSvc.GetUserOrganizations(ctx, obj.ID)
 	if err != nil {
 		return nil, fmt.Errorf("failed to get user organizations: %w", err)
 	}
 
-	// If the user doesn't have any organizations, return an empty slice
+	// If the user doesn't have any organizations, return an empty connection
 	if len(organizationIDs) == 0 {
-		return []*types.Organization{}, nil
+		return &types.OrganizationConnection{
+			Edges:    []*types.OrganizationEdge{},
+			PageInfo: &types.PageInfo{},
+		}, nil
 	}
 
 	// Get the organization details for each organization ID
-	var organizations []*types.Organization
+	var edges []*types.OrganizationEdge
 	for _, organizationID := range organizationIDs {
 		organization, err := r.proboSvc.GetOrganization(ctx, organizationID)
 		if err != nil {
 			return nil, fmt.Errorf("failed to get organization details: %w", err)
 		}
-		organizations = append(organizations, types.NewOrganization(organization))
+		edges = append(edges, types.NewOrganizationEdge(organization))
 	}
 
-	return organizations, nil
+	return &types.OrganizationConnection{
+		Edges:    edges,
+		PageInfo: &types.PageInfo{},
+	}, nil
 }
 
 // Control returns schema.ControlResolver implementation.