summaryrefslogtreecommitdiff
path: root/rhnop-client/db.lua
blob: ce3b15f5ada75244d919c2a8b06913cb6b36f8cc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
--
--  rhnop
--
--  Copyright (C) 2011 Christian Pointner <equinox@helsinki.at>
--
--  This file is part of rhnop.
--
--  rhnop is free software: you can redistribute it and/or modify
--  it under the terms of the GNU General Public License as published by
--  the Free Software Foundation, either version 3 of the License, or
--  any later version.
--
--  rhnop is distributed in the hope that it will be useful,
--  but WITHOUT ANY WARRANTY; without even the implied warranty of
--  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
--  GNU General Public License for more details.
--
--  You should have received a copy of the GNU General Public License
--  along with rhnop. If not, see <http://www.gnu.org/licenses/>.
--

require "luasql.mysql"

-- CREATE DATABASE rhnop
-- GRANT select,insert,update ON rhnop.* TO 'nopsyncd' IDENTIFIED BY '<password>';
-- GRANT select ON rhnop.* TO 'nopcollectd' IDENTIFIED BY '<password>';
-- USE rhnop
-- CREATE TABLE IF NOT EXISTS now (timestamp BIGINT UNSIGNED PRIMARY KEY NOT NULL, cart INT NOT NULL, len INT, showtitle VARCHAR(255), title VARCHAR(255), artist VARCHAR(255), album VARCHAR(255), ismusic BOOLEAN);

local db = {}

function db:init(db, user, pwd, host, port, table)
   self.table = table

   local err

   self.env, err = luasql.mysql()
   if self.env == nil then
      return nil, err
   end

   self.con, err = self.env:connect(db, user, pwd, host, port)
   if self.con == nil then
      return nil, err
   end

   local ret, err = self.con:setautocommit(true)
   if ret == nil then
      return nil, err
   end

   return true
end

function db:getLastEntry()
   local cur, err = self.con:execute("SELECT MAX(timestamp) FROM " .. self.table)
   if cur == nil then
      return nil, err
   end

   local timestamp = cur:fetch()
   if timestamp == nil then timestamp = 0 end
   return timestamp
end

function db:findMissingEntries(lasttimestamp)
   local lastts = self.con:escape(lasttimestamp)
   return = self.con:execute("SELECT * FROM " .. self.table .. " WHERE timestamp > '" .. lastts .. "')")
end

function db:getNextMissingEntry(cur)
   local data = {}
   data = cur:fetch(data, "a")
   if data == nil then
      return nil, "that's all folks"
   end
   return data
end

function db:getEntry(timestamp)
   local ts = self.con:escape(timestamp)
   local cur, err = self.con:execute("SELECT * FROM " .. self.table .. " WHERE timestamp = '" .. ts .. "')")
   if cur == nil then
      return nil, err
   end

   local data = {}
   data = cur:fetch(data, "a")
   if data == nil then
      return nil, "nothing found"
   end
   return data
end

function db:addEntry(data)
   local timestamp = self.con:escape(data.timestamp)
   local cart = self.con:escape(data.cart)
   local len = self.con:escape(data.len)
   local showtitle = self.con:escape(data.showtitle)
   local title = self.con:escape(data.title)
   local artist = self.con:escape(data.artist)
   local album = self.con:escape(data.album)
   local ismusic = self.con:escape(data.ismusic)

   local cur, err = self.con:execute("REPLACE INTO " .. self.table .. " VALUES('" .. timestamp .. "', '" .. cart .. "', '" .. len .. "', '" .. showtitle .. "', '" .. title .. "', '" .. artist .."', '" .. album .. "', '" .. ismusic "')")
   if cur == nil then
      return nil, err
   end

   return true
end

function db:close()
   if self.con then
      self.con:close()
   end

   if self.env then
      self.env:close()
   end
end

return db