Convert into gem

lib/openvpn-status-web.rb

@@ -0,0 +1,96 @@
+#!/usr/bin/env ruby
+
+require 'etc'
+require 'logger'
+require 'ipaddr'
+require 'yaml'
+require 'rack'
+require 'metriks'
+
+require 'openvpn-status-web/int_patch'
+require 'openvpn-status-web/version'
+
+module OpenVPNStatusWeb
+  def self.logger
+    @logger
+  end
+
+  def self.logger=(logger)
+    @logger = logger
+  end
+
+  class LogFormatter
+    def call(lvl, time, progname, msg)
+      "[%s] %-5s %s\n" % [Time.now.strftime('%Y-%m-%d %H:%M:%S'), lvl, msg.to_s]
+    end
+  end
+
+  class Daemon
+    def initialize(name, file)
+      @name = name
+      @file = file
+    end
+
+    def call(env)
+      main_tmpl = read_template(File.join(File.dirname(__FILE__), 'openvpn-status-web/main.html.erb'))
+      # variables for template
+      name = @name
+      client_list, routing_table, global_stats = read_status_log(@file)
+      
+      html = main_tmpl.result(binding)
+      [200, {"Content-Type" => "text/html"}, [html]]
+    end
+
+    def read_template(file)
+      text = File.open(file, 'rb') do |f| f.read end
+    
+      ERB.new(text)
+    end
+  
+    def read_status_log(file)
+      text = File.open(file, 'rb') do |f| f.read end
+
+      current_section = :none
+      client_list = []
+      routing_table = []
+      global_stats = []
+
+      text.lines.each do |line|
+        (current_section = :cl; next) if line == "OpenVPN CLIENT LIST\n"
+        (current_section = :rt; next) if line == "ROUTING TABLE\n"
+        (current_section = :gs; next) if line == "GLOBAL STATS\n"
+        (current_section = :end; next) if line == "END\n"
+    
+        case current_section
+        when :cl then client_list << line.strip.split(',')
+        when :rt then routing_table << line.strip.split(',')
+        when :gs then global_stats << line.strip.split(',')
+        end
+      end
+
+      [client_list[2..-1], routing_table[1..-1], global_stats]
+    end
+
+    def self.run!
+      if ARGV.length != 4
+        puts "Usage: openvpn-status-web vpn-name status-log listen-host listen-port"
+        exit 1
+      end
+
+      OpenVPNStatusWeb.logger = Logger.new(STDOUT)
+
+      OpenVPNStatusWeb.logger.progname = "openvpn-status-web"
+      OpenVPNStatusWeb.logger.formatter = LogFormatter.new
+
+      OpenVPNStatusWeb.logger.info "Starting..."
+
+      Signal.trap('INT') do
+        OpenVPNStatusWeb.logger.info "Quitting..."
+        Rack::Handler::WEBrick.shutdown
+      end
+
+      app = Daemon.new(ARGV[0], ARGV[1])
+      Rack::Handler::WEBrick.run app, :Host => ARGV[2], :Port => ARGV[3]
+    end
+  end
+end