From fe43eb924ee1b321a751dd0b3ef428e920be49ff Mon Sep 17 00:00:00 2001
From: 0x1eef <0x1eef@protonmail.com>
Date: Thu, 23 May 2024 18:23:13 -0300
Subject: [PATCH] Add Twenty::Milestone#project

---
 .../migration/6_add_project_id_to_milestones.rb       | 11 +++++++++++
 server/lib/twenty/server/model/milestone.rb           |  2 ++
 2 files changed, 13 insertions(+)
 create mode 100644 server/lib/twenty/server/migration/6_add_project_id_to_milestones.rb

diff --git a/server/lib/twenty/server/migration/6_add_project_id_to_milestones.rb b/server/lib/twenty/server/migration/6_add_project_id_to_milestones.rb
new file mode 100644
index 0000000..77d23f1
--- /dev/null
+++ b/server/lib/twenty/server/migration/6_add_project_id_to_milestones.rb
@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+Sequel.migration do
+  up do
+    add_column :milestones, :project_id, Integer, null: false
+  end
+
+  down do
+    drop_column :milestones, :project_id
+  end
+end
diff --git a/server/lib/twenty/server/model/milestone.rb b/server/lib/twenty/server/model/milestone.rb
index a5c45e8..191a622 100644
--- a/server/lib/twenty/server/model/milestone.rb
+++ b/server/lib/twenty/server/model/milestone.rb
@@ -2,7 +2,9 @@ module Twenty
   class Milestone < Sequel::Model
     include Model
     validates_presence_of :name
+    validates_presence_of :project
 
+    many_to_one :project
     many_to_many :tasks,
                  class_name: "Twenty::Task",
                  join_table: :tasks_milestones