Extract endpoints into config file

lib/cronitor.rb

@@ -64,7 +64,7 @@ def self.job(key, &block)
   end
 
   def self.monitor_api_url
-    'https://cronitor.io/api/monitors'
+    "#{Cronitor.monitor_url}/api/monitors"
   end
 end
 

lib/cronitor/config.rb

@@ -9,7 +9,7 @@ module Cronitor
   YAML_KEYS = MONITOR_TYPES.map { |t| "#{t}s" }
 
   class << self
-    attr_accessor :api_key, :api_version, :environment, :logger, :config, :timeout, :ping_timeout, :auto_discover_sidekiq
+    attr_accessor :api_key, :api_version, :environment, :logger, :config, :timeout, :ping_timeout, :auto_discover_sidekiq, :ping_url, :monitor_url
 
     def configure(&block)
       block.call(self)
@@ -23,6 +23,8 @@ def configure(&block)
   self.ping_timeout = ENV.fetch('CRONITOR_PING_TIMEOUT', nil) || 5
   self.config = ENV.fetch('CRONITOR_CONFIG', nil)
   self.auto_discover_sidekiq = ENV.fetch('CRONITOR_AUTO_DISCOVER_SIDEKIQ', 'true').casecmp('true').zero? # https://github.com/cronitorio/cronitor-sidekiq
+  self.ping_url = ENV.fetch('CRONITOR_PING_URL', 'https://cronitor.link')
+  self.monitor_url = ENV.fetch('CRONITOR_MONITOR_URL', 'https://cronitor.io')
   self.logger = Logger.new($stdout)
   logger.level = Logger::INFO
 end

lib/cronitor/monitor.rb

@@ -180,7 +180,7 @@ def unpause
     end
 
     def ping_api_url
-      "https://cronitor.link/p/#{api_key}/#{key}"
+      "#{Cronitor.ping_url}/p/#{api_key}/#{key}"
     end
 
     def fallback_ping_api_url

spec/cronitor_spec.rb

@@ -35,6 +35,40 @@
     end
   end
 
+  describe '#ping_url and #monitor_url' do
+    let(:env_vars) { ENV }
+
+    before do
+      stub_const('ENV', env_vars)
+      # We need to reset the ping_url and monitor_url since the previous spec sets them globally
+      Cronitor.configure do |cronitor|
+        cronitor.ping_url = cronitor.default_ping_url
+        cronitor.monitor_url = cronitor.default_monitor_url
+      end
+    end
+
+    context 'when custom CRONITOR_PING_URL and CRONITOR_MONITOR_URL ENV variables are set' do
+      let(:env_vars) do
+        ENV.to_hash.merge(
+          'CRONITOR_PING_URL' => 'https://ping.com',
+          'CRONITOR_MONITOR_URL' => 'https://monitor.com'
+        )
+      end
+
+      it 'returns the custom Cronitor URLs' do
+        expect(Cronitor.ping_url).to eq('https://ping.com')
+        expect(Cronitor.monitor_url).to eq('https://monitor.com')
+      end
+    end
+
+    context 'when no CRONITOR_PING_URL or CRONITOR_MONITOR_URL ENV variables are set' do
+      it 'returns the default Cronitor URLs' do
+        expect(Cronitor.ping_url).to eq('https://cronitor.link')
+        expect(Cronitor.monitor_url).to eq('https://cronitor.io')
+      end
+    end
+  end
+
   describe 'YAML configuration' do
     before(:all) do
       Cronitor.configure do |cronitor|