komi-san-slack_2-3.rb

require 'slack-ruby-client'
require 'net/http'
require 'uri'
require 'json'
require 'logger'

TOKEN = ENV["SLACK_TOKEN"]

R18_LIST = File.foreach('wordlist.txt').map do |word|
  word.chomp
end
R18_LIST.freeze

def r18_word? text
  trim_str  = text.split('しよう')[0]
  
  return r18_check_google(trim_str)

end

#text = 'ぴよぴよ'
#google searchで「ぴよぴよ」を検索
#trueならエロ
def r18_check_google text
  total_off  = google_search(text, "off")
  total_high = google_search(text, "high")
  
  puts text 
  puts "safe off : #{total_off}"
  puts "safe high: #{total_high}" 
  
  #とりあえず、50%くらいで
  judge_res = (total_off.to_i - total_high.to_i) > total_high.to_i
  puts judge_res

  return judge_res
end


#検索後、検索件数を返す
def google_search text, r18 = "off"
  # [ロガー]
  # カレントディレクトリのwebapi.logというファイルに出力
  logger = Logger.new('./webapi.log')

  # [クエリパラメータ]
  # URI.encode_www_formを使って「application/x-www-form-urlencoded」形式の文字列に変換
  # 文字列はURLエンコードされた形式に変換（半角スペースの"+"への変換等）
  #
  # （変換例）
  # 'bar baz' => 'bar+baz'
  # 'あ' => '%E3%81%82'
  #params = URI.encode_www_form({ param1: 'foo', param2: 'bar baz' , param3: 'あ' })
    params = URI.encode_www_form(key: ENV["SEARCH_KEY"], cx: ENV["GOO_ENGINE_ID"], q: text, safe: r18)

  # [URI]
  # URI.parseは与えられたURIからURI::Genericのサブクラスのインスタンスを返す
  # -> 今回はHTTPプロトコルなのでURI::HTTPクラスのインスタンスが返される
  #
  # オブジェクトからは以下のようにして構成要素を取得できる
  # uri.scheme => 'http'
  # uri.host   => 'mogulla3.com'
  # uri.port   => 4567
  # uri.path   => ''
  # uri.query  => 'param1=foo&param2=bar+baz&param3=%E3%81%82'
  #uri = URI.parse("http://www.google.com?#{params}")
  uri = URI.parse("https://www.googleapis.com/customsearch/v1?#{params}")
  puts uri

  begin
    # [GETリクエスト]
    # Net::HTTP.startでHTTPセッションを開始する
    # 既にセッションが開始している場合はIOErrorが発生
    #response = Net::HTTP.start(uri.host, uri.port) do |http|

    #sslを許可
    http_obj = Net::HTTP.new(uri.host, uri.port)
    http_obj.use_ssl =true
    response = http_obj.start do |http|
      # Net::HTTP.open_timeout=で接続時に待つ最大秒数の設定をする
      # タイムアウト時はTimeoutError例外が発生
      http.open_timeout = 5

      # Net::HTTP.read_timeout=で読み込み1回でブロックして良い最大秒数の設定をする
      # デフォルトは60秒
      # タイムアウト時はTimeoutError例外が発生
      http.read_timeout = 10

      # Net::HTTP#getでレスポンスの取得
      # 返り値はNet::HTTPResponseのインスタンス
      http.get(uri.request_uri)
    end

    # [レスポンス処理]
    # 2xx系以外は失敗として終了することにする
    # ※ リダイレクト対応できると良いな..
    #
    # ステータスコードに応じてレスポンスのクラスが異なる
    # 1xx系 => Net::HTTPInformation
    # 2xx系 => Net::HTTPSuccess
    # 3xx系 => Net::HTTPRedirection
    # 4xx系 => Net::HTTPClientError
    # 5xx系 => Net::HTTPServerError
    case response
      # 2xx系
    when Net::HTTPSuccess
      # [JSONパース処理]
      # JSONオブジェクトをHashへパースする
      # JSON::ParserErrorが発生する可能性がある
      #p JSON.parse(response.body)
      res_json = JSON.parse(response.body)
      total = res_json.dig("queries","request",0,"totalResults")
      puts total
      return total
      # 3xx系
    when Net::HTTPRedirection
      # リダイレクト先のレスポンスを取得する際は
      # response['Location']でリダイレクト先のURLを取得してリトライする必要がある
      logger.warn("Redirection: code=#{response.code} message=#{response.message}")
    else
      logger.error("HTTP ERROR: code=#{response.code} message=#{response.message}")
    end

    # [エラーハンドリング]
    # 各種処理で発生しうるエラーのハンドリング処理
    # 各エラーごとにハンドリング処理が書けるようにrescue節は小さい単位で書く
    # (ここでは全て同じ処理しか書いていない)
  rescue IOError => e
    logger.error(e.message)
  rescue TimeoutError => e
    logger.error(e.message)
  rescue JSON::ParserError => e
    logger.error(e.message)
  rescue => e
    logger.error(e.message)
  end
end

Slack.configure do |config|
  config.token = TOKEN
end

client = Slack::RealTime::Client.new

client.on :hello do
  puts '...'
end

client.on :message do |data|
  if !data['text'][/しよう(。|\.)*/].nil?
    if r18_word?(data['text'])
      sleep(3)
      client.message channel: data['channel'], text: '"そして古見さんは石になった。"'
    else
      client.message channel: data['channel'], text: 'ﾌﾝｽ!'
    end
  else 
    client.message channel: data['channel'], text:'...'
  end
end

client.start!