platform_sh 0.2.3 → 0.2.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 3d7dc484d0414e3a3d3033ca3a5435910932c4c9
-  data.tar.gz: 5e234ccea2bc43683dbefe4ec7ee60547116c745
+  metadata.gz: 557c7b76ab879a1d8ed7683b02b6e5440b7003ba
+  data.tar.gz: 73f9f29e020929824a6a1e8140532dd27e844dd8
 SHA512:
-  metadata.gz: 51c0fa7f383216fd004bc70287936e8f857b3f4d1e52e4fc59f303a3e668371155bc909d4ca2c611c9d82e031e2d8ec968feb3024b1a91b80ec298c1ef0d0139
-  data.tar.gz: 4f4ff7373bb4c045817222e20decddb343bf8534bea8feadadca7f45dafc684ae8a5fbd9621587fde8834d979d7a0d0ba673cdc648bd5d892ae89d6e216111ab
+  metadata.gz: cc9631b086ae4b5e76ae843e09193b23f65c49b030d299d1ccb6ca2adad718988b5313deb80fcbb40ed97a9ee402195d9a608307a27849b509890b23a88fb9e5
+  data.tar.gz: e2b1b759d75c824c93f0c3b51ea09f50be34f2cc919ed03ea21748f66a9ba91fcb8bad2f38d597f3387838b4e4be68366689c6394bcdb08a5b64e2391035c4e5

data/lib/platform_sh.rb

@@ -55,29 +55,78 @@ class PlatformSH
     JSON.parse(File.read('/run/config.json'))
   end
   
+  
   #Tries to guess relational database url
   def self.guess_database_url
-    databases = PlatformSH::config["relationships"].select {|k,v| v[0]["scheme"]=="mysql" || v[0]["scheme"]=="pgsql"}
-    case databases.length
-      when 0
-        $stderr.puts "Could not find a relational database"
-        return nil
-      when 1
-        database = databases.first[1][0]
-        #This is needed to choose between mysql and mysql2 gems, magically.
-        case database['scheme']
-          when "mysql"
-            scheme = "mysql2"
-          when "pgsql"
-            scheme = "postgresql"
-          else
-            scheme = database['scheme']
-        end
-        database_url = "#{scheme}://#{database['username']}:#{database['password']}@#{database['host']}:#{database['port']}/#{database['path']}"
-        return database_url
-      else
-        $stderr.puts "More than one database, giving up, set configuration by hand"
+    postgresql_url = self.guess_postgresql_url
+    mysql_url = self.guess_mysql_url
+    if !mysql_url.nil? && !postgresql_url.nil?
+      $stderr.puts "More than one relational database, giving up, set configuration by hand"
+      return nil
+    end
+    if (mysql_url.nil? && postgresql_url.nil?)
+      $stderr.puts "Could not find a relational database"
+      return nil
     end
+    return mysql_url || postgresql_url
   end
 
-end
+  def self.guess_url(service_type, platform_scheme, url_template)
+      services = PlatformSH::config["relationships"].select {|k,v| v[0]["scheme"]==platform_scheme}
+      case services.length
+        when 0
+          $stderr.puts "Could not find an #{service_type}"
+          return nil
+        when 1
+          service = services.first[1][0]
+          service =  service.each_with_object({}){|(k,v), h| h[k.to_sym] = v} #keys need to be symbols
+          return url_template % service
+        else
+          $stderr.puts "More than one #{service_type}, giving up, set configuration by hand"
+      end
+  end
+  
+  def self.guess_elasticsearch_url
+    self.guess_url("elasticsearch", "http", "http://%{host}:%{port}")
+  end
+  
+  def self.guess_redis_url
+    self.guess_url("redis", "redis", "redis://%{host}:%{port}")
+  end
+  
+  def self.guess_mongodb_url
+    self.guess_url("mongodb", "mongodb", "mongodb://%{username}:%{password}@%{host}:%{port}/%{path}")
+  end
+  
+  def self.guess_solr_url
+    self.guess_url("solr", "solr", "http://%{host}:%{port}/%{path}")
+  end
+  
+  def self.guess_mysql_url
+    #fallback to mysql url if mysql2 gem is not loaded
+    if Gem::Specification::find_all_by_name("mysql2").empty?
+      template = "mysql://%{username}:%{password}@%{host}:%{port}/%{path}"
+    else
+      template = "mysql2://%{username}:%{password}@%{host}:%{port}/%{path}"
+    end
+    self.guess_url("mysql", "mysql",template)
+  end
+  
+  def self.guess_rabbitmq_url
+    self.guess_url("rabbitmq", "amqp","amqp://%{username}:%{password}@%{host}:%{port}")
+  end
+  
+  def self.guess_postgresql_url
+    self.guess_url("postgresql", "pgsql","postgresql://%{username}:%{password}@%{host}:%{port}")
+  end
+  
+  def self.export_services_urls
+    ENV['DATABASE_URL']=PlatformSH::guess_database_url
+    ENV['MONGODB_URL']=PlatformSH::guess_mongodb_url
+    ENV['REDIS_URL']=PlatformSH::guess_redis_url
+    ENV['ELASTICSEARCH_URL']=PlatformSH::guess_redis_url
+    ENV['RABBITMQ_URL']=PlatformSH::guess_rabbitmq_url
+    ENV['SOLR_URL']=PlatformSH::guess_solr_url
+  end
+  
+end

data/lib/platform_sh/version.rb

@@ -1,3 +1,3 @@
 class PlatformSH
-  VERSION = "0.2.3"
+  VERSION = "0.2.4"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: platform_sh
 version: !ruby/object:Gem::Version
-  version: 0.2.3
+  version: 0.2.4
 platform: ruby
 authors:
 - Ori Pekelman
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2016-12-07 00:00:00.000000000 Z
+date: 2016-12-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler