login

Author: shuzijun

package.json

@@ -13,6 +13,7 @@
     "@jetbrains/logos": "^2.2.7",
     "@plasmohq/storage": "^0.12.2",
     "antd": "^4.23.5",
+    "crypto-js": "^4.1.1",
     "react": "18.2.0",
     "react-dom": "18.2.0"
   },
@@ -23,7 +24,7 @@
     "@types/node": "18.8.3",
     "@types/react": "18.0.21",
     "@types/react-dom": "18.0.6",
-    "plasmo": "0.57.1",
+    "plasmo": "0.57.2",
     "prettier": "2.7.1",
     "typescript": "4.8.4"
   },
@@ -34,6 +35,8 @@
       "https://leetcode.com/*",
       "https://codetop.cc/*"
     ],
-    "permissions": []
+    "permissions": [
+      "cookies"
+    ]
   }
 }

pnpm-lock.yaml

@@ -0,0 +1,15 @@
+import CryptoJS from 'crypto-js'
+console.log("HELLO WORLD")
+
+chrome.runtime.onMessage.addListener((message, sender, sendResponse) => {
+    if (message === 'get-user-data') {
+        chrome.cookies.getAll({domain: sender.origin.replace("https://", "")}).then((cookies) => {
+            const cookieString = cookies.map(c => c.name + "=" + c.value).join(';');
+            console.log(cookieString);
+            const encodedWord = CryptoJS.enc.Utf8.parse(cookieString);
+            const encoded = CryptoJS.enc.Base64.stringify(encodedWord);
+            sendResponse(encoded);
+        });
+    }
+    return true
+});

src/background.ts

@@ -0,0 +1,56 @@
+import type {PlasmoContentScript, PlasmoGetInlineAnchor} from "plasmo"
+import {Button, Tooltip} from 'antd';
+import cssText from "data-text:~/src/index.less"
+
+import {useStorage} from "@plasmohq/storage/hook"
+
+import {GetLogoSize} from "~logo"
+
+export const config: PlasmoContentScript = {
+    matches: ["https://leetcode.cn/*"]
+}
+
+export const getStyle = () => {
+    const style = document.createElement("style")
+    style.textContent = cssText
+    return style
+}
+
+// Use this to optimize unmount lookups
+export const getShadowHostId = () => "leetcode-editor-login-id"
+
+export const getInlineAnchor: PlasmoGetInlineAnchor = () =>
+document.querySelector("a[href='/accounts/logout/']")
+
+
+const LeetcodeCnLogin = () => {
+
+    const [showIcon] = useStorage("ShowIcon", true)
+    const logo = GetLogoSize(14)
+    const url = GetUrl()
+
+    function click() {
+        chrome.runtime.sendMessage('get-user-data', (response) => {
+            window.open(url+response)
+        });
+
+    }
+
+    return (
+            <Tooltip title="IDE Login">
+                <Button hidden={!showIcon} type="text" onClick={click} icon={logo} target='_blank'> Login </Button>
+            </Tooltip>
+
+            )
+}
+
+
+const GetUrl = () => {
+    const [editor] = useStorage("Editor", "leetcode-editor-pro")
+    const [product] = useStorage("Product", "idea")
+    const [project] = useStorage("Project", "")
+
+    return "jetbrains://" + product + "/" + editor + "/login?project=" + project + "&token="
+}
+
+export default LeetcodeCnLogin

src/contents/leetcode-cn-login.tsx

@@ -0,0 +1,56 @@
+import type {PlasmoContentScript, PlasmoGetInlineAnchor} from "plasmo"
+import {Button, Tooltip} from 'antd';
+import cssText from "data-text:~/src/index.less"
+
+import {useStorage} from "@plasmohq/storage/hook"
+
+import {GetLogoSize} from "~logo"
+
+export const config: PlasmoContentScript = {
+    matches: ["https://leetcode.com/*"]
+}
+
+export const getStyle = () => {
+    const style = document.createElement("style")
+    style.textContent = cssText
+    return style
+}
+
+// Use this to optimize unmount lookups
+export const getShadowHostId = () => "leetcode-editor-login-id"
+
+export const getInlineAnchor: PlasmoGetInlineAnchor = () =>
+document.querySelector(".user-menu__JejQ > li:last-child ")
+
+
+const LeetcodeComLogin = () => {
+
+    const [showIcon] = useStorage("ShowIcon", true)
+    const logo = GetLogoSize(14)
+    const url = GetUrl()
+
+    function click() {
+        chrome.runtime.sendMessage('get-user-data', (response) => {
+            window.open(url+response)
+        });
+
+    }
+
+    return (
+            <Tooltip title="IDE Login">
+                <Button hidden={!showIcon} type="text" onClick={click} icon={logo} target='_blank'> Login </Button>
+            </Tooltip>
+
+            )
+}
+
+
+const GetUrl = () => {
+    const [editor] = useStorage("Editor", "leetcode-editor-pro")
+    const [product] = useStorage("Product", "idea")
+    const [project] = useStorage("Project", "")
+
+    return "jetbrains://" + product + "/" + editor + "/login?project=" + project + "&token="
+}
+
+export default LeetcodeComLogin