diff options
| -rwxr-xr-x | lognotify.rb | 26 |
1 files changed, 17 insertions, 9 deletions
diff --git a/lognotify.rb b/lognotify.rb index f789d08..dc99119 100755 --- a/lognotify.rb +++ b/lognotify.rb @@ -25,15 +25,25 @@ require 'ftools' +# Global settings. CACHE_DIR="~/.cache/lognotify" -CONFIG_DIR="~/.config/lognotify" +CONF_DIR="~/.config/lognotify" + +# Methods for quickly getting file paths. +class String + def to_cache_path + return File.expand_path(CACHE_DIR + '/' + self + '.log') + end + + def to_conf_path + return File.expand_path(CONF_DIR + '/' + self + '.conf') + end +end # Configuration file parser. def parse identifier conf = Hash.new - path = File.expand_path(CONFIG_DIR + '/' + identifier + '.conf') - - File.open(path) do |file| + File.open(identifier.to_conf_path) do |file| file.each_line do |line| # Remove whitespace from beginning of line, allowing for indentation. line.lstrip! @@ -55,8 +65,7 @@ end # Count lines in cached log. def count_lines identifier - path = File.expand_path(CACHE_DIR + '/' + identifier + '.log') - return File.open(path).readlines.length + return File.open(identifier.to_cache_path).readlines.length end # Retrieve new lines via SSH. @@ -69,8 +78,7 @@ end # Append new lines to cached log. def append_lines identifier, lines - path = File.expand_path(CACHE_DIR + '/' + identifier + '.log') - file = File.open(path, 'a') + file = File.open(identifier.to_cache_path, 'a') file.print lines file.close @@ -88,7 +96,7 @@ ARGV.each do |identifier| conf = parse(identifier) # Create cache file, if nonexistent. - path = File.expand_path(CACHE_DIR + '/' + identifier + '.log') + path = identifier.to_cache_path File.open(path, 'w').close unless File.exist?(path) print '* Counting lines in cached log... ' |