implements project languages saving them onto the database

1 files changed, 21 insertions, 0 deletions

diff --git a/db/migrate/20160229193553_add_main_language_to_repository.rb b/db/migrate/20160229193553_add_main_language_to_repository.rb
new file mode 100644
index 00000000000..da24abd2f04
--- /dev/null
+++ b/db/migrate/20160229193553_add_main_language_to_repository.rb
@@ -0,0 +1,21 @@
+class AddMainLanguageToRepository < ActiveRecord::Migration
+  require 'rugged'
+  require 'linguist'
+
+  def up
+    add_column :projects, :main_language, :string, default: nil
+
+    Project.all.each do |project|
+      unless project.repository.empty?
+        language = Linguist::Repository.new(
+          project.repository.rugged,
+          project.repository.rugged.head.target_id).language
+        project.update_attributes(main_language: language)
+      end
+    end
+  end
+
+  def down
+    remove_column :projects, :main_language
+  end
+end