mirror of
https://github.com/cmur2/openvpn-status-web.git
synced 2025-06-29 00:30:24 +02:00
Convert into gem
This commit is contained in:
cn
96
lib/openvpn-status-web.rb
Normal file
96
lib/openvpn-status-web.rb
Normal file
@ -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
|
||||
16
lib/openvpn-status-web/int_patch.rb
Normal file
16
lib/openvpn-status-web/int_patch.rb
Normal file
@ -0,0 +1,16 @@
|
||||
|
||||
class Integer
|
||||
def as_bytes
|
||||
return "1 Byte" if self == 1
|
||||
|
||||
label = ["Bytes", "KiB", "MiB", "GiB", "TiB"]
|
||||
i = 0
|
||||
num = self.to_f
|
||||
while num >= 1024 do
|
||||
num = num / 1024
|
||||
i += 1
|
||||
end
|
||||
|
||||
"#{format('%.2f', num)} #{label[i]}"
|
||||
end
|
||||
end
|
||||
108
lib/openvpn-status-web/main.html.erb
Normal file
108
lib/openvpn-status-web/main.html.erb
Normal file
@ -0,0 +1,108 @@
|
||||
<!DOCTYPE html>
|
||||
<html lang="en">
|
||||
<head>
|
||||
<meta charset="utf-8">
|
||||
<meta http-equiv="refresh" content="60">
|
||||
<title>OpenVPN Status</title>
|
||||
<style type="text/css">
|
||||
table {
|
||||
border-collapse: collapse;
|
||||
}
|
||||
h1 {
|
||||
font-size: 24px;
|
||||
}
|
||||
h2 {
|
||||
font-size: 16px;
|
||||
font-weight: normal;
|
||||
margin-top: 20px;
|
||||
margin-bottom: 10px;
|
||||
}
|
||||
div {
|
||||
margin-left: 15px;
|
||||
}
|
||||
td {
|
||||
margin: 0px;
|
||||
}
|
||||
td.first {
|
||||
padding: 0px 6px 0px 0px;
|
||||
}
|
||||
td.middle {
|
||||
padding: 0px 6px 0px 6px;
|
||||
border-left: 1px solid #ccc;
|
||||
border-right: 1px solid #ccc;
|
||||
}
|
||||
td.last {
|
||||
padding: 0px 0px 0px 6px;
|
||||
}
|
||||
thead {
|
||||
font-weight: bold;
|
||||
border-bottom: 1px solid #ccc;
|
||||
}
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
|
||||
<h1>OpenVPN Status for <%= name %></h1>
|
||||
|
||||
<h2>Client List</h2>
|
||||
<div>
|
||||
<table>
|
||||
<thead>
|
||||
<td class="first">Common Name</td>
|
||||
<td class="middle">Real Address</td>
|
||||
<td class="middle">Data Received</td>
|
||||
<td class="middle">Data Sent</td>
|
||||
<td class="last">Connected Since</td>
|
||||
</thead>
|
||||
<tbody>
|
||||
<% client_list.each do |client| %>
|
||||
<tr>
|
||||
<td class="first"><%= client[0] %></td>
|
||||
<td class="middle"><%= client[1] %></td>
|
||||
<td class="middle"><%= client[2].to_i.as_bytes %></td>
|
||||
<td class="middle"><%= client[3].to_i.as_bytes %></td>
|
||||
<td class="last"><%= client[4] %></td>
|
||||
</tr>
|
||||
<% end %>
|
||||
</tbody>
|
||||
</table>
|
||||
</div>
|
||||
|
||||
<h2>Routing Table</h2>
|
||||
<div>
|
||||
<table>
|
||||
<thead>
|
||||
<td class="first">Virtual Address</td>
|
||||
<td class="middle">Common Name</td>
|
||||
<td class="middle">Real Address</td>
|
||||
<td class="last">Last Ref</td>
|
||||
</thead>
|
||||
<tbody>
|
||||
<% routing_table.each do |e| %>
|
||||
<tr>
|
||||
<td class="first"><%= e[0] %></td>
|
||||
<td class="middle"><%= e[1] %></td>
|
||||
<td class="middle"><%= e[2] %></td>
|
||||
<td class="last"><%= e[3] %></td>
|
||||
</tr>
|
||||
<% end %>
|
||||
</tbody>
|
||||
</table>
|
||||
</div>
|
||||
|
||||
<h2>Global Stats</h2>
|
||||
<div>
|
||||
<table>
|
||||
<tbody>
|
||||
<% global_stats.each do |e| %>
|
||||
<tr>
|
||||
<td><%= e[0] %>:</td>
|
||||
<td><%= e[1] %></td>
|
||||
</tr>
|
||||
<% end %>
|
||||
</tbody>
|
||||
</table>
|
||||
</div>
|
||||
|
||||
</body>
|
||||
</html>
|
||||
4
lib/openvpn-status-web/version.rb
Normal file
4
lib/openvpn-status-web/version.rb
Normal file
@ -0,0 +1,4 @@
|
||||
|
||||
module OpenVPNStatusWeb
|
||||
VERSION = "0.0.1"
|
||||
end
|
||||
Reference in New Issue
Block a user