#!/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