CloudFunctionsから上限までツイート保存

CloudFunctions から YouTubeAPI で特定チャンネル動画を取得するのハッカソンで TwitterAPI の実装も行ったので共有します。 Twitter API を使うで大体分かります。

ツイートの保存

サンプルコードは以下です。

async function saveUserTweet(uid: string) {
  // Firestoreのuidを渡す
  const twitterData = (await db.doc(`private/${uid}`).get()).data(); // Firestoreからあらかじめ保存したTwitterのユーザートークンを取得
  if (twitterData) {
    const twitterClient = new Twitter({
      consumer_key: config.twitter.consumer_key, // TwitterAPIのAPPにあるものを環境変数に入れておき、ここで入れる
      consumer_secret: config.twitter.secret_key,
      access_token_key: twitterData.access_token_key, // ユーザートークンを入れる
      access_token_secret: twitterData.access_token_secret,
    });
    const twitterUid = twitterData.twitterUid; // あらかじめ保存したTwitterのuidを取得
    return getTweetAndSave(twitterClient, twitterUid, uid);
  } else {
    throw new Error("認証に失敗しました");
  }
}
async function getTweetAndSave(
  twitterUid: string,
  uid: string,
  next_id?: number
) {
  let params;
  if (next_id) {
    params = {
      user_id: twitterUid,
      max_id: next_id, // ここが空だと何も返ってこないため、冗長的な記述
      count: 200, // 一回のリクエストで最大200件まで取得できます
      exclude_replies: true,
      include_rts: false,
    };
  } else {
    params = {
      user_id: twitterUid,
      count: 200,
      exclude_replies: true,
      include_rts: false,
    };
  }
  await twitterClient
    .get("statuses/user_timeline", params)
    .then(async (tweets) => {
      const nextTweetId = +tweets[tweets.length - 1]?.id_str - 1; //　最後のTweetIdの一つ前のIdにする
      const batch = db.batch();
      tweets.forEach(
        (tweetContent: {
          id_str: string;
          favorite_count: number;
          retweeted: boolean;
        }) => {
          const tweetId = tweetContent.id_str; // ツイートのIdは.idでも取れるが、JavaScriptによって数値が勝手にまるめられてしまうため、stringの方を取得
          const likeCount = tweetContent.favorite_count;
          const tweetRef = db.doc(`private/${uid}/tweets/${tweetId}`);
          batch.set(tweetRef, {
            tweetId,
            likeCount,
          });
        }
      );
      await batch.commit(); // バッチ処理で書き込む
      if (tweets.length) {
        // 返ってきたツイートの配列が空でなければ次を取得
        await getTweetAndSave(twitterClient, twitterUid, uid, nextTweetId);
      }
      return;
    })
    .catch((error) => {
      throw error;
    });
}

特に、max_id の記述が詰まりました。max_id は指定されたツイート Id より過去のツイートを取得できるのですが、API の仕様？によって max_id を指定しているのに空だと空の配列が返ってきます。そのため、次の nextTweetId があれば max_id を指定する形にしています。また、API の仕様で次のリクエストでは max_id で指定したツイートから取得するため、最後のツイート Id を指定するとそのツイートを重複して取得することになります。そのため、数値に-1 することで、次に取得するべきツイート Id を指定しています。

コード

twitter-project/functions/src/user.function.ts こうリファクタリングした方がよいなどご指摘をお待ちしております。