From 4492974ac6de21a8bd7dd869b7a46f6d78281340 Mon Sep 17 00:00:00 2001
From: "sir.kokabi" <sir.koakbi@gmail.com>
Date: Fri, 28 Jun 2024 15:22:26 +0330
Subject: [PATCH] add profile note feature

---
 src/contents/ProfileNote.tsx      | 100 ++++++++++++++++++++++++++++++
 src/contents/ProfileNoteAlert.tsx |  45 ++++++++++++++
 src/storage.ts                    |   9 +++
 src/utils.js                      |  21 ++++++-
 4 files changed, 173 insertions(+), 2 deletions(-)
 create mode 100644 src/contents/ProfileNote.tsx
 create mode 100644 src/contents/ProfileNoteAlert.tsx

diff --git a/src/contents/ProfileNote.tsx b/src/contents/ProfileNote.tsx
new file mode 100644
index 0000000..7a1b1af
--- /dev/null
+++ b/src/contents/ProfileNote.tsx
@@ -0,0 +1,100 @@
+import cssText from "data-text:~contents/styles.css";
+import type { PlasmoCSConfig, PlasmoGetInlineAnchor } from "plasmo";
+import React, { useState, useEffect, useCallback } from "react";
+import * as utils from "~utils";
+
+export const config: PlasmoCSConfig = {
+  matches: ["https://twitter.com/*", "https://x.com/*"]
+};
+
+export const getInlineAnchor: PlasmoGetInlineAnchor = () =>
+  document.querySelector(`div[data-testid='UserName'] > div > div`);
+
+export const getStyle = () => {
+  const style = document.createElement("style");
+  style.textContent = cssText;
+  return style;
+};
+
+const ProfileNote = () => {
+  const [note, setNote] = useState("");
+  const [originalNote, setOriginalNote] = useState("");
+  const [isPersian, setIsPersian] = useState(false);
+  const [showToast, setShowToast] = useState(false);
+  const [remainingLength, setRemainingLength] = useState(120);
+  const username = window.location.toString().split('/').pop();
+
+  useEffect(() => {
+    utils.getProfileNote(username).then((value) => {
+      if (value !== undefined) {
+        setNote(value);
+        setOriginalNote(value);
+        setIsPersian(utils.isPersian(value));
+        setRemainingLength(120 - value.length);
+      }
+    });
+  }, [username]);
+
+  const saveNote = useCallback(async () => {
+    if (note.trim() === '')  await utils.deleteProfileNote(username);
+    else {
+      utils.saveProfileNote(username, note)
+    }
+    setShowToast(true);
+    setOriginalNote(note);
+    setTimeout(() => {
+      setShowToast(false);
+    }, 800);
+
+  }, [note, username]);
+
+  const handleChange = (e) => {
+    const newNote = e.target.value;
+    setNote(newNote);
+    setIsPersian(utils.isPersian(newNote));
+    setRemainingLength(120 - newNote.length);
+  };
+
+  const handleKeyDown = (e) => {
+    if (e.key === "Enter") {
+      e.preventDefault();
+      saveNote();
+    }
+  };
+
+  const isNoteChanged = note !== originalNote;
+
+  return (
+    <div className="relative w-full">
+      <textarea
+        className="w-full shadow-[0px_2px_3px_-1px_rgba(0,0,0,0.1),0px_1px_0px_0px_rgba(25,28,33,0.02),0px_0px_0px_1px_rgba(25,28,33,0.08)] p-3 rounded-md mt-2"
+        style={{ direction: isPersian ? "rtl" : "ltr", border: "none" }}
+        value={note}
+        maxLength={120}
+        onChange={handleChange}
+        onKeyDown={handleKeyDown}
+        placeholder="Add a short note"
+      />
+      {isNoteChanged && (
+        <>
+          <button
+            className={`mt-2 px-4 py-1 bg-[#299cee] text-white rounded ${isPersian ? 'mr-auto' : 'ml-auto'} block`}
+            onClick={saveNote}
+          >
+            Save
+          </button>
+          <div className={`absolute bottom-2 ${isPersian ? 'right-2' : 'left-2'} text-xs px-2 py-1`}>
+            {remainingLength}
+          </div>
+        </>
+      )}
+      {showToast && (
+        <div className={`absolute bottom-2 ${isPersian ? 'left-2' : 'right-2'} text-xs px-2 py-1`}>
+          Saved
+        </div>
+      )}
+    </div>
+  );
+};
+
+export default ProfileNote;
diff --git a/src/contents/ProfileNoteAlert.tsx b/src/contents/ProfileNoteAlert.tsx
new file mode 100644
index 0000000..5c8d609
--- /dev/null
+++ b/src/contents/ProfileNoteAlert.tsx
@@ -0,0 +1,45 @@
+import cssText from "data-text:~contents/styles.css";
+import type { PlasmoCSConfig, PlasmoGetInlineAnchorList } from "plasmo";
+import React, { useState, useEffect, useCallback } from "react";
+import * as utils from "~utils";
+
+export const config: PlasmoCSConfig = {
+  matches: ["https://twitter.com/*", "https://x.com/*"]
+};
+
+export const getInlineAnchorList: PlasmoGetInlineAnchorList = () =>
+  document.querySelectorAll(`div[data-testid="User-Name"]`);
+
+export const getStyle = () => {
+  const style = document.createElement("style");
+  style.textContent = cssText;
+  return style;
+};
+
+const truncateString = (str = '', maxLength = 50) =>
+  str.length > maxLength ? `${str.slice(0, maxLength)}…` : str;
+
+const ProfileNoteAlert = ({anchor}) => {
+  const [note, setNote] = useState("");
+  const [isPersian, setIsPersian] = useState(false);
+
+  const username = anchor.element.querySelector('div[data-testid="User-Name"] > div:nth-child(2) > div > div').textContent.replace("@","");
+
+  useEffect(() => {
+    utils.getProfileNote(username).then((value) => {
+      if (typeof value === 'string' && value.trim() !== ''){
+        setNote(value);
+        setIsPersian(utils.isPersian(value))      
+      }          
+    });
+  }, [anchor]);
+
+
+  return (
+    <span title={note} style={{ direction: isPersian ? "rtl" : "ltr", border: "none" }} className="font-bold text-sm">
+      {truncateString(note, 50)}
+    </span>
+  );
+};
+
+export default ProfileNoteAlert;
diff --git a/src/storage.ts b/src/storage.ts
index 45bec41..e055857 100644
--- a/src/storage.ts
+++ b/src/storage.ts
@@ -2,6 +2,15 @@ import { Storage } from "@plasmohq/storage";
 
 const storage = new Storage();
 
+export const readStorage = async (key:string)=>{
+  return storage.get(key);
+}
+
+export const deleteStorageKey = async (key:string)=>{
+  await storage.remove(key);
+}
+
+
 export const readStorageAsBoolean = async (key:string)=>{
     const value = await storage.get(key);    
     return value=="true";
diff --git a/src/utils.js b/src/utils.js
index f5f5c88..7bac70b 100644
--- a/src/utils.js
+++ b/src/utils.js
@@ -5,7 +5,7 @@ import persianRex from "persian-rex"
 import rtl from "rtl-detect"
 import Virastar from "virastar"
 
-import { readStorageAsList } from "./storage"
+import { readStorageAsList, readStorage, writeStorage, deleteStorageKey } from "./storage"
 
 var virastar = new Virastar({ cleanup_begin_and_end: false })
 
@@ -20,6 +20,7 @@ function cleanupText(text) {
 }
 
 function isPersian(text) {
+  if (!text) return false;  
   var totalCharacters = text.length
 
   var persianCharacters = 0
@@ -97,6 +98,19 @@ async function hasMuttedText(tweetText) {
   })
 }
 
+async function getProfileNote(username) {
+  const profileNote = await readStorage(username)
+  return profileNote 
+}
+
+async function saveProfileNote(username, note) {
+  await writeStorage(username, note)   
+}
+
+async function deleteProfileNote(username) {
+  await deleteStorageKey(username)   
+}
+
 function evaluateXpath(xpath) {
   return document.evaluate(
     xpath,
@@ -198,5 +212,8 @@ export {
   evaluateXpath,
   bgColorForCurrentTheme,
   getCurrentTheme,
-  getSimplifiedVersion
+  getSimplifiedVersion,
+  getProfileNote,
+  saveProfileNote,
+  deleteProfileNote
 }