Add imported boolean to project. Prevent shell from creating repo after it was imported

app/contexts/projects/create_context.rb

@@ -51,6 +51,7 @@ module Projects
         if shell.import_repository(@project.path_with_namespace, @project.import_url)
           # We should create satellite for imported repo
           @project.satellite.create unless @project.satellite.exists?
+          @project.imported = true
           true
         else
           @project.errors.add(:import_url, 'cannot clone repo')

app/helpers/application_helper.rb

@@ -142,7 +142,7 @@ module ApplicationHelper
   end
 
   def user_color_scheme_class
-    COLOR_SCHEMES[current_user.try(:color_scheme_id)]
+    COLOR_SCHEMES[current_user.try(:color_scheme_id)] if defined?(current_user)
   end
 
   # Define whenever show last push event

app/models/project.rb

@@ -413,6 +413,10 @@ class Project < ActiveRecord::Base
     !(forked_project_link.nil? || forked_project_link.forked_from_project.nil?)
   end
 
+  def imported?
+    imported
+  end
+
   def rename_repo
     old_path_with_namespace = File.join(namespace_dir, path_was)
     new_path_with_namespace = File.join(namespace_dir, path)

app/observers/project_observer.rb

 class ProjectObserver < BaseObserver
   def after_create(project)
-    unless project.forked?
+    return true if project.forked? || project.imported?
+
     GitlabShellWorker.perform_async(
       :add_repository,
       project.path_with_namespace
@@ -8,7 +9,6 @@ class ProjectObserver < BaseObserver
 
     log_info("#{project.owner.name} created a new project \"#{project.name_with_namespace}\"")
   end
-  end
 
   def after_update(project)
     project.send_move_instructions if project.namespace_id_changed?

db/migrate/20130614132337_add_improted_to_project.rb

+class AddImprotedToProject < ActiveRecord::Migration
+  def change
+    add_column :projects, :imported, :boolean, default: false, null: false
+  end
+end

db/schema.rb

@@ -11,7 +11,7 @@
 #
 # It's strongly recommended to check this file into your version control system.
 
-ActiveRecord::Schema.define(:version => 20130613173246) do
+ActiveRecord::Schema.define(:version => 20130614132337) do
 
   create_table "deploy_keys_projects", :force => true do |t|
     t.integer  "deploy_key_id", :null => false
@@ -172,6 +172,7 @@ ActiveRecord::Schema.define(:version => 20130613173246) do
     t.string   "issues_tracker_id"
     t.boolean  "snippets_enabled",       :default => true,     :null => false
     t.datetime "last_activity_at"
+    t.boolean  "imported",               :default => false,    :null => false
   end
 
   add_index "projects", ["creator_id"], :name => "index_projects_on_owner_id"