kmts 2.0.0

Sign up to get free protection for your applications and to get access to all the features.
Files changed (72) hide show
  1. data/.gitignore +5 -0
  2. data/CHANGELOG +4 -0
  3. data/Gemfile +4 -0
  4. data/Gemfile.lock +29 -0
  5. data/LICENSE +202 -0
  6. data/README.md +64 -0
  7. data/README.rdoc +14 -0
  8. data/Rakefile +5 -0
  9. data/bin/km_send +35 -0
  10. data/doc/Accept.html +546 -0
  11. data/doc/Gemfile.html +110 -0
  12. data/doc/Hash.html +283 -0
  13. data/doc/Helper.html +318 -0
  14. data/doc/KMError.html +159 -0
  15. data/doc/KMTS.html +493 -0
  16. data/doc/KMTS/IdentError.html +159 -0
  17. data/doc/KMTS/InitError.html +159 -0
  18. data/doc/KMTS/SaaS.html +451 -0
  19. data/doc/Object.html +211 -0
  20. data/doc/README_rdoc.html +122 -0
  21. data/doc/Rakefile.html +111 -0
  22. data/doc/String.html +244 -0
  23. data/doc/bin/km_send.html +54 -0
  24. data/doc/created.rid +15 -0
  25. data/doc/images/brick.png +0 -0
  26. data/doc/images/brick_link.png +0 -0
  27. data/doc/images/bug.png +0 -0
  28. data/doc/images/bullet_black.png +0 -0
  29. data/doc/images/bullet_toggle_minus.png +0 -0
  30. data/doc/images/bullet_toggle_plus.png +0 -0
  31. data/doc/images/date.png +0 -0
  32. data/doc/images/find.png +0 -0
  33. data/doc/images/loadingAnimation.gif +0 -0
  34. data/doc/images/macFFBgHack.png +0 -0
  35. data/doc/images/package.png +0 -0
  36. data/doc/images/page_green.png +0 -0
  37. data/doc/images/page_white_text.png +0 -0
  38. data/doc/images/page_white_width.png +0 -0
  39. data/doc/images/plugin.png +0 -0
  40. data/doc/images/ruby.png +0 -0
  41. data/doc/images/tag_green.png +0 -0
  42. data/doc/images/wrench.png +0 -0
  43. data/doc/images/wrench_orange.png +0 -0
  44. data/doc/images/zoom.png +0 -0
  45. data/doc/index.html +142 -0
  46. data/doc/js/darkfish.js +116 -0
  47. data/doc/js/jquery.js +32 -0
  48. data/doc/js/quicksearch.js +114 -0
  49. data/doc/js/thickbox-compressed.js +10 -0
  50. data/doc/lib/km/saas_rb.html +54 -0
  51. data/doc/lib/km/version_rb.html +52 -0
  52. data/doc/lib/km_rb.html +60 -0
  53. data/doc/rdoc.css +730 -0
  54. data/doc/spec/accept_rb.html +62 -0
  55. data/doc/spec/km_old_rb.html +54 -0
  56. data/doc/spec/km_saas_spec_rb.html +56 -0
  57. data/doc/spec/km_send_spec_rb.html +54 -0
  58. data/doc/spec/km_spec_rb.html +54 -0
  59. data/doc/spec/setup_rb.html +60 -0
  60. data/doc/spec/watchr_rb.html +52 -0
  61. data/kmts.gemspec +26 -0
  62. data/lib/kmts.rb +245 -0
  63. data/lib/kmts/saas.rb +39 -0
  64. data/lib/kmts/version.rb +3 -0
  65. data/spec/accept.rb +91 -0
  66. data/spec/km_old.rb +105 -0
  67. data/spec/km_saas_spec.rb +107 -0
  68. data/spec/km_send_spec.rb +98 -0
  69. data/spec/km_spec.rb +175 -0
  70. data/spec/setup.rb +77 -0
  71. data/spec/watchr.rb +3 -0
  72. metadata +190 -0
