Merge pull request #1 from DGGua/everyday

feat: add everyday problem

Author: DGGua

.vscode/settings.json

@@ -10,5 +10,12 @@
         ".vscode-test": true
     },
     "tslint.autoFixOnSave": true,
-    "tslint.ignoreDefinitionFiles": true
-}
+    "tslint.ignoreDefinitionFiles": true,
+    "editor.defaultFormatter": null,
+    "[typescript]": {
+        "editor.defaultFormatter": "vscode.typescript-language-features"
+    },
+    "[json]": {
+        "editor.defaultFormatter": "vscode.json-language-features"
+    }
+}

CHANGELOG.md

@@ -3,6 +3,12 @@ All notable changes to the "leetcode" extension will be documented in this file.
 
 Check [Keep a Changelog](http://keepachangelog.com/) for recommendations on how to structure this file.
 
+## [0.19.0]
+
+### Added
+- Add `Daily` category for current month daily problems.
+- Add `pickDaily` command for quickly pick daily command.
+
 ## [0.18.0]
 ### Added
 - Add `star` command in shortcuts [PR#601](https://github.com/LeetCode-OpenSource/vscode-leetcode/pull/601)

README.md

@@ -1,3 +1,13 @@
+# 🐱🐱🐱 this repository is forked from [vscode-leetcode](https://github.com/LeetCode-OpenSource/vscode-leetcode)
+
+This repository is forked for personal needs. I do not want merge it to the origin repository, for my irregular development. I made some features, if you like, please give us a ⭐Star!
+
+### Features
+
+- daily problems. easily use command or choose from sidebar to get problem today!
+
+below are the origin README.md
+
 # LeetCode
 
 > Solve LeetCode problems in VS Code
@@ -23,6 +33,7 @@
 - English Document | [中文文档](https://github.com/LeetCode-OpenSource/vscode-leetcode/blob/master/docs/README_zh-CN.md)
 
 ## ❗️ Attention ❗️- Workaround to login to LeetCode endpoint
+
 > Note: If you are using `leetcode.cn`, you can just ignore this section.
 
 Recently we observed that [the extension cannot login to leetcode.com endpoint anymore](https://github.com/LeetCode-OpenSource/vscode-leetcode/issues/478). The root cause of this issue is that leetcode.com changed its login mechanism and so far there is no ideal way to fix that issue.
@@ -32,9 +43,10 @@ Thanks for [@yihong0618](https://github.com/yihong0618) provided a workaround wh
 > Note: If you want to use third-party login(**Recommended**), please make sure your account has been connected to the third-party. If you want to use `Cookie` login, click [here](https://github.com/LeetCode-OpenSource/vscode-leetcode/issues/478#issuecomment-564757098) to see the steps.
 
 ## Requirements
+
 - [VS Code 1.30.1+](https://code.visualstudio.com/)
 - [Node.js 10+](https://nodejs.org)
-    > NOTE: Please make sure that `Node` is in your `PATH` environment variable. You can also use the setting `leetcode.nodePath` to specify the location of your `Node.js` executable.
+  > NOTE: Please make sure that `Node` is in your `PATH` environment variable. You can also use the setting `leetcode.nodePath` to specify the location of your `Node.js` executable.
 
 ## Quick Start
 
@@ -43,6 +55,7 @@ Thanks for [@yihong0618](https://github.com/yihong0618) provided a workaround wh
 ## Features
 
 ### Sign In/Out
+
 <p align="center">
   <img src="https://raw.githubusercontent.com/LeetCode-OpenSource/vscode-leetcode/master/docs/imgs/sign_in.png" alt="Sign in" />
 </p>
@@ -52,16 +65,19 @@ Thanks for [@yihong0618](https://github.com/yihong0618) provided a workaround wh
 - You can also use the following command to sign in/out:
   - **LeetCode: Sign in**
   - **LeetCode: Sign out**
+
 ---
 
 ### Switch Endpoint
+
 <p align="center">
   <img src="https://raw.githubusercontent.com/LeetCode-OpenSource/vscode-leetcode/master/docs/imgs/endpoint.png" alt="Switch Endpoint" />
 </p>
 
 - By clicking the button ![btn_endpoint](https://raw.githubusercontent.com/LeetCode-OpenSource/vscode-leetcode/master/docs/imgs/btn_endpoint.png) at the **explorer's navigation bar**, you can switch between different endpoints.
 
 - The supported endpoints are:
+
   - **leetcode.com**
   - **leetcode.cn**
 
@@ -70,14 +86,15 @@ Thanks for [@yihong0618](https://github.com/yihong0618) provided a workaround wh
 ---
 
 ### Pick a Problem
+
 <p align="center">
   <img src="https://raw.githubusercontent.com/LeetCode-OpenSource/vscode-leetcode/master/docs/imgs/pick_problem.png" alt="Pick a Problem" />
 </p>
 
 - Directly click on the problem or right click the problem in the `LeetCode Explorer` and select `Preview Problem` to see the problem description.
 - Select `Show Problem` to directly open the file with the problem description.
 
-  > Note：You can specify the path of the workspace folder to store the problem files by updating the setting `leetcode.workspaceFolder`. The default value is：**$HOME/.leetcode/**.
+  > Note：You can specify the path of the workspace folder to store the problem files by updating the setting `leetcode.workspaceFolder`. The default value is：**\$HOME/.leetcode/**.
 
   > You can specify whether including the problem description in comments or not by updating the setting `leetcode.showCommentDescription`.
 
@@ -86,11 +103,13 @@ Thanks for [@yihong0618](https://github.com/yihong0618) provided a workaround wh
 ---
 
 ### Editor Shortcuts
+
 <p align="center">
   <img src="https://raw.githubusercontent.com/LeetCode-OpenSource/vscode-leetcode/master/docs/imgs/shortcuts.png" alt="Editor Shortcuts" />
 </p>
 
 - The extension supports 5 editor shortcuts (aka Code Lens):
+
   - `Submit`: Submit your answer to LeetCode.
   - `Test`: Test your answer with customized test cases.
   - `Star/Unstar`: Star or unstar the current problem.
@@ -102,6 +121,7 @@ Thanks for [@yihong0618](https://github.com/yihong0618) provided a workaround wh
 ---
 
 ### Search problems by Keywords
+
 <p align="center">
   <img src="https://raw.githubusercontent.com/LeetCode-OpenSource/vscode-leetcode/master/docs/imgs/search.png" alt="Search problems by Keywords" />
 </p>
@@ -111,13 +131,13 @@ Thanks for [@yihong0618](https://github.com/yihong0618) provided a workaround wh
 ---
 
 ### Manage Session
+
 <p align="center">
   <img src="https://raw.githubusercontent.com/LeetCode-OpenSource/vscode-leetcode/master/docs/imgs/session.png" alt="Manage Session" />
 </p>
 
 - To manage your LeetCode sessions, just clicking the `LeetCode: ***` at the bottom of the status bar. You can **switch** between sessions or **create**, **delete** a session.
 
-
 ## Settings
 
 | Setting Name                      | Description                                                                                                                                                                                                                                                   | Default Value      |

package.json

@@ -1,20 +1,19 @@
 {
-    "name": "vscode-leetcode",
-    "displayName": "LeetCode",
-    "description": "Solve LeetCode problems in VS Code",
-    "version": "0.18.1",
+    "name": "vscode-leetcode-dggua",
+    "displayName": "LeetCodeEveryday",
+    "description": "Solve LeetCode problems in VS Code EveryDay",
+    "version": "0.19.0",
     "author": "Sheng Chen",
-    "publisher": "LeetCode",
+    "publisher": "dggua",
     "license": "MIT",
     "icon": "resources/LeetCode.png",
     "engines": {
         "vscode": "^1.57.0"
     },
     "repository": {
         "type": "git",
-        "url": "https://github.com/LeetCode-OpenSource/vscode-leetcode"
+        "url": "https://github.com/dggua/vscode-leetcode"
     },
-    "homepage": "https://github.com/LeetCode-OpenSource/vscode-leetcode/blob/master/README.md",
     "categories": [
         "Other",
         "Snippets"
@@ -32,6 +31,7 @@
         "onCommand:leetcode.manageSessions",
         "onCommand:leetcode.refreshExplorer",
         "onCommand:leetcode.pickOne",
+        "onCommand:leetcode.pickToday",
         "onCommand:leetcode.showProblem",
         "onCommand:leetcode.previewProblem",
         "onCommand:leetcode.searchProblem",
@@ -82,6 +82,11 @@
                 "title": "Pick One",
                 "category": "LeetCode"
             },
+            {
+                "command": "leetcode.pickToday",
+                "title": "Pick Problem Today",
+                "category": "LeetCode"
+            },
             {
                 "command": "leetcode.showProblem",
                 "title": "Show Problem",
@@ -709,7 +714,8 @@
         "vscode:prepublish": "npm run compile",
         "compile": "tsc -p ./",
         "watch": "tsc -watch -p ./",
-        "lint": "tslint --project tsconfig.json -e src/*.d.ts -t verbose"
+        "lint": "tslint --project tsconfig.json -e src/*.d.ts -t verbose",
+        "publish": "vsce package"
     },
     "devDependencies": {
         "@types/fs-extra": "^9.0.11",
@@ -723,6 +729,7 @@
         "typescript": "^4.3.2"
     },
     "dependencies": {
+        "axios": "^0.27.2",
         "fs-extra": "^10.0.0",
         "highlight.js": "^10.7.2",
         "lodash": "^4.17.21",

src/commands/list.ts

@@ -1,10 +1,11 @@
 // Copyright (c) jdneo. All rights reserved.
 // Licensed under the MIT license.
 
+import axios from "axios"
 import * as vscode from "vscode";
 import { leetCodeExecutor } from "../leetCodeExecutor";
 import { leetCodeManager } from "../leetCodeManager";
-import { IProblem, ProblemState, UserStatus } from "../shared";
+import { DailyProblem, DailyProblemSet, IProblem, ProblemState, UserStatus } from "../shared";
 import * as settingUtils from "../utils/settingUtils";
 import { DialogType, promptForOpenOutputChannel } from "../utils/uiUtils";
 
@@ -45,6 +46,32 @@ export async function listProblems(): Promise<IProblem[]> {
     }
 }
 
+export async function listDailyProblems(context: vscode.ExtensionContext): Promise<DailyProblem[]> {
+    try {
+        const cookie = context.globalState.get<string>("cookie")
+        if (leetCodeManager.getStatus() === UserStatus.SignedOut || cookie === undefined) {
+            return [];
+        }
+        const date = new Date();
+        const res = await axios.post<DailyProblemSet>(
+            "https://leetcode.cn/graphql/", {
+            "query": "\n    query dailyQuestionRecords($year: Int!, $month: Int!) {\n  dailyQuestionRecords(year: $year, month: $month) {\n    date\n    userStatus\n    question {\n      questionFrontendId\n      title\n      titleSlug\n      translatedTitle\n    }\n  }\n}\n    ",
+            "variables": {
+                "year": date.getFullYear(),
+                "month": date.getMonth() + 1,
+            }
+        }, {
+            headers: {
+                cookie,
+            },
+        })
+        return res.data.data.dailyQuestionRecords;
+    } catch (e) {
+        await promptForOpenOutputChannel("Failed to list daily Problems. Please open the output channel for details.", DialogType.error);
+        return []
+    }
+}
+
 function parseProblemState(stateOutput: string): ProblemState {
     if (!stateOutput) {
         return ProblemState.Unknown;

src/commands/show.ts

@@ -52,6 +52,18 @@ export async function pickOne(): Promise<void> {
     await showProblemInternal(randomProblem);
 }
 
+export async function pickToday(context: vscode.ExtensionContext): Promise<void> {
+    const dailyProblems = await list.listDailyProblems(context);
+    const problemToday = dailyProblems.find(problem => Number.parseInt(problem.date.split("-")[2]) == new Date().getDate())
+    const problems: IProblem[] = await list.listProblems();
+    const targetProblem = problems.find(problem => problem.id == problemToday?.question.questionFrontendId)
+    if (!targetProblem) {
+        vscode.window.showErrorMessage("Get problem today failed.");
+        return;
+    }
+    await showProblemInternal(targetProblem)
+}
+
 export async function showProblem(node?: LeetCodeNode): Promise<void> {
     if (!node) {
         return;

src/explorer/LeetCodeNode.ts

@@ -6,7 +6,7 @@ import { IProblem, ProblemState } from "../shared";
 
 export class LeetCodeNode {
 
-    constructor(private data: IProblem, private isProblemNode: boolean = true) { }
+    constructor(protected data: IProblem, private isProblemNode: boolean = true) { }
 
     public get locked(): boolean {
         return this.data.locked;
@@ -47,6 +47,10 @@ export class LeetCodeNode {
         return this.isProblemNode;
     }
 
+    public get dataClone(): IProblem {
+        return Object.assign({}, this.data)
+    }
+
     public get previewCommand(): Command {
         return {
             title: "Preview Problem",
@@ -67,5 +71,11 @@ export class LeetCodeNode {
             query: `difficulty=${this.difficulty}`,
         });
     }
+}
 
+export class LeetCodeDailyNode extends LeetCodeNode {
+
+    constructor(node: LeetCodeNode, public date: Date) {
+        super(node.dataClone);
+    }
 }

src/explorer/LeetCodeTreeDataProvider.ts

@@ -7,7 +7,7 @@ import * as vscode from "vscode";
 import { leetCodeManager } from "../leetCodeManager";
 import { Category, defaultProblem, ProblemState } from "../shared";
 import { explorerNodeManager } from "./explorerNodeManager";
-import { LeetCodeNode } from "./LeetCodeNode";
+import { LeetCodeDailyNode, LeetCodeNode } from "./LeetCodeNode";
 
 export class LeetCodeTreeDataProvider implements vscode.TreeDataProvider<LeetCodeNode> {
 
@@ -46,7 +46,11 @@ export class LeetCodeTreeDataProvider implements vscode.TreeDataProvider<LeetCod
         }
 
         return {
-            label: element.isProblem ? `[${element.id}] ${element.name}` : element.name,
+            label: element.isProblem ?
+                `[${element instanceof LeetCodeDailyNode ?
+                    `${element.date.getMonth()}-${element.date.getDate()}` :
+                    element.id}] ${element.name}` :
+                element.name,
             tooltip: this.getSubCategoryTooltip(element),
             collapsibleState: element.isProblem ? vscode.TreeItemCollapsibleState.None : vscode.TreeItemCollapsibleState.Collapsed,
             iconPath: this.parseIconPathFromProblemState(element),
@@ -69,6 +73,8 @@ export class LeetCodeTreeDataProvider implements vscode.TreeDataProvider<LeetCod
             return explorerNodeManager.getRootNodes();
         } else {
             switch (element.id) { // First-level
+                case Category.Daily:
+                    return explorerNodeManager.getDailyNodes();
                 case Category.All:
                     return explorerNodeManager.getAllNodes();
                 case Category.Favorite: