fix minor bugs

Author: Sorosliu1029

jupyterlab_leetcode/handlers/cookie_handler.py

@@ -17,8 +17,9 @@ def get(self):
             cookie = get_leetcode_cookie(
                 browser, self.settings, self.request.headers.get("User-Agent", "")
             )
-            self.set_cookie("leetcode_browser", browser, expires=cookie["expires"])
+            if not cookie["expired"] and cookie["expires"]:
+                self.set_cookie("leetcode_browser", browser, expires=cookie["expires"])
             self.finish(json.dumps(cookie))
         except Exception as e:
             self.set_status(400)
-            self.finish(json.dumps({"message": str(e.args)}))
+            self.finish(json.dumps({"message": str(e)}))

jupyterlab_leetcode/utils/utils.py

@@ -55,11 +55,9 @@ def get_leetcode_cookie(browser: str, settings: dict[str, Any], ua: str):
     try:
         cj = BROWSER_COOKIE_METHOD_MAP[browser](domain_name="leetcode.com")
     except browser_cookie3.BrowserCookieError as e:
-        raise Exception(
-            "Failed to retrieve cookies. Maybe not installed the browser?"
-        ) from e
+        raise Exception("Failed to retrieve cookies. Maybe not installed the browser?")
     except Exception as e:
-        raise Exception(f"An error occurred: {str(e)}") from e
+        raise Exception(f"An error occurred: {str(e)}")
 
     cookie_session = first(cj, lambda c: c.name == "LEETCODE_SESSION")
     cookie_csrf = first(cj, lambda c: c.name == "csrftoken")
@@ -70,7 +68,9 @@ def get_leetcode_cookie(browser: str, settings: dict[str, Any], ua: str):
     )
     checked = exist and not expired
 
-    expires = cast(Cookie, cookie_session).expires
+    expires = None
+    if checked:
+        expires = cast(Cookie, cookie_session).expires
 
     settings.update(
         leetcode_headers=HTTPHeaders(

src/components/Actions.tsx

@@ -50,7 +50,7 @@ const Actions: React.FC<{
           {Data.map(item => (
             <Tooltip label={item.label} key={item.label}>
               <Anchor
-                size="xs"
+                size="sm"
                 target="_blank"
                 underline="never"
                 href={item.href}

src/components/BrowserMenu.tsx

@@ -104,8 +104,8 @@ const BrowserMenu: React.FC<{
       .then(resp => {
         if (!resp['checked']) {
           Notification.error(
-            `Failed to check cookie for ${browser}. Please ensure you are logged in to LeetCode in ${browser}.`,
-            { autoClose: 3000 }
+            `Failed to check cookie for ${browser}. Have you logged in LeetCode in ${browser}?`,
+            { autoClose: 5000 }
           );
           return;
         }
@@ -134,7 +134,7 @@ const BrowserMenu: React.FC<{
         </Button>
       </Menu.Target>
       <Menu.Dropdown>
-        <Menu.Label>With LeetCode logged in</Menu.Label>
+        <Menu.Label>Where LeetCode logged in:</Menu.Label>
         {Browsers.map(({ name, icon }) => (
           <Menu.Item
             key={name}

src/components/QuestionCompanyFilter.tsx

@@ -5,8 +5,10 @@ import classes from '../styles/Filter.module.css';
 
 const CheckedIcon = <IconCheck size={12} stroke={1.5} />;
 
+// TODO: fill data
 const Data = ['facebook', 'google'];
 
+// TODO: show 'wont work if not premium'
 const renderMultiSelectOption: MultiSelectProps['renderOption'] = ({
   option,
   checked

src/components/QuestionTopicFilter.tsx

@@ -5,6 +5,7 @@ import classes from '../styles/Filter.module.css';
 
 const CheckedIcon = <IconCheck size={12} stroke={1.5} />;
 
+// TODO: fill data
 const Data = ['array', 'hash-table'];
 
 const renderMultiSelectOption: MultiSelectProps['renderOption'] = ({

src/components/SubmissionCalendar.tsx

@@ -13,7 +13,7 @@ const dates = Array(ShowWeeks * 7 + today.getUTCDay() + 1) // ensure table colum
   .map((_, i) => new Date(today.getTime() - i * ADay));
 const months = Array.from(
   new Map(
-    dates.map(d => [d.getUTCFullYear() * 100 + d.getMonth(), d])
+    dates.map(d => [d.getUTCFullYear() * 100 + d.getUTCMonth(), d])
   ).entries()
 ).sort(([a], [b]) => a - b);
 const monthHead = (

src/widget.tsx

@@ -19,7 +19,7 @@ export class JupyterMainWidget extends ReactWidget {
     this.id = 'JupyterlabLeetcodeWidget';
     this.addClass('jupyterlab-leetcode-widget');
     this.docManager = docManager;
-    // TODO: fix mantine override body line-height
+    // FIXME: fix mantine override body line-height
     this.theme = createTheme({});
   }