From 23b5baae1e31e73e2308fbc3ab21d1c6f1b30e83 Mon Sep 17 00:00:00 2001
From: dangered wolf <d@ngeredwolf.me>
Date: Thu, 13 Apr 2023 17:27:07 -0400
Subject: [PATCH] Add view counts

---
 src/api/status.ts           | 6 ++++++
 src/constants.ts            | 1 +
 src/helpers/author.ts       | 3 +++
 src/types/twitterTypes.d.ts | 9 +++++++++
 src/types/types.d.ts        | 1 +
 5 files changed, 20 insertions(+)

diff --git a/src/api/status.ts b/src/api/status.ts
index adf9f21..feb84bb 100644
--- a/src/api/status.ts
+++ b/src/api/status.ts
@@ -49,6 +49,12 @@ const populateTweetProperties = async (
   apiTweet.created_at = tweet.created_at;
   apiTweet.created_timestamp = new Date(tweet.created_at).getTime() / 1000;
 
+  if (tweet.ext_views?.state === 'EnabledWithCount') {
+    apiTweet.views = parseInt(tweet.ext_views.count || '0') ?? null;
+  } else {
+    apiTweet.views = null;
+  }
+
   if (tweet.lang !== 'unk') {
     apiTweet.lang = tweet.lang;
   } else {
diff --git a/src/constants.ts b/src/constants.ts
index 689571b..53a2957 100644
--- a/src/constants.ts
+++ b/src/constants.ts
@@ -26,6 +26,7 @@ export const Constants = {
     'cards_platform=Web-12',
     'include_cards=1',
     'include_ext_alt_text=true',
+    'include_ext_views=true',
     'include_quote_count=true',
     'include_reply_count=1',
     'tweet_mode=extended',
diff --git a/src/helpers/author.ts b/src/helpers/author.ts
index 18164e0..e2c55fe 100644
--- a/src/helpers/author.ts
+++ b/src/helpers/author.ts
@@ -14,6 +14,9 @@ export const getAuthorText = (tweet: APITweet): string | null => {
     if (tweet.likes > 0) {
       authorText += `${formatNumber(tweet.likes)} ❤️    `;
     }
+    if (tweet.views && tweet.views > 0) {
+      authorText += `${formatNumber(tweet.views)} 👁️    `;
+    }
     authorText = authorText.trim();
 
     return authorText;
diff --git a/src/types/twitterTypes.d.ts b/src/types/twitterTypes.d.ts
index 279eeff..022d384 100644
--- a/src/types/twitterTypes.d.ts
+++ b/src/types/twitterTypes.d.ts
@@ -128,6 +128,14 @@ type TweetEntities = {
   media?: TweetMedia[];
 };
 
+/* TODO: Are there more states for ext_views?
+   Legacy Tweets use Enabled but have no count, while newer tweets have EnabledWithCount
+   and count is populated with a string. */
+type ExtViews = {
+  state: 'Enabled' | 'EnabledWithCount',
+  count?: string
+}
+
 type TweetPartial = {
   card?: TweetCard;
   conversation_id_str: string;
@@ -135,6 +143,7 @@ type TweetPartial = {
   display_text_range: [number, number];
   entities: TweetEntities;
   extended_entities: TweetEntities;
+  ext_views?: ExtViews;
   favorite_count: number;
   in_reply_to_screen_name?: string;
   in_reply_to_status_id_str?: string;
diff --git a/src/types/types.d.ts b/src/types/types.d.ts
index 855ade1..21c3a60 100644
--- a/src/types/types.d.ts
+++ b/src/types/types.d.ts
@@ -116,6 +116,7 @@ interface APITweet {
   likes: number;
   retweets: number;
   replies: number;
+  views?: number | null;
 
   color: string;