-
Notifications
You must be signed in to change notification settings - Fork 0
/
model.rb
146 lines (120 loc) · 3.32 KB
/
model.rb
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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
require 'byebug'
class Model
@@conditions = {}
def self.find_col
col_names = QuestionsDatabase.instance.execute(<<-SQL)
PRAGMA
table_info("#{self.get_table}")
SQL
col_names = col_names.map { |hash| hash['name'] }
end
def self.find_by(search_str)
data = QuestionsDatabase.instance.execute(<<-SQL)
SELECT
*
FROM
#{self.get_table}
WHERE
#{search_str}
SQL
data.empty? ? nil : data.map { |datum| self.new(datum) }
end
def self.method_missing(method_name, *args)
if method_name[0..7] == "find_by_"
cols = method_name[8..-1].split("_and_")
raise "You can't search by that" unless self.check_ivar(cols)
self.find_by(self.make_search_str(cols,args))
end
end
def self.check_ivar(cols)
cols.all? { |col| find_col.include?(col) }
end
def self.find_by_id(id)
data = QuestionsDatabase.instance.execute(<<-SQL, id)
SELECT
*
FROM
#{self.get_table}
WHERE
id = ?
SQL
data.empty? ? nil : self.new(data.first)
end
def self.make_search_str(cols, values)
search_arr = []
cols.length.times do |idx|
if values[idx].is_a?(Fixnum)
search_arr << "#{cols[idx]} = #{values[idx]}"
else
search_arr << "#{cols[idx]} = '#{values[idx]}'"
end
end
search_str = search_arr.join(" AND ")
end
# def self.where(options = {})
# cols = options.keys.map(&:to_s)
# raise "You can't search by that" unless self.check_ivar(cols)
# values = options.values
# self.find_by(self.make_search_str(cols, values))
# end
# Reply.where(:parent_id => 1).where(:question_id => 1).where
def self.where(options = {})
@@conditions.merge!(options)
if options.empty?
self.where!
else
self
end
end
def self.where!
cols = @@conditions.keys.map(&:to_s)
raise "You can't search by that" unless self.check_ivar(cols)
values = @@conditions.values
@@conditions = {}
self.find_by(self.make_search_str(cols, values))
end
def self.all
all_data = QuestionsDatabase.instance.execute("SELECT * FROM #{self.get_table}")
all_data.map { |data| self.new(data) }
end
def self.get_table
return "replies" if self == Reply
class_name = self.to_s.downcase
name_array = class_name.split /(?=[A-Z])/
name_array.join("_") + "s"
end
def where(options = {})
end
def save
ivar_names = get_ivars_as_strings.drop(1)
ivar_values = ivar_names.map { |ivar| send(ivar) }
if id.nil?
ivar_string = ivar_names.join(", ")
question_mark_string = (["?"] * ivar_names.length).join(", ")
QuestionsDatabase.instance.execute(<<-SQL, *ivar_values)
INSERT INTO
#{self.class.get_table}(#{ivar_string})
VALUES
(#{question_mark_string})
SQL
self.id = QuestionsDatabase.instance.last_insert_row_id
else
vars = []
ivar_values.length.times do |idx|
vars << "#{ivar_names[idx]} = '#{ivar_values[idx]}'"
end
var_equalities = vars.join(", ")
QuestionsDatabase.instance.execute(<<-SQL, id)
UPDATE
#{self.class.get_table}
SET
#{var_equalities}
WHERE
id = ?
SQL
end
end
def get_ivars_as_strings
self.instance_variables.map { |ivar| ivar.to_s[1..-1] }
end
end