should be good for a first pass

app/controllers/graphql_controller.rb

@@ -15,6 +15,7 @@ def execute
     context = {
       # Query context goes here, for example:
       current_user: current_user,
+      current_company: current_company
     }
     result = StaffplanReduxSchema.execute(query, variables: variables, context: context, operation_name: operation_name)
     render json: result

app/graphql/mutations/base_mutation.rb

@@ -1,10 +1,7 @@
 # frozen_string_literal: true
 
 module Mutations
-  class BaseMutation < GraphQL::Schema::RelayClassicMutation
-    argument_class Types::BaseArgument
-    field_class Types::BaseField
-    input_object_class Types::BaseInputObject
-    object_class Types::BaseObject
+  class BaseMutation < GraphQL::Schema::Mutation
+    null false
   end
 end

app/graphql/mutations/create_work_week.rb

@@ -0,0 +1,14 @@
+module Mutations
+  class CreateWorkWeek < BaseMutation
+    # arguments passed to the `resolve` method
+    # argument :description, String, required: true
+    # argument :url, String, required: true
+
+    # return type from the mutation
+    type Types::StaffPlan::WorkWeekType
+
+    def resolve
+      # TODO: implement
+    end
+  end
+end

app/graphql/mutations/set_current_company.rb

@@ -0,0 +1,13 @@
+module Mutations
+  class SetCurrentCompany < BaseMutation
+    # arguments passed to the `resolve` method
+    argument :company_id, ID, required: true
+
+    # return type from the mutation
+    type Types::StaffPlan::CompanyType
+
+    def resolve
+      # TODO: implement
+    end
+  end
+end

app/graphql/mutations/update_work_week.rb

@@ -0,0 +1,14 @@
+module Mutations
+  class UpdateWorkWeek < BaseMutation
+    # arguments passed to the `resolve` method
+    # argument :description, String, required: true
+    # argument :url, String, required: true
+
+    # return type from the mutation
+    type Types::StaffPlan::WorkWeekType
+
+    def resolve
+      # TODO: implement
+    end
+  end
+end

app/graphql/types/mutation_type.rb

@@ -2,11 +2,8 @@
 
 module Types
   class MutationType < Types::BaseObject
-    # TODO: remove me
-    field :test_field, String, null: false,
-      description: "An example field added by the generator"
-    def test_field
-      "Hello World"
-    end
+    field :set_current_company, mutation: Mutations::SetCurrentCompany
+    field :create_work_week, mutation: Mutations::CreateWorkWeek
+    field :update_work_week, mutation: Mutations::UpdateWorkWeek
   end
 end

app/graphql/types/query_type.rb

@@ -23,9 +23,30 @@ def clients
       context[:current_user].current_company.clients.all
     end
 
-    field :assignments, [Types::StaffPlan::AssignmentType], null: false
-    def assignments
-      context[:current_user].assignments.all
+    field :project_assignments, [Types::StaffPlan::AssignmentType], null: false do
+      argument :project_id, ID, required: true, description: "ID of the project to fetch assignments for."
+    end
+
+    def project_assignments(project_id: nil)
+      context[:current_company]
+        .projects
+        .find(project_id)
+        .assignments
+        .all
+    end
+
+    field :user_assignments, [Types::StaffPlan::AssignmentType], null: false do
+      argument :user_id, ID, required: false,
+               description: "ID of the user to fetch assignments for. The current user's assignments will be returned if this argument is not provided."
+    end
+    def user_assignments(user_id: nil)
+      target = if user_id.present?
+        context[:current_company].users.find(user_id)
+      else
+        context[:current_user]
+      end
+
+      target.assignments.all
     end
 
     field :users, [Types::StaffPlan::UserType], null: false

app/graphql/types/staff_plan/assignment_type.rb

@@ -4,8 +4,8 @@ module Types
   module StaffPlan
     class AssignmentType < Types::BaseObject
       field :id, ID, null: false
-      field :user_id, ID, null: false
-      field :project_id, ID, null: false
+      field :user, Types::StaffPlan::UserType, null: false
+      field :project, Types::StaffPlan::ProjectType, null: false
       field :status, String, null: false
       field :created_at, GraphQL::Types::ISO8601DateTime, null: false
       field :updated_at, GraphQL::Types::ISO8601DateTime, null: false

app/graphql/types/staff_plan/project_type.rb

@@ -4,7 +4,7 @@ module Types
   module StaffPlan
     class ProjectType < Types::BaseObject
       field :id, ID, null: false
-      field :client_id, ID, null: false
+      field :client, Types::StaffPlan::ClientType, null: false
       field :name, String, null: false
       field :status, String, null: false
       field :payment_frequency, String, null: false

app/graphql/types/staff_plan/work_week_type.rb

@@ -4,6 +4,7 @@ module Types
   module StaffPlan
     class WorkWeekType < Types::BaseObject
       field :id, ID, null: false
+      field :user, Types::StaffPlan::UserType, null: false
       field :assignment_id, Integer, null: false
       field :cweek, Integer, null: false
       field :year, Integer, null: false
@@ -13,8 +14,6 @@ class WorkWeekType < Types::BaseObject
       field :created_at, GraphQL::Types::ISO8601DateTime, null: false
       field :updated_at, GraphQL::Types::ISO8601DateTime, null: false
 
-      field :user, Types::StaffPlan::UserType, null: false
-
       def user
         object.user
       end