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
|
# tdiary_vote.rb $Revision: 2 $
# Copyright (C) 2006 Michitaka Ohno <elpeo@mars.dti.ne.jp>
# You can redistribute it and/or modify it under GPL2.
#
# ref. http://elpeo.jp/diary/20060622.html#p01
#
# .tdiary-vote {
# float: left;
# background-color: aqua;
# }
require 'digest/md5'
require 'pstore'
@tdiary_vote_db = "#{@cache_path}/tdiary_vote"
@tdiary_vote_label = "投票"
@tdiary_vote_date = nil
def vote( *items )
return '' unless @tdiary_vote_date
h, voted = get_vote( @tdiary_vote_date, @cgi.cookies['tdiary_vote'][0] )
max = h.empty? ? 1 : h.values.max
r = %Q[<table border="1">]
items.sort{|a, b| h[b] <=> h[a]}.each do |item|
num = h[item]
r << %Q[<tr>]
r << %Q[<td>#{CGI.escapeHTML( item )}</td>]
r << %Q[<td><span class="tdiary-vote" style="height: 2.5ex; width: #{(100*num/max).to_i}px"></span>#{num}</td>]
unless voted then
r << %Q[<td>]
r << %Q[<form submit="#{@conf.index}" method="POST" style="margin: 0px; padding: 0px">]
r << %Q[#{csrf_protection}]
r << %Q[<input type="hidden" name="date" value="#{@tdiary_vote_date.strftime( "%Y%m%d" )}">]
r << %Q[<input type="hidden" name="name" value="">]
r << %Q[<input type="hidden" name="body" value="">]
r << %Q[<input type="hidden" name="vote" value="#{CGI.escapeHTML( item )}">]
r << %Q[<input type="submit" name="comment" value="#{@tdiary_vote_label}">]
r << %Q[</form>]
r << %Q[</td>]
end
r << %Q[</tr>]
end
r << %Q[</table>]
end
def get_vote( date, uid )
h = Hash.new(0)
voted = false
file = "#{@tdiary_vote_db}/#{date.strftime( "%Y%m%d" )}.db"
if File.exist?( file ) then
PStore.new( file ).transaction do |db|
h.update( db['vote'] ) if db.root?( 'vote' )
voted = db['voter'].include?( uid ) if db.root?( 'voter' )
db.abort
end
end
[h, voted]
end
def add_vote( date, item, uid )
Dir::mkdir( @tdiary_vote_db ) unless File::directory?( @tdiary_vote_db )
file = "#{@tdiary_vote_db}/#{date.strftime( "%Y%m%d" )}.db"
PStore.new( file ).transaction do |db|
db['voter'] = Hash.new unless db.root?( 'voter' )
db.abort if db['voter'].include?( uid )
db['voter'][uid] = @cgi.remote_addr
db['vote'] = Hash.new(0) unless db.root?( 'vote' )
db['vote'][item] += 1
end
end
add_body_enter_proc do |date|
@tdiary_vote_date = date
''
end
add_body_leave_proc do |date|
@tdiary_vote_date = nil
''
end
unless bot? then
if @mode == 'comment' && @cgi.valid?( 'vote' ) && @cgi.cookies['tdiary_vote'][0] then
add_vote( @date, @cgi.params['vote'][0], @cgi.cookies['tdiary_vote'][0] )
end
add_footer_proc do
uid = @cgi.cookies['tdiary_vote'][0] || Digest::MD5.hexdigest( @cgi.remote_addr + Time.now.to_s + rand.to_s )
cookie_path = File::dirname( @cgi.script_name )
cookie_path += '/' if cookie_path !~ /\/$/
cookie = CGI::Cookie::new(
'name' => 'tdiary_vote',
'value' => uid,
'path' => cookie_path,
'expires' => Time.now.gmtime + 30*24*60*60
)
add_cookie( cookie )
''
end
end
|
