custom parameters

Update version to 2.0.1
Add custom parameters

README.md

@@ -25,11 +25,11 @@ Or install it yourself as:
 ```ruby
 require 'ngrok/tunnel'
 
-# spawn ngrok at local port 3001
-Ngrok::Tunnel.start(3001)
+# spawn ngrok (default port 3001)
+Ngrok::Tunnel.start
 
 # ngrok local_port
-Ngrok::Tunnel.local_port
+Ngrok::Tunnel.port
 => 3001
 
 # ngrok external url
@@ -50,7 +50,7 @@ Ngrok::Tunnel.pid
 => 27384
 
 # ngrok log file descriptor
-Ngrok::Tunnel.log_file
+Ngrok::Tunnel.log
 => #<File:/tmp/ngrok20141022-27376-cmmiq4>
 
 # kill ngrok
@@ -59,6 +59,17 @@ Ngrok::Tunnel.stop
 
 ```
 
+```ruby
+# ngrok custom parameters
+Ngrok::Tunnel.start(port: 3333, 
+                    subdomain: 'MY_SUBDOMAIN', 
+                    authtoken: 'MY_TOKEN', 
+                    log: 'ngrok.log', 
+                    config: '~/.ngrok')
+
+```
+
+
 ## Contributing
 
 1. Fork it ( https://github.com/bogdanovich/ngrok-tunnel/fork )

lib/ngrok/tunnel.rb

@@ -5,29 +5,31 @@ module Ngrok
 
   class NotFound < StandardError; end
   class FetchUrlError < StandardError; end
+  class Error < StandardError; end
 
   class Tunnel
 
     class << self
-      attr_reader :pid, :local_port, :ngrok_url, :ngrok_url_https, :log_file, :status
+      attr_reader :pid, :ngrok_url, :ngrok_url_https, :status
 
-      def init
-        @status = :stopped
+      def init(params = {})
+        @params = {port: 3001, timeout: 10, config: '/dev/null'}.merge(params)
+        @status = :stopped unless @status
       end
 
-      def start(port, options = {})
-        options[:timeout] ||= 10
-        @options = options
-
+      def start(params = {})
         ensure_binary
-
-        @local_port = port.to_i
-        @log_file   = Tempfile.new('ngrok')
+        init(params)
+
+        if stopped?
+          @params[:log] = (@params[:log]) ? File.open(@params[:log], 'w+') : Tempfile.new('ngrok')
+          @pid = spawn("exec ngrok " + ngrok_exec_params)
+          at_exit { Ngrok::Tunnel.stop }      
+          @status = :running
+          fetch_urls
+        end
 
-        @pid = spawn("exec ngrok -log=stdout #{@local_port} > #{@log_file.path}")
-        at_exit { Ngrok::Tunnel.stop }      
-        @status = :running
-        fetch_urls
+        @ngrok_url
       end
 
       def stop
@@ -36,6 +38,7 @@ def stop
           @ngrok_url = @ngrok_url_https = @pid = nil
           @status = :stopped
         end
+        @status
       end
 
       def running?
@@ -46,18 +49,49 @@ def stopped?
         @status == :stopped
       end
 
+      def port
+        @params[:port]
+      end
+
+      def log
+        @params[:log]
+      end
+
+      def subdomain
+        @params[:subdomain]
+      end
+
+      def authtoken
+        @params[:authtoken]
+      end
+
       def inherited(subclass)
         init
       end
 
       private
 
+      def ngrok_exec_params
+        exec_params = "-log=stdout " 
+        exec_params << "-authtoken=#{@params[:authtoken]} " if @params[:authtoken]
+        exec_params << "-subdomain=#{@params[:subdomain]} " if @params[:subdomain]
+        exec_params << "-config=#{@params[:config]} #{@params[:port].to_i} > #{@params[:log].path}"
+      end
+
       def fetch_urls
-        @options[:timeout].times do
-          @ngrok_url, @ngrok_url_https = @log_file.read.scan(/"Url":"([^"]+)"/).flatten
+        @params[:timeout].times do
+          log_content = @params[:log].read
+          @ngrok_url, @ngrok_url_https = log_content.scan(/"Url":"([^"]+)"/).flatten
           return @ngrok_url if @ngrok_url
+
+          error = log_content.scan(/"Error":"([^"]+)"/).flatten
+          unless error.empty?
+            self.stop
+            raise Ngrok::Error, error.first
+          end
+
           sleep 1
-          @log_file.rewind
+          @params[:log].rewind
         end
         raise FetchUrlError, "Unable to fetch external url"
         @ngrok_url

lib/ngrok/tunnel/version.rb

@@ -1,5 +1,5 @@
 module Ngrok
   class Tunnel
-    VERSION = "1.0.1"
+    VERSION = "2.0.1"
   end
 end

spec/tunnel/tunnel_spec.rb

@@ -20,8 +20,7 @@
   describe "Run process" do
 
     before(:all) do
-      @local_port = 3001
-      Ngrok::Tunnel.start(3001)
+      Ngrok::Tunnel.start
     end
 
     after(:all) { Ngrok::Tunnel.stop }
@@ -39,7 +38,7 @@
     end
 
     it "should match local_port" do
-      expect(Ngrok::Tunnel.local_port).to eq(@local_port)
+      expect(Ngrok::Tunnel.port).to eq(3001)
     end
 
     it "should have valid ngrok_url" do
@@ -60,4 +59,24 @@
 
   end
 
+  describe "Custom log file" do
+    it "should be able to use custom log file" do
+      Ngrok::Tunnel.start(log: 'test.log')
+      expect(Ngrok::Tunnel.running?).to eq true
+      expect(Ngrok::Tunnel.log.path).to eq 'test.log'
+      Ngrok::Tunnel.stop
+      expect(Ngrok::Tunnel.stopped?).to eq true
+    end
+  end
+
+  describe "Custom subdomain" do
+    it "should fail without authtoken" do
+      expect {Ngrok::Tunnel.start(subdomain: 'test-subdomain')}.to raise_error
+    end
+
+    it "should fail with incorrect authtoken" do
+      expect {Ngrok::Tunnel.start(subdomain: 'test-subdomain', authtoken: 'incorrect_token')}.to raise_error
+    end
+  end
+
 end