Add a Rake task to make the current node the primary Geo node

To support #1664, we need a way to add a Geo Node to the database with an input SSH key.

changelogs/unreleased-ee/sh-add-primary-node-rake-task.yml

+---
+title: Add a Rake task to make the current node the primary Geo node
+merge_request:
+author:

lib/tasks/geo.rake

@@ -72,4 +72,39 @@ namespace :geo do
       Rails.application.config = @previous_config[:config]
     end
   end
+
+  desc 'Make this node the Geo primary'
+  task :add_primary_node, [:ssh_key_filename] => :environment do |_, args|
+    filename = args[:ssh_key_filename]
+    abort 'GitLab Geo is not supported with this license. Please contact sales@gitlab.com.' unless Gitlab::Geo.license_allows?
+    abort 'You must specify a filename of an SSH public key' unless filename.present?
+    abort 'GitLab Geo primary node already present' if Gitlab::Geo.primary_node.present?
+
+    public_key = load_ssh_public_key(filename)
+
+    abort "Invalid SSH public key in #{filename}, aborting" unless public_key
+
+    add_primary_geo_node(public_key)
+  end
+
+  def load_ssh_public_key(filename)
+    File.open(filename).read
+  rescue => e
+    puts "Error opening #{filename}: #{e}".color(:red)
+    nil
+  end
+
+  def add_primary_geo_node(public_key)
+    params = { host: Gitlab.config.gitlab.host,
+               port: Gitlab.config.gitlab.port,
+               relative_url_root: Gitlab.config.gitlab.relative_url_root,
+               primary: true,
+               geo_node_key_attributes: { key: public_key } }
+
+    node = GeoNode.new(params)
+    puts "Saving primary GeoNode with URL #{node.url}".color(:green)
+    node.save
+
+    puts "Error saving GeoNode:\n#{node.errors.full_messages.join("\n")}".color(:red) unless node.persisted?
+  end
 end

spec/tasks/geo_rake_spec.rb

+require 'rake_helper'
+
+describe 'geo rake tasks' do
+  before do
+    Rake.application.rake_require 'tasks/geo'
+  end
+
+  describe 'add_primary_node task' do
+    let(:ssh_key) { 'ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABAQDUkxk8m9rVYZ1q4/5xpg3TwTM9QFw3TinPFkyWsiACFKjor3byV6g3vHWTuIS70E7wk2JTXGL0wdrfUG6iQDJuP0BYNxjkluB14nIAfPuXN7V73QY/cqvHogw5o6pPRFD+Szke6FzouNQ70Z/qrM1k7me3e9DMuscMMrMTOR2HLKppNQyP4Jp0WJOyncdWB2NxKXTezy/ZnHv+BdhC0q0JW3huIx9qkBCHio7x8BdyJLMF9KxNYIuCkbP3exs5wgb+qGrjSri6LfAVq8dJ2VYibWxdsUG6iITJF+G4qbcyQjgiMLbxCfNd9bjwmkxSGvFn2EPsAFKzxyAvYFWb/y91 test@host' }
+
+    before do
+      expect(Gitlab::Geo).to receive(:license_allows?).and_return(true)
+    end
+
+    it 'creates a GeoNode' do
+      begin
+        file = Tempfile.new('geo-test-')
+        file.write(ssh_key)
+        path = file.path
+        file.close
+
+        expect(GeoNode.count).to eq(0)
+
+        run_rake_task('geo:add_primary_node', path)
+
+        expect(GeoNode.count).to eq(1)
+        node = GeoNode.first
+        expect(node.primary).to be_truthy
+        expect(node.geo_node_key.key).to eq(ssh_key)
+      ensure
+        file.unlink
+      end
+    end
+  end
+end