tdiary 3.2.2.20130507

data/tdiary/io/pstore.rb

@@ -0,0 +1,252 @@
+# -*- coding: utf-8; -*-
+#
+# pstoreio.rb: tDiary IO class of tdiary 1.x format. $Revision: 1.25 $
+#
+# Copyright (C) 2001-2005, TADA Tadashi <t@tdtds.jp>
+# You can redistribute it and/or modify it under GPL2.
+#
+require 'pstore'
+
+module TDiary
+	class PStoreIO
+		def initialize( tdiary )
+			@data_path = tdiary.conf.data_path
+		end
+
+		#
+		# block must be return boolean which dirty diaries.
+		#
+		def transaction( date )
+			diaries = {}
+			filename = date.strftime( "#{@data_path}%Y%m" )
+			begin
+				PStore::new( filename ).transaction do |db|
+					dirty = false
+					if db.root?( 'diary' ) then
+						diaries.update( db['diary'] )
+					end
+					dirty = yield( diaries ) if iterator?
+					if dirty != TDiary::TDiaryBase::DIRTY_NONE then
+						db['diary'] = diaries
+					else
+						db.abort
+					end
+				end
+			rescue PStore::Error, NameError, Errno::EACCES
+				raise PermissionError::new( "make your @data_path to writable via httpd. #$!" )
+			end
+			begin
+				File::delete( filename ) if diaries.empty?
+			rescue Errno::ENOENT
+			end
+			return diaries
+		end
+
+		def calendar
+			calendar = {}
+			Dir["#{@data_path}??????"].sort.each do |file|
+				year, month = file.scan( %r[/(\d{4})(\d\d)$] )[0]
+				next unless year
+				calendar[year] = [] unless calendar[year]
+				calendar[year] << month
+			end
+			calendar
+		end
+
+		def diary_factory( date, title, body, style = nil )
+			Diary::new( date, title, body )
+		end
+	end
+end
+
+=begin
+== class Comment
+  Management a comment.
+=end
+class Comment
+	attr_reader :name, :mail, :body, :date
+
+	def initialize( name, mail, body, date = Time::now )
+		@name, @mail, @body, @date = name, mail, body, date
+		@show = true
+	end
+
+	def shorten( len = 120 )
+		lines = NKF::nkf( "-e -m0 -f#{len}", @body.gsub( /\n/, ' ' ) ).split( /\n/ )
+		lines[0].concat( '..' ) if lines[0] and lines[1]
+		lines[0] || ''
+	end
+
+	def visible?; @show; end
+	def show=( s ); @show = s; end
+
+	def ==( c )
+		(@name == c.name) and (@mail == c.mail) and (@body == c.body)
+	end
+end
+
+=begin
+== Paragraph class
+Management a paragraph.
+=end
+class Paragraph
+	attr_reader :subtitle, :body
+	alias :body_to_html :body
+	alias :subtitle_to_html :subtitle
+
+	def initialize( fragment, author = nil )
+		@author = author
+		lines = fragment.split( /\n+/ )
+		if lines.size > 1 then
+			if /^<</ =~ lines[0]
+				@subtitle = lines.shift.chomp.sub( /^</, '' )
+			elsif /^[　 <]/u !~ lines[0]
+				@subtitle = lines.shift.chomp
+			end
+		end
+		@body = lines.join( "\n" )
+	end
+
+	def to_src
+		s = ''
+		if @subtitle then
+			s += "[#{@author}]" if @author
+			s += '<' if /^</ =~ @subtitle
+			s += @subtitle + "\n"
+		end
+		"#{s}#{@body}\n\n"
+	end
+
+	def to_s
+		"subtitle=#{@subtitle}, body=#{@body}"
+	end
+
+	def author
+		@author = @auther unless @author
+		@author
+	end
+end
+
+=begin
+== Diary class
+Management a day of diary
+=end
+class Diary
+	include TDiary::DiaryBase
+	include TDiary::UncategorizableDiary
+
+	def initialize( date, title, body )
+		init_diary
+		replace( date, title, body )
+	end
+
+	def style
+		'tDiary'
+	end
+
+	def replace( date, title, body )
+		set_date( date )
+		set_title( title )
+		@paragraphs = []
+		append( body )
+	end
+
+	def append( body, author = nil )
+		body.gsub( /\r/, '' ).split( /\n\n+/ ).each do |fragment|
+			paragraph = Paragraph::new( fragment, author )
+			@paragraphs << paragraph if paragraph
+		end
+		@last_modified = Time::now
+		self
+	end
+
+	def each_section
+		@paragraphs.each do |paragraph|
+			yield paragraph
+		end
+	end
+
+	def to_src
+		src = ''
+		each_section do |para|
+			src << para.to_src
+		end
+		src
+	end
+
+	def to_html( opt, mode = :HTML )
+		case mode
+		when :CHTML
+			to_chtml( opt )
+		else
+			to_html4( opt )
+		end
+	end
+
+	def to_html4( opt )
+		idx = 1
+		r = ''
+		each_section do |paragraph|
+			r << %Q[<div class="section">\n]
+			if paragraph.subtitle then
+				r << %Q[<h3><a ]
+				if opt['anchor'] then
+					r << %Q[name="p#{'%02d' % idx}" ]
+				end
+				r << %Q[href="#{opt['index']}<%=anchor "#{@date.strftime( '%Y%m%d' )}#p#{'%02d' % idx}" %>">#{opt['section_anchor']}</a> ]
+				if opt['multi_user'] and paragraph.author then
+					r << %Q|[#{paragraph.author}]|
+				end
+				r << %Q[#{paragraph.subtitle}</h3>]
+			end
+			if /^</ =~ paragraph.body then
+				r << %Q[#{paragraph.body}]
+			elsif paragraph.subtitle
+				r << %Q[<p>#{paragraph.body.collect{|l|l.chomp.sub( /^[　 ]/u, '' )}.join( "</p>\n<p>" )}</p>]
+			else
+				r << %Q[<p><a ]
+				if opt['anchor'] then
+					r << %Q[name="p#{'%02d' % idx}" ]
+				end
+				r << %Q[href="#{opt['index']}<%=anchor "#{@date.strftime( '%Y%m%d' )}#p#{'%02d' % idx}" %>">#{opt['section_anchor']}</a> #{paragraph.body.collect{|l|l.chomp.sub( /^[　 ]/u, '' )}.join( "</p>\n<p>" )}</p>]
+			end
+			r << %Q[</div>]
+			idx += 1
+		end
+		r
+	end
+
+	def to_chtml( opt )
+		idx = 0
+		r = ''
+		each_section do |paragraph|
+			if paragraph.subtitle then
+				r << %Q[<P><A NAME="p#{'%02d' % idx += 1}">*</A> #{paragraph.subtitle}</P>]
+			end
+			if /^</ =~ paragraph.body then
+				idx += 1
+				r << paragraph.body
+			elsif paragraph.subtitle
+				r << %Q[<P>#{paragraph.body.collect{|l|l.chomp.sub( /^[　 ]/u, '' )}.join( "</P>\n<P>" )}</P>]
+			else
+				r << %Q[<P><A NAME="p#{'%02d' % idx += 1}">*</A> ]
+				if opt['multi_user'] and paragraph.author then
+					r << %Q|[#{paragraph.author}]|
+				end
+				r << %Q[#{paragraph.body.collect{|l|l.chomp.sub( /^[　 ]/u, '' )}.join( "</P>\n<P>" )}</P>]
+			end
+		end
+		r
+	end
+
+	def to_s
+		"date=#{@date.strftime('%Y%m%d')}, title=#{@title}, body=[#{@paragraphs.join('][')}]"
+	end
+end
+
+# Local Variables:
+# mode: ruby
+# indent-tabs-mode: t
+# tab-width: 3
+# ruby-indent-level: 3
+# End:

data/tdiary/io/rdb.rb

@@ -0,0 +1,213 @@
+# -*- coding: utf-8 -*-
+#
+# rdb.rb: RDB IO for tDiary 3.x
+#
+# NAME             rdb
+#
+# DESCRIPTION      RDB 向け tDiary IO クラス
+#                  日記データは RDB に保存、リンク元の記録には未対応
+#
+# 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 'sequel'
+
+module TDiary
+  module CommentIO
+    def restore_comment(diaries)
+      diaries.each do |date, diary_object|
+        db[:comments].filter(:diary_id => date).order_by(:no).select(:name, :mail, :last_modified, :visible, :comment).each do |row|
+          comment = Comment.new(row[:name], row[:mail], row[:comment], Time.at(row[:last_modified].to_i))
+          comment.show = row[:visible]
+          diary_object.add_comment(comment)
+        end
+      end
+    end
+
+    def store_comment(diaries)
+      diaries.each do |diary_id, diary|
+        no = 0
+        diary.each_comment(diary.count_comments(true)) do |com|
+          no += 1
+          date = {
+            :diary_id => diary_id,
+            :no => no
+          }
+          body = {
+            :name => com.name,
+            :mail => com.mail,
+            :last_modified => com.date.to_i,
+            :visible => com.visible?,
+            :comment => com.body
+          }
+          comment = db[:comments].filter(date)
+          if comment.count > 0
+            comment.update(body)
+          else
+            db[:comments].insert(date.merge(body))
+          end
+        end
+      end
+    end
+  end
+
+  module RefererIO
+    def restore_referer(diaries)
+      # not implemented yet
+      return
+    end
+
+    def store_referer(diaries)
+      # not implemented yet
+      return
+    end
+  end
+
+  class RdbIO < BaseIO
+    include CommentIO
+    include RefererIO
+    include CacheIO
+
+    class << self
+      def load_cgi_conf(conf)
+        if cgi_conf = db(conf)[:conf].select(:body).first
+          cgi_conf[:body]
+        else
+          ""
+        end
+      end
+
+      def save_cgi_conf(conf, result)
+        if db(conf)[:conf].count > 0
+          db(conf)[:conf].update(:body => result)
+        else
+          db(conf)[:conf].insert(:body => result)
+        end
+      end
+
+      def db(conf)
+        @@_db ||= Sequel.connect(conf.database_url || ENV['DATABASE_URL'])
+
+        @@_db.create_table :conf do
+          String :body, :text => true
+        end unless @@_db.table_exists?(:conf)
+
+        @@_db.create_table :diaries do
+          String :diary_id, :size => 8
+          String :year, :size => 4
+          String :month, :size => 2
+          String :day, :size => 2
+          String :title, :text => true
+          String :body, :text => true
+          String :style, :text => true
+          Fixnum :last_modified
+          TrueClass :visible
+          primary_key :diary_id
+        end unless @@_db.table_exists?(:diaries)
+
+        @@_db.create_table :comments do
+          String :diary_id, :size => 8
+          Fixnum :no
+          String :name, :text => true
+          String :mail, :text => true
+          String :comment, :text => true
+          Fixnum :last_modified
+          TrueClass :visible
+          primary_key [:diary_id, :no]
+        end unless @@_db.table_exists?(:comments)
+
+        @@_db
+      end
+    end
+
+    #
+    # block must be return boolean which dirty diaries.
+    #
+    def transaction(date)
+      diaries = {}
+
+      if cache = restore_parser_cache(date)
+        diaries.update(cache)
+      else
+        restore(date.strftime("%Y%m%d"), diaries)
+        restore_comment(diaries)
+      end
+
+      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_parser_cache(date, diaries) if dirty || !cache
+    end
+
+    def calendar
+      calendar = Hash.new{|hash, key| hash[key] = []}
+      db[:diaries].select(:year, :month).group_by(:year, :month).order_by(:year, :month).each do |row|
+        calendar[row[:year]] << row[:month]
+      end
+      calendar
+    end
+
+    def cache_dir
+      @tdiary.conf.cache_path || "#{Dir.tmpdir}/cache"
+    end
+
+    private
+
+    def restore(date, diaries, month = true)
+      query = db[:diaries].select(:diary_id, :title, :last_modified, :visible, :body, :style)
+      query = if month && /(\d\d\d\d)(\d\d)(\d\d)/ =~ date
+                query.filter(:year => $1, :month => $2)
+              else
+                query.filter(:diary_id => date)
+              end
+      query.each do |row|
+        style = if row[:style].nil? || row[:style].empty?
+                  'wiki'
+                else
+                  row[:style].downcase
+                end
+        diary = eval("#{style(style)}::new(row[:diary_id], row[:title], row[:body], Time::at(row[:last_modified].to_i))")
+        diary.show(row[:visible])
+        diaries[row[:diary_id]] = diary
+      end
+    end
+
+    def store(diaries)
+      diaries.each do |diary_id, diary|
+        date = if /(\d\d\d\d)(\d\d)(\d\d)/ =~ diary_id
+                 {
+                   :year => $1,
+                   :month => $2,
+                   :day => $3,
+                   :diary_id => diary_id
+                 }
+               end
+        body = {
+          :title => diary.title,
+          :last_modified => diary.last_modified.to_i,
+          :style => diary.style,
+          :visible => diary.visible?,
+          :body => diary.to_src
+        }
+
+        entry = db[:diaries].filter(date)
+        if entry.count > 0
+          entry.update(body)
+        else
+          db[:diaries].insert(date.merge(body))
+        end
+      end
+    end
+
+    def db
+      self.class.db(@tdiary.conf)
+    end
+  end
+end