firestore_client.rb

# require 'google/apis/firestore_v1beta1'
require 'net/http'
require 'dotenv'
Dotenv.load
require "google/cloud/firestore"


class FirestoreClient

  # インスタンスを作成
  # @param [nil] 
  # @return [nil]
  def initialize()    
    @firestore_client = Google::Cloud::Firestore.new(
      project_id: "pocketmentor-455ae",
      credentials: "pocketmentor-455ae-firebase-adminsdk-bw1dg-9626bcc8d7.json"
    )
  end

  # ない場合は新規作成される
  # ある場合は更新される
  # users/tweet_id(不変ID)/ここ更新
  # @param [Object] Tweetオブジェクト 
  # @return [nil]
  def saveUser(tweet)
    doc_ref = @firestore_client.doc "users/#{tweet.user.id}"
    doc_ref.set(
      screen_name: tweet.user.screen_name,
      last_update: Time.new
    )
  end

  # ~したいの最初のツイートを保存
  # ない場合は新規作成されるある場合は更新される
  # users/twitter_id/tweets/tweetのID/ここを更新
  # @param [Object] Tweetオブジェクト 
  # @return [nil]
  def saveTweet(tweet)
    tweet_ref = @firestore_client.doc "users/#{tweet.user.id}/tweets/#{tweet.id}"
    tweet_ref.set(
      tweet: tweet.text,
      created_at: tweet.created_at,
      is_done: false
    )
  end

  # botのリプライを保存する
  # リプライ先のツイートの直に下保存する
  # users/twitter_id/tweets/tweetのID/replies/ここを更新
  # @param [Object] Tweetオブジェクト 
  # @return [nil]
  def saveReply(tweet)
    reply_ref = @firestore_client.doc "users/#{tweet.user.id}/tweets/#{tweet.id}/replies/#{tweet.id}"
    reply_ref.set(
      tweet: tweet.text,
      created_at: tweet.created_at,
      is_done: false
    )
  end

  # ユーザーが呟いた全てのやりたいことツイート
  # リプライ先のツイートの直に下保存する
  # @param [user_id] user_id (tweet.user.id)不変のid
  # @return [String] firestoreに保存してるツイートの中身
  # tweet.data →{tweet: "tweet text"}
  def getTweets(user_id)
    tweets_ref = @firestore_client.col "users/#{user_id}/tweets"
    tweets=[]
    tweets_ref.get do |tweet|
      tweets.push(tweet.data)
    end
    return tweets
  end

  # ユーザが事前に登録したやりたいことの中で最新のツイートを返す
  # @param [Integer] ユーザーID
  # @return [Object] firebaseのデータオブジェクト
  def getUserRecentWishTweet(user_id)
    recent_wish_tweets = getTweets(user_id)
    
    return recent_wish_tweets.sample(1)[0] unless recent_wish_tweets.empty?
  end

end