# -*- coding: utf-8 -*-

require 'rubygems'

require 'active_record'

require 'kconv'

require 'yaml'

gem 'twitter'

require 'twitter'

require File.dirname(__FILE__) + "/model_ngram.rb"

require File.dirname(__FILE__) + '/Twitterers.rb'

require File.dirname(__FILE__) + '/Buzzwords.rb'

$KCODE = 'UTF8'

module Post

  def Post.markovStr(search=nil)

    # マルコフ連鎖

    words = Array.new

    ngs = Ngram.find(:all)

    if search != nil

      puts "make post with '#{search}'"

      tmp = Array.new

      ngs.each{|ng|

        tmp.push(ng) if ng.a+ng.b+ng.c =~ /#{search}/i

      }

      if tmp.size > 0

        ngs = tmp

      else

        puts "no match '#{search}'"

      end

    end

    start = ngs[rand(ngs.size)]

    words.push(start.a, start.b, start.c)

    puts start.to_s

    # 左へ伸ばす

    puts '---search left 3-grams---'

    left = start

    while left.head != true do

      begin

        ngs = Ngram.find(:all, :conditions => ["b=? and c=?", left.a, left.b])

        left = ngs[rand(ngs.size)]

        puts left.to_s

        words.unshift(left.a)

      rescue

        break

      end

    end

    # 右へ伸ばす

    puts '---search right 3-grams---'

    right = start

    while true do

      begin

        ngs = Ngram.find(:all, :conditions => ["a=? and b=?",right.b , right.c])

        right = ngs[rand(ngs.size)]

        puts right.to_s

        words.push(right.c)

      rescue

        break

      end

      break if right.tail && 0.7 > rand

    end

    # ほぼ確実にreply先を変更する

    followers = nil

    for i in 0...words.size do

      s = words[i]

      if s =~ /(@[a-zA-Z0-9_]+)/

        if 0.9 > rand

          if followers == nil

            tws =  Twitterers.new(config["user"], config["pass"])

            followers = tws.followers

          end

          if followers.size > 0

            reply = followers[rand(followers.size)]

            followers.delete(reply)

            words[i] = "@#{reply}"

          end

        end

      end

      for i in 0...words.size-1 do

        if words[i] == '@' && words[i+1] =~ /[a-zA-Z0-9_]+/ # @とusernameが分かれている時

          if 0.9 > rand

            if followers == nil

              tws = Twitterers.new(config["user"], config["pass"])

              followers = tws.followers

            end

            if followers.size > 0

              reply = followers[rand(followers.size)]

              followers.delete(reply)

              words[i+1] = reply

            end

          end

        end

      end

    end

    # すごくたまにfollowerに話しかける

    if 0.02 > rand

      if followers == nil

        tws = Twitterers.new(config["user"], config["pass"])

        followers = tws.followers

      end

      if followers.size > 0

        reply = followers[rand(followers.size)]

        followers.delete(reply)

        words.unshift("@#{reply}")

      end

    end

    # String化

    post = ''

    words.each{|s|

      post += ' ' if post =~ /[a-zA-Z]\Z/ && s =~ /\A[a-zA-Z]/ # 英単語はスペース開けて連結

      if post =~/[@\#]\Z/ && s =~ /[a-zA-Z0-9_]/ # @replyや#hashtagが、@とusernameに分かれていた時

        post += s + ' '

        next

      end

      if s =~ /([@#][a-zA-Z0-9_]+)/ # @replyか#hashtag

        post += ' ' if post.last != ' '

        post += s + ' '

      else

        post += s

      end

    }

    return post.strip.to_s

  end

  def Post.has_ngWord(post)

    ngwords = Array.new

    open(File.dirname(__FILE__)+"/ngwords").read.each{ |line|

      ngwords.push(line.chomp)

    }

    ngwords.each{ |w|

      if post =~ /#{w}/

        return true

      end

    }

    return false

  end

  def Post.tweet_length_check(post)

    min = config["tweet_len_min"] || 0

    max = config["tweet_len_max"] || 200

    len = post.split(//u).size

    return (min <= len && len <= max)

  end

  # 括弧が対応していなかったら修正する

  def Post.fix_brackets(str)

    brackets = [['<','>'],['＜','＞'],['(',')'],['（','）'], ['「','」'],['『','』'],['[',']'],['"','"']]

    delimiter = brackets.join('|').split('|').map{|b|

      if b =~ /\(|\)|\[|\]|\"|\'/

        "\\#{b}" 

      else

        b

      end

    }.join('|')

    words = str.split(/(#{delimiter})/)

    pairs = Array.new

    for i in 0...words.size do

      brackets.each{|b|

        if words[i] == b[0] # 左括弧

          for j in i...words.size do

            if words[j] == b[1] && !pairs.index(i) && !pairs.index(j) # 対になる括弧が右にあるか

              next if pairs.size>0 && pairs.max > i # 括弧同士の重複

              pairs << i

              pairs << j

            end

          end

        end

      }

    end

    for i in 0...words.size do

      if words[i] =~ /#{delimiter}/ && !pairs.index(i)

        words[i] = ""

      end

    end

    words.join

  end

end

require File.dirname(__FILE__) + '/Post.rb'

post = Post.markovStr(search).toutf8

    break if Post.tweet_length_check(post) && !Post.has_ngWord(post)

    post = Post.markovStr(search).toutf8

post = Post.fix_brackets(post)