TaskableAssignee basic implementation

Author: despo

app/models/todoro/task.rb

@@ -4,6 +4,10 @@ class Task < ApplicationRecord
     has_many :reminders, dependent: :destroy
     has_many :task_steps, class_name: "Todoro::TaskStep", dependent: :destroy
 
+    has_many :task_assignments, dependent: :destroy
+    has_many :assignees, through: :task_assignments, source: :assignee
+
+
     validates :title, presence: true
     validates :status, presence: true, inclusion: { in: %w[pending completed] }
 
@@ -17,6 +21,14 @@ def complete!
       update(status: "completed", completed_at: Time.zone.now)
     end
 
+    def assign_to(assignee)
+      task_assignments.find_or_create_by(assignee: assignee)
+    end
+
+    def unassign(assignee)
+      task_assignments.where(assignee: assignee).destroy_all
+    end
+
     private
 
     def set_default_reminders

app/models/todoro/task_assignment.rb

@@ -0,0 +1,16 @@
+module Todoro
+  class TaskAssignment < ApplicationRecord
+    belongs_to :task, class_name: "Todoro::Task"
+    belongs_to :assignee, polymorphic: true
+
+    validate :assignee_must_be_taskable
+
+    private
+
+    def assignee_must_be_taskable
+      unless Todoro.assignable_models.include?(assignee_type)
+        errors.add(:assignee, "#{assignee_type} is not a registered taskable model")
+      end
+    end
+  end
+end

lib/generators/todoro/templates/create_todoro_task_assignments.rb

@@ -0,0 +1,10 @@
+class CreateTodoroTaskAssignments < ActiveRecord::Migration[ActiveRecord::Migration.current_version]
+  def change
+    create_table :todoro_task_assignments do |t|
+      t.references :task, null: false, foreign_key: { to_table: :todoro_tasks }
+      t.references :assignee, polymorphic: true, null: false
+
+      t.timestamps
+    end
+  end
+end

lib/todoro.rb

@@ -3,6 +3,9 @@
 require "todoro/taskable"
 
 module Todoro
+  mattr_accessor :assignable_models
+  self.assignable_models = []
+
   def self.taskable_models
     Rails.application.eager_load!
 

lib/todoro/taskable_assignee.rb

@@ -0,0 +1,16 @@
+module Todoro
+  module TaskableAssignee
+    extend ActiveSupport::Concern
+
+    included do
+      has_many :task_assignments, as: :assignee, class_name: "Todoro::TaskAssignment", dependent: :destroy
+      has_many :tasks, through: :task_assignments, source: :task
+    end
+
+    class_methods do
+      def taskable_assignee
+        Todoro.assignable_models << self.name unless Todoro.assignable_models.include?(self.name)
+      end
+    end
+  end
+end