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
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
|
# -*- coding: utf-8 -*-
#
# dbi_io.rb: DBI IO for tDiary 2.x.
#
# NAME dbi_io
#
# DESCRIPTION tDiary向けDBI用IOクラス
# 詳細は、README.jaを参照してください
#
# Copyright (C) 2003 ma2tak <ma2tak@ma2tak.dyndns.org>
# (C) 2004 moonwolf <moonwolf@mooonwolf.com>
# (C) 2005 Kazuhiko <kazuhiko@fdiary.net>
# (C) 2012 hsbt <shibata.hiroshi@gmail.com>
#
# You can distribute this under GPL.
require 'tdiary/io/base'
require 'tempfile'
require 'dbi'
module TDiary
module CommentIO
def restore_comment(diaries)
begin
diaries.each do |date, diary_object|
@dbh.select_all("SELECT diary_id, name, mail, last_modified, visible, no, author, comment FROM commentdata WHERE author=? AND diary_id=? ORDER BY no;", @dbi_author, date) do |diary_id, name, mail, last_modified, visible, no, author, comment|
comment = Comment.new(name, mail, comment, Time.at(last_modified.to_i))
comment.show = visible
diary_object.add_comment(comment)
end
end
rescue Errno::ENOENT
end
end
def store_comment(diaries)
begin
diaries.each do |date, diary|
no = 0
diary.each_comment(diary.count_comments(true)) do |com|
no += 1
param = [com.name, com.mail, com.date.to_i, com.visible?, com.body, @dbi_author, date, no]
sth = @dbh.execute("UPDATE commentdata SET name=?, mail=?, last_modified=?, visible=?, comment=? WHERE author=? AND diary_id=? AND no=?;", *param)
if sth.rows == 0
@dbh.execute("INSERT INTO commentdata (name, mail, last_modified, visible, comment, author, diary_id, no) VALUES (?,?,?,?,?,?,?,?);", *param)
end
end
@dbh.execute("DELETE FROM commentdata where author=? AND diary_id=? AND no>?", @dbi_author, date, no)
end
rescue Errno::ENOENT
end
end
end
module RefererIO
def restore_referer(diaries)
return
end
def store_referer(diaries)
return
end
end
class DbiIO < BaseIO
include CommentIO
include RefererIO
def initialize(tdiary)
@tdiary = tdiary
@dbi_url = tdiary.conf.dbi_driver_url
@dbi_user = tdiary.conf.dbi_user
@dbi_passwd = tdiary.conf.dbi_passwd
@dbi_author = tdiary.conf.dbi_author || 'default'
@dbh = DBI.connect(@dbi_url, @dbi_user, @dbi_passwd)
load_styles
end
class << self
def load_cgi_conf(conf)
end
def save_cgi_conf(conf, result)
end
end
#
# block must be return boolean which dirty diaries.
#
def transaction(date)
File.open("#{Dir.tmpdir}/dbi_io.lock", 'w') do |file|
file.flock(File::LOCK_EX)
@dbh.transaction do
date_string = date.strftime("%Y%m%d")
diaries = {}
cache = @tdiary.restore_parser_cache(date, 'defaultio')
if cache
diaries.update(cache)
else
restore(date_string, diaries)
restore_comment(diaries)
restore_referer(diaries)
end
@old_referers = {}
diaries.each_pair{|k,v| @old_referers[k] = v.instance_variable_get('@referers').dup}
dirty = yield(diaries) if iterator?
store(diaries) if dirty & TDiary::TDiaryBase::DIRTY_DIARY != 0
store_comment(diaries) if dirty & TDiary::TDiaryBase::DIRTY_COMMENT != 0
store_referer(diaries) if dirty & TDiary::TDiaryBase::DIRTY_REFERER != 0
if dirty or not cache
@tdiary.store_parser_cache(date, 'defaultio', diaries)
end
end
end
end
def calendar
calendar = Hash.new{|hash, key| hash[key] = []}
sql = "SELECT year, month FROM diarydata WHERE author=? GROUP BY year, month ORDER BY year, month;"
@dbh.select_all(sql, @dbi_author) do |year, month|
calendar[year] << month
end
calendar
end
def diary_factory(date, title, body, style = 'tDiary')
styled_diary_factory(date, title, body, style)
end
# HNF移行ツールのため、作成
def restore_diary(date)
diaries = {}
restore(date, diaries, false)
diaries
end
private
def restore(date, diaries, month=true)
sql = "SELECT diary_id, title, last_modified, visible, body, style FROM DiaryData WHERE author='#{@dbi_author}' and diary_id='#{date}';"
if month == true
if /(\d\d\d\d)(\d\d)(\d\d)/ =~ date
sql = "SELECT diary_id, title, last_modified, visible, body, style FROM DiaryData WHERE author='#{@dbi_author}' AND year='#{$1}' AND month='#{$2}';"
end
end
@dbh.select_all(sql) do |diary_id, title, last_modified, visible, body, style|
style = 'tdiary' if style.nil? || style.empty?
style = style.downcase
diary = eval("#{style(style)}::new(diary_id, title, body, Time::at(last_modified.to_i))")
diary.show(visible)
diaries[diary_id] = diary
end
end
def store(diaries)
diaries.each do |date, diary|
# save diaries
if /(\d\d\d\d)(\d\d)(\d\d)/ =~ date
year = $1
month = $2
day = $3
end
visible = (diary.visible? ? "true" : "false")
param = [year, month, day, diary.title, diary.last_modified.to_i, visible, diary.to_src, diary.style, @dbi_author, date]
sth = @dbh.execute("UPDATE diarydata SET year=?, month=?, day=?, title=?, last_modified=?, visible=?, body=?, style=? WHERE author=? AND diary_id=?", *param)
if sth.rows == 0
@dbh.execute("INSERT INTO diarydata (year, month, day, title, last_modified, visible, body, style, author, diary_id) VALUES (?,?,?,?,?,?,?,?,?,?);", *param)
end
end
end
# 追加メソッド for test
def delete(diaries)
diaries.each do |date, diary|
sql = "DELETE FROM diarydata WHERE author=#{@dbi_author} AND diary_id=#{date};"
@dbh.execute(sql)
end
end
end
end
|