data/doc/spec/accept_rb.html ADDED
@@ -0,0 +1,62 @@
1
+ <?xml version="1.0" encoding="utf-8"?>
2
+ <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN"
3
+ "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
4
+
5
+ <html xmlns="http://www.w3.org/1999/xhtml" xml:lang="en" lang="en">
6
+ <head>
7
+ <meta content="text/html; charset=UTF-8" http-equiv="Content-Type" />
8
+
9
+ <title>File: accept.rb [RDoc Documentation]</title>
10
+
11
+ <link type="text/css" media="screen" href="../rdoc.css" rel="stylesheet" />
12
+
13
+ <script src="../js/jquery.js" type="text/javascript"
14
+ charset="utf-8"></script>
15
+ <script src="../js/thickbox-compressed.js" type="text/javascript"
16
+ charset="utf-8"></script>
17
+ <script src="../js/quicksearch.js" type="text/javascript"
18
+ charset="utf-8"></script>
19
+ <script src="../js/darkfish.js" type="text/javascript"
20
+ charset="utf-8"></script>
21
+ </head>
22
+
23
+ <body class="file file-popup">
24
+ <div id="metadata">
25
+ <dl>
26
+ <dt class="modified-date">Last Modified</dt>
27
+ <dd class="modified-date">2011-02-04 09:44:35 -0800</dd>
28
+
29
+
30
+ <dt class="requires">Requires</dt>
31
+ <dd class="requires">
32
+ <ul>
33
+
34
+ <li>socket</li>
35
+
36
+ <li>rubygems</li>
37
+
38
+ <li>json</li>
39
+
40
+ <li>uri</li>
41
+
42
+ <li>cgi</li>
43
+
44
+ </ul>
45
+ </dd>
46
+
47
+
48
+
49
+ </dl>
50
+ </div>
51
+
52
+ <div id="documentation">
53
+
54
+ <div class="description">
55
+ <h2>Description</h2>
56
+
57
+ </div>
58
+
59
+ </div>
60
+ </body>
61
+ </html>
62
+
data/doc/spec/km_old_rb.html ADDED
@@ -0,0 +1,54 @@
1
+ <?xml version="1.0" encoding="utf-8"?>
2
+ <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN"
3
+ "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
4
+
5
+ <html xmlns="http://www.w3.org/1999/xhtml" xml:lang="en" lang="en">
6
+ <head>
7
+ <meta content="text/html; charset=UTF-8" http-equiv="Content-Type" />
8
+
9
+ <title>File: km_old.rb [RDoc Documentation]</title>
10
+
11
+ <link type="text/css" media="screen" href="../rdoc.css" rel="stylesheet" />
12
+
13
+ <script src="../js/jquery.js" type="text/javascript"
14
+ charset="utf-8"></script>
15
+ <script src="../js/thickbox-compressed.js" type="text/javascript"
16
+ charset="utf-8"></script>
17
+ <script src="../js/quicksearch.js" type="text/javascript"
18
+ charset="utf-8"></script>
19
+ <script src="../js/darkfish.js" type="text/javascript"
20
+ charset="utf-8"></script>
21
+ </head>
22
+
23
+ <body class="file file-popup">
24
+ <div id="metadata">
25
+ <dl>
26
+ <dt class="modified-date">Last Modified</dt>
27
+ <dd class="modified-date">2011-02-04 09:43:26 -0800</dd>
28
+
29
+
30
+ <dt class="requires">Requires</dt>
31
+ <dd class="requires">
32
+ <ul>
33
+
34
+ <li>setup</li>
35
+
36
+ </ul>
37
+ </dd>
38
+
39
+
40
+
41
+ </dl>
42
+ </div>
43
+
44
+ <div id="documentation">
45
+
46
+ <div class="description">
47
+ <h2>Description</h2>
48
+
49
+ </div>
50
+
51
+ </div>
52
+ </body>
53
+ </html>
54
+
data/doc/spec/km_saas_spec_rb.html ADDED
@@ -0,0 +1,56 @@
1
+ <?xml version="1.0" encoding="utf-8"?>
2
+ <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN"
3
+ "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
4
+
5
+ <html xmlns="http://www.w3.org/1999/xhtml" xml:lang="en" lang="en">
6
+ <head>
7
+ <meta content="text/html; charset=UTF-8" http-equiv="Content-Type" />
8
+
9
+ <title>File: km_saas_spec.rb [RDoc Documentation]</title>
10
+
11
+ <link type="text/css" media="screen" href="../rdoc.css" rel="stylesheet" />
12
+
13
+ <script src="../js/jquery.js" type="text/javascript"
14
+ charset="utf-8"></script>
15
+ <script src="../js/thickbox-compressed.js" type="text/javascript"
16
+ charset="utf-8"></script>
17
+ <script src="../js/quicksearch.js" type="text/javascript"
18
+ charset="utf-8"></script>
19
+ <script src="../js/darkfish.js" type="text/javascript"
20
+ charset="utf-8"></script>
21
+ </head>
22
+
23
+ <body class="file file-popup">
24
+ <div id="metadata">
25
+ <dl>
26
+ <dt class="modified-date">Last Modified</dt>
27
+ <dd class="modified-date">2011-02-04 12:02:23 -0800</dd>
28
+
29
+
30
+ <dt class="requires">Requires</dt>
31
+ <dd class="requires">
32
+ <ul>
33
+
34
+ <li>setup</li>
35
+
36
+ <li>km/saas</li>
37
+
38
+ </ul>
39
+ </dd>
40
+
41
+
42
+
43
+ </dl>
44
+ </div>
45
+
46
+ <div id="documentation">
47
+
48
+ <div class="description">
49
+ <h2>Description</h2>
50
+
51
+ </div>
52
+
53
+ </div>
54
+ </body>
55
+ </html>
56
+
data/doc/spec/km_send_spec_rb.html ADDED
@@ -0,0 +1,54 @@
1
+ <?xml version="1.0" encoding="utf-8"?>
2
+ <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN"
3
+ "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
4
+
5
+ <html xmlns="http://www.w3.org/1999/xhtml" xml:lang="en" lang="en">
6
+ <head>
7
+ <meta content="text/html; charset=UTF-8" http-equiv="Content-Type" />
8
+
9
+ <title>File: km_send_spec.rb [RDoc Documentation]</title>
10
+
11
+ <link type="text/css" media="screen" href="../rdoc.css" rel="stylesheet" />
12
+
13
+ <script src="../js/jquery.js" type="text/javascript"
14
+ charset="utf-8"></script>
15
+ <script src="../js/thickbox-compressed.js" type="text/javascript"
16
+ charset="utf-8"></script>
17
+ <script src="../js/quicksearch.js" type="text/javascript"
18
+ charset="utf-8"></script>
19
+ <script src="../js/darkfish.js" type="text/javascript"
20
+ charset="utf-8"></script>
21
+ </head>
22
+
23
+ <body class="file file-popup">
24
+ <div id="metadata">
25
+ <dl>
26
+ <dt class="modified-date">Last Modified</dt>
27
+ <dd class="modified-date">2011-02-04 11:57:15 -0800</dd>
28
+
29
+
30
+ <dt class="requires">Requires</dt>
31
+ <dd class="requires">
32
+ <ul>
33
+
34
+ <li>setup</li>
35
+
36
+ </ul>
37
+ </dd>
38
+
39
+
40
+
41
+ </dl>
42
+ </div>
43
+
44
+ <div id="documentation">
45
+
46
+ <div class="description">
47
+ <h2>Description</h2>
48
+
49
+ </div>
50
+
51
+ </div>
52
+ </body>
53
+ </html>
54
+
data/doc/spec/km_spec_rb.html ADDED
@@ -0,0 +1,54 @@
1
+ <?xml version="1.0" encoding="utf-8"?>
2
+ <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN"
3
+ "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
4
+
5
+ <html xmlns="http://www.w3.org/1999/xhtml" xml:lang="en" lang="en">
6
+ <head>
7
+ <meta content="text/html; charset=UTF-8" http-equiv="Content-Type" />
8
+
9
+ <title>File: km_spec.rb [RDoc Documentation]</title>
10
+
11
+ <link type="text/css" media="screen" href="../rdoc.css" rel="stylesheet" />
12
+
13
+ <script src="../js/jquery.js" type="text/javascript"
14
+ charset="utf-8"></script>
15
+ <script src="../js/thickbox-compressed.js" type="text/javascript"
16
+ charset="utf-8"></script>
17
+ <script src="../js/quicksearch.js" type="text/javascript"
18
+ charset="utf-8"></script>
19
+ <script src="../js/darkfish.js" type="text/javascript"
20
+ charset="utf-8"></script>
21
+ </head>
22
+
23
+ <body class="file file-popup">
24
+ <div id="metadata">
25
+ <dl>
26
+ <dt class="modified-date">Last Modified</dt>
27
+ <dd class="modified-date">2011-02-04 11:12:57 -0800</dd>
28
+
29
+
30
+ <dt class="requires">Requires</dt>
31
+ <dd class="requires">
32
+ <ul>
33
+
34
+ <li>setup</li>
35
+
36
+ </ul>
37
+ </dd>
38
+
39
+
40
+
41
+ </dl>
42
+ </div>
43
+
44
+ <div id="documentation">
45
+
46
+ <div class="description">
47
+ <h2>Description</h2>
48
+
49
+ </div>
50
+
51
+ </div>
52
+ </body>
53
+ </html>
54
+
data/doc/spec/setup_rb.html ADDED
@@ -0,0 +1,60 @@
1
+ <?xml version="1.0" encoding="utf-8"?>
2
+ <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN"
3
+ "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
4
+
5
+ <html xmlns="http://www.w3.org/1999/xhtml" xml:lang="en" lang="en">
6
+ <head>
7
+ <meta content="text/html; charset=UTF-8" http-equiv="Content-Type" />
8
+
9
+ <title>File: setup.rb [RDoc Documentation]</title>
10
+
11
+ <link type="text/css" media="screen" href="../rdoc.css" rel="stylesheet" />
12
+
13
+ <script src="../js/jquery.js" type="text/javascript"
14
+ charset="utf-8"></script>
15
+ <script src="../js/thickbox-compressed.js" type="text/javascript"
16
+ charset="utf-8"></script>
17
+ <script src="../js/quicksearch.js" type="text/javascript"
18
+ charset="utf-8"></script>
19
+ <script src="../js/darkfish.js" type="text/javascript"
20
+ charset="utf-8"></script>
21
+ </head>
22
+
23
+ <body class="file file-popup">
24
+ <div id="metadata">
25
+ <dl>
26
+ <dt class="modified-date">Last Modified</dt>
27
+ <dd class="modified-date">2011-02-04 09:59:12 -0800</dd>
28
+
29
+
30
+ <dt class="requires">Requires</dt>
31
+ <dd class="requires">
32
+ <ul>
33
+
34
+ <li>ap</li>
35
+
36
+ <li>km</li>
37
+
38
+ <li>fileutils</li>
39
+
40
+ <li>accept</li>
41
+
42
+ </ul>
43
+ </dd>
44
+
45
+
46
+
47
+ </dl>
48
+ </div>
49
+
50
+ <div id="documentation">
51
+
52
+ <div class="description">
53
+ <h2>Description</h2>
54
+
55
+ </div>
56
+
57
+ </div>
58
+ </body>
59
+ </html>
60
+
data/doc/spec/watchr_rb.html ADDED
@@ -0,0 +1,52 @@
1
+ <?xml version="1.0" encoding="utf-8"?>
2
+ <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN"
3
+ "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
4
+
5
+ <html xmlns="http://www.w3.org/1999/xhtml" xml:lang="en" lang="en">
6
+ <head>
7
+ <meta content="text/html; charset=UTF-8" http-equiv="Content-Type" />
8
+
9
+ <title>File: watchr.rb [RDoc Documentation]</title>
10
+
11
+ <link type="text/css" media="screen" href="../rdoc.css" rel="stylesheet" />
12
+
13
+ <script src="../js/jquery.js" type="text/javascript"
14
+ charset="utf-8"></script>
15
+ <script src="../js/thickbox-compressed.js" type="text/javascript"
16
+ charset="utf-8"></script>
17
+ <script src="../js/quicksearch.js" type="text/javascript"
18
+ charset="utf-8"></script>
19
+ <script src="../js/darkfish.js" type="text/javascript"
20
+ charset="utf-8"></script>
21
+ </head>
22
+
23
+ <body class="file file-popup">
24
+ <div id="metadata">
25
+ <dl>
26
+ <dt class="modified-date">Last Modified</dt>
27
+ <dd class="modified-date">2011-02-04 16:43:58 -0800</dd>
28
+
29
+
30
+ <dt class="requires">Requires</dt>
31
+ <dd class="requires">
32
+ <ul>
33
+
34
+ </ul>
35
+ </dd>
36
+
37
+
38
+
39
+ </dl>
40
+ </div>
41
+
42
+ <div id="documentation">
43
+
44
+ <div class="description">
45
+ <h2>Description</h2>
46
+
47
+ </div>
48
+
49
+ </div>
50
+ </body>
51
+ </html>
52
+
data/kmts.gemspec ADDED
@@ -0,0 +1,26 @@
1
+ # -*- encoding: utf-8 -*-
2
+ require File.expand_path("../lib/kmts/version", __FILE__)
3
+
4
+ Gem::Specification.new do |s|
5
+ s.name = "kmts"
6
+ s.version = KMTS::VERSION
7
+ s.platform = Gem::Platform::RUBY
8
+ s.authors = ["KISSmetrics"]
9
+ s.email = ["support@kissmetrics.com"]
10
+ s.homepage = "https://github.com/kissmetrics/kmts"
11
+ s.summary = "KISSmetrics threadsafe ruby API gem"
12
+ s.description = "KISSmetrics threadsafe ruby API gem"
13
+
14
+ s.required_rubygems_version = ">= 1.3.6"
15
+ s.rubyforge_project = "kissmetrics"
16
+
17
+ s.add_development_dependency "bundler", ">= 1.0.0"
18
+ s.add_development_dependency "rspec", "~> 2.4.0"
19
+ s.add_development_dependency "rake"
20
+ s.add_development_dependency "json"
21
+
22
+ s.files = `git ls-files`.split("\n")
23
+ s.test_files = `git ls-files -- {test,spec,features}/*`.split("\n")
24
+ s.executables = `git ls-files`.split("\n").map{|f| f =~ /^bin\/(.*)/ ? $1 : nil}.compact
25
+ s.require_path = 'lib'
26
+ end
data/lib/kmts.rb ADDED
@@ -0,0 +1,245 @@
1
+ require 'uri'
2
+ require 'socket'
3
+ require 'net/http'
4
+ require 'fileutils'
5
+ require 'kmts/saas'
6
+
7
+ class KMError < StandardError; end
8
+
9
+ class KMTS
10
+ @key = nil
11
+ @logs = {}
12
+ @host = 'trk.kissmetrics.com:80'
13
+ @log_dir = '/tmp'
14
+ @to_stderr = true
15
+ @use_cron = false
16
+ @dryrun = false
17
+
18
+ class << self
19
+ class IdentError < StandardError; end
20
+ class InitError < StandardError; end
21
+
22
+ def init(key, options={})
23
+ default = {
24
+ :host => @host,
25
+ :log_dir => @log_dir,
26
+ :to_stderr => @to_stderr,
27
+ :use_cron => @use_cron,
28
+ :dryrun => @dryrun,
29
+ :env => set_env,
30
+ }
31
+ options = default.merge(options)
32
+
33
+ begin
34
+ @key = key
35
+ @host = options[:host]
36
+ @log_dir = options[:log_dir]
37
+ @use_cron = options[:use_cron]
38
+ @to_stderr = options[:to_stderr]
39
+ @dryrun = options[:dryrun]
40
+ @env = options[:env]
41
+ log_dir_writable?
42
+ rescue Exception => e
43
+ log_error(e)
44
+ end
45
+ end
46
+
47
+ def set_env
48
+ @env = Rails.env if defined? Rails
49
+ @env ||= ENV['RACK_ENV']
50
+ @env ||= 'production'
51
+ end
52
+
53
+ def record(id, action, props={})
54
+ props = hash_keys_to_str(props)
55
+ begin
56
+ return unless is_initialized?
57
+ return set(id, action) if action.class == Hash
58
+
59
+ props.update('_n' => action)
60
+ generate_query('e', props, id)
61
+ rescue Exception => e
62
+ log_error(e)
63
+ end
64
+ end
65
+
66
+ def alias(name, alias_to)
67
+ begin
68
+ return unless is_initialized?
69
+ generate_query('a', { '_n' => alias_to, '_p' => name }, false)
70
+ rescue Exception => e
71
+ log_error(e)
72
+ end
73
+ end
74
+
75
+ def set(id, data)
76
+ begin
77
+ return unless is_initialized?
78
+ generate_query('s', data, id)
79
+ rescue Exception => e
80
+ log_error(e)
81
+ end
82
+ end
83
+
84
+ def send_logged_queries # :nodoc:
85
+ line = nil
86
+ begin
87
+ query_log = log_name(:query_old)
88
+ query_log = log_name(:query) unless File.exists?(query_log)
89
+ return unless File.exists?(query_log) # can't find logfile to send
90
+ FileUtils.move(query_log, log_name(:send))
91
+ File.open(log_name(:send)) do |fh|
92
+ while not fh.eof?
93
+ begin
94
+ line = fh.readline.chomp
95
+ send_query(line)
96
+ rescue Exception => e
97
+ log_query(line) if line
98
+ log_error(e)
99
+ end
100
+ end
101
+ end
102
+ FileUtils.rm(log_name(:send))
103
+ rescue Exception => e
104
+ log_error(e)
105
+ end
106
+ end
107
+
108
+ def log_dir
109
+ @log_dir
110
+ end
111
+ def host
112
+ @host
113
+ end
114
+
115
+ # :stopdoc:
116
+ protected
117
+ def hash_keys_to_str(hash)
118
+ Hash[*hash.map { |k,v| k.class == Symbol ? [k.to_s,v] : [k,v] }.flatten] # convert all keys to strings
119
+ end
120
+ def reset
121
+ @id = nil
122
+ @key = nil
123
+ @logs = {}
124
+ @host = 'trk.kissmetrics.com:80'
125
+ @log_dir = '/tmp'
126
+ @to_stderr = true
127
+ @use_cron = false
128
+ @env = nil
129
+ @force = false
130
+ end
131
+
132
+ def log_name(type)
133
+ return @logs[type] if @logs[type]
134
+ fname = ''
135
+ env = @env ? "_#{@env}" : ''
136
+ case type
137
+ when :error
138
+ fname = "kissmetrics#{env}_error.log"
139
+ when :query
140
+ fname = "kissmetrics#{env}_query.log"
141
+ when :query_old # backwards compatibility
142
+ fname = "kissmetrics_query.log"
143
+ when :sent
144
+ fname = "kissmetrics#{env}_sent.log"
145
+ when :send
146
+ fname = Time.now.to_i.to_s + "kissmetrics_#{env}_sending.log"
147
+ end
148
+ @logs[type] = File.join(@log_dir,fname)
149
+ end
150
+
151
+ def log_query(msg)
152
+ log(:query,msg)
153
+ end
154
+
155
+ def log_sent(msg)
156
+ log(:sent,msg)
157
+ end
158
+
159
+ def log_send(msg)
160
+ log(:send,msg)
161
+ end
162
+
163
+ def log_error(error)
164
+ if defined?(HoptoadNotifier)
165
+ HoptoadNotifier.notify_or_ignore(KMError.new(error))
166
+ end
167
+ msg = Time.now.strftime("<%c> ") + error.message
168
+ $stderr.puts msg if @to_stderr
169
+ log(:error, msg)
170
+ rescue Exception # rescue incase hoptoad has issues
171
+ end
172
+
173
+ def log(type,msg)
174
+ begin
175
+ File.open(log_name(type), 'a') do |fh|
176
+ fh.flock(File::LOCK_EX)
177
+ fh.puts(msg)
178
+ end
179
+ rescue Exception => e
180
+ raise KMError.new(e) if type.to_s == 'query'
181
+ # just discard at this point otherwise
182
+ end
183
+ end
184
+
185
+
186
+ def generate_query(type, data, id = nil)
187
+ data = hash_keys_to_str(data)
188
+ query_arr = []
189
+ query = ''
190
+ data.update('_p' => id) if id
191
+ data.update('_k' => @key)
192
+ data.update '_d' => 1 if data['_t']
193
+ data['_t'] ||= Time.now.to_i
194
+
195
+ unsafe = Regexp.new("[^#{URI::REGEXP::PATTERN::UNRESERVED}]", false, 'N')
196
+
197
+ data.inject(query) do |query,key_val|
198
+ query_arr << key_val.collect { |i| URI.escape(i.to_s, unsafe) }.join('=')
199
+ end
200
+ query = '/' + type + '?' + query_arr.join('&')
201
+ if @use_cron
202
+ log_query(query)
203
+ else
204
+ begin
205
+ send_query(query)
206
+ rescue Exception => e
207
+ log_query(query)
208
+ log_error(e)
209
+ end
210
+ end
211
+ end
212
+
213
+ def send_query(line)
214
+ if @dryrun
215
+ log_sent(line)
216
+ else
217
+ begin
218
+ host,port = @host.split(':')
219
+ proxy = URI.parse(ENV['http_proxy'] || ENV['HTTP_PROXY'] || '')
220
+ res = Net::HTTP::Proxy(proxy.host, proxy.port, proxy.user, proxy.password).start(host, port) do |http|
221
+ http.get(line)
222
+ end
223
+ rescue Exception => e
224
+ raise KMError.new("#{e} for host #{@host}")
225
+ end
226
+ log_sent(line)
227
+ end
228
+ end
229
+
230
+ def log_dir_writable?
231
+ if not FileTest.writable? @log_dir
232
+ $stderr.puts("Could't open #{log_name(:query)} for writing. Does #{@log_dir} exist? Permissions?") if @to_stderr
233
+ end
234
+ end
235
+
236
+ def is_initialized?
237
+ if @key == nil
238
+ log_error InitError.new("Need to initialize first (KMTS::init <your_key>)")
239
+ return false
240
+ end
241
+ return true
242
+ end
243
+ # :startdoc:
244
+ end
245
+ end