#!/usr/bin/ruby
#
# SimplePaste
#
###################################################

BASE_DIR   = '/home/ko1/src/simplepaste/' # File.dirname(__FILE__)
ERB_DIR    = 'sp_erb'
STORE_DIR  = 'sp_store'
BACKUP_DIR = 'sp_backup'

###################################################
require File.join(BASE_DIR, 'simplepaste')

StoreClass = SimpleStore

class TextSimplePaste < SimplePaste
  def edit_line_number text
    i = 1
    text.split(/\n/).map{|line|
      "#{'%04d' % (i+=1)}: #{line}"
    }.join("\n")
  end

  EDIT_COMMANDS = ['line_number']

  def on_view page_id, *args
    move_to :make, page_id unless StoreClass.exist? page_id
    store = StoreClass.open(page_id)
    text = store['body']

    args.each{|option|
      edit_command = "edit_#{option}"
      if respond_to? edit_command
        text = __send__(edit_coomand, text)
      end
    }

    store['body']    = escape(text)
    store['page_id'] = page_id
    store['header']  = '...'
    store['v_url']   = "#{script_url}/view/#{page_id}"
    store['ro_url']  = "#{script_url}/readonly/#{page_id}"
    show :view, store
  end

  def on_readonly page_id
    store = StoreClass.open(page_id)
    show :readonly, 'body' => escape(store['body'])
  end

  def on_raw page_id
    raise "You can't view raw page with MSIE" if @cgi.user_agent =~ /MSIE/
    store = StoreClass.open(page_id)
    raise RawTextNotice.new(store['body'])
  end

  def commit_new_entry store
    store['body']   = q('paste_body')
  end
end

TextSimplePaste.new(
  File.join(BASE_DIR, ERB_DIR),
  File.join(BASE_DIR, STORE_DIR),
  File.join(BASE_DIR, BACKUP_DIR)).run