diff --git a/Makefile b/Makefile index ecbfafa..2aac5d0 100644 --- a/Makefile +++ b/Makefile @@ -1,4 +1,7 @@ lint: golangci-lint run ./... -.PHONY: lint +problemset: + bash ./query-problem-set.sh > problemset.jsonl + +.PHONY: lint problemset diff --git a/create.sh b/create.sh index cd8458f..79603fe 100755 --- a/create.sh +++ b/create.sh @@ -1,21 +1,42 @@ #!/bin/bash -num=$1 +id=$1 -echo "$num" | grep -q "^[1-9][0-9]\{,4\}$" || { - echo "Err: no valid number given" >&2 +echo "$id" | grep -q "^[1-9][0-9]\{,4\}$" || { + echo "Err: no valid problem ID provided." >&2 exit 1 } -parent=$((num / 100)) -pdir="solutions/${parent}/q${num}" +parent=$((id / 100)) +pdir="solutions/${parent}/q${id}" sol_file="${pdir}/solution.go" if [ -f "$sol_file" ]; then exit 0 fi -echo "Creating template for question No. $num" >&2 +title_slug=$(jq -r "select(.id == $id) | .titleSlug" ./problemset.jsonl) +[ -n "$title_slug" ] || { + echo "Err: problem $id not found in the problem set." >&2 + exit 1 +} +problem_url="https://leetcode.com/problems/$title_slug/" + +echo -e "Creating template for problem \033[32m${id}: $title_slug\033[0m" >&2 mkdir -pv "$pdir" -echo "package q$num" > "$sol_file" -echo "Created $sol_file" +echo "// Package q$id implements a solution for $problem_url" > "$sol_file" +echo -e "package q$id\n" >> "$sol_file" + +echo "Created: $sol_file" + +echo "Fetching code snippet for Go ..." +curl -sSLf "${problem_url}description/" \ + -H 'accept: text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8' \ + -H 'accept-language: en-US,en;q=0.5' \ + -H 'user-agent: Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/145.0.0.0 Safari/537.36' \ + | grep -o 'NEXT_DATA[^>]*>.*' \ + | sed 's/^[^>]*>//;s/<\/script.*//' \ + | jq -r '[ .props.pageProps.dehydratedState.queries[].state.data.question.codeSnippets | select( . != null) | .[] | select(.langSlug == "golang") ][0].code' \ + >> "$sol_file" + +echo "Done." diff --git a/query-problem-set.sh b/query-problem-set.sh new file mode 100644 index 0000000..72fa3a4 --- /dev/null +++ b/query-problem-set.sh @@ -0,0 +1,37 @@ +#!/bin/bash +set -e + +query() { + page=$1 + offset=$((page * 100)) + + # shellcheck disable=SC2016 + curl -sSLf 'https://leetcode.com/graphql/' \ + --compressed \ + -H 'content-type: application/json' \ + --data-raw '{ + "query": "\n query problemsetQuestionListV2($filters: QuestionFilterInput, $limit: Int, $searchKeyword: String, $skip: Int, $sortBy: QuestionSortByInput, $categorySlug: String) {\n problemsetQuestionListV2(\n filters: $filters\n limit: $limit\n searchKeyword: $searchKeyword\n skip: $skip\n sortBy: $sortBy\n categorySlug: $categorySlug\n ) {\n questions {\n id\n titleSlug\n title\n translatedTitle\n questionFrontendId\n paidOnly\n difficulty\n topicTags {\n name\n slug\n nameTranslated\n }\n status\n isInMyFavorites\n frequency\n acRate\n contestPoint\n }\n totalLength\n finishedLength\n hasMore\n }\n}\n ", + "variables": { + "skip": '"$offset"', + "limit": 100, + "categorySlug": "all-code-essentials", + "filters": {"filterCombineType":"ALL","statusFilter":{"questionStatuses":[],"operator":"IS"},"difficultyFilter":{"difficulties":[],"operator":"IS"},"languageFilter":{"languageSlugs":[],"operator":"IS"},"topicFilter":{"topicSlugs":[],"operator":"IS"},"acceptanceFilter":{},"frequencyFilter":{},"frontendIdFilter":{},"lastSubmittedFilter":{},"publishedFilter":{},"companyFilter":{"companySlugs":[],"operator":"IS"},"positionFilter":{"positionSlugs":[],"operator":"IS"},"positionLevelFilter":{"positionLevelSlugs":[],"operator":"IS"},"contestPointFilter":{"contestPoints":[],"operator":"IS"},"premiumFilter":{"premiumStatus":[],"operator":"IS"}}, + "searchKeyword": "", + "sortBy": {"sortField":"CUSTOM","sortOrder":"ASCENDING"}, + "filtersV2": {"filterCombineType":"ALL","statusFilter":{"questionStatuses":[],"operator":"IS"},"difficultyFilter":{"difficulties":[],"operator":"IS"},"languageFilter":{"languageSlugs":[],"operator":"IS"},"topicFilter":{"topicSlugs":[],"operator":"IS"},"acceptanceFilter":{},"frequencyFilter":{},"frontendIdFilter":{},"lastSubmittedFilter":{},"publishedFilter":{},"companyFilter":{"companySlugs":[],"operator":"IS"},"positionFilter":{"positionSlugs":[],"operator":"IS"},"positionLevelFilter":{"positionLevelSlugs":[],"operator":"IS"},"contestPointFilter":{"contestPoints":[],"operator":"IS"},"premiumFilter":{"premiumStatus":[],"operator":"IS"}} + }, + "operationName": "problemsetQuestionListV2" + }' | jq -c '.data.problemsetQuestionListV2.questions[] | { id, title, titleSlug, difficulty, paidOnly, tags: [ .topicTags[] | .slug ] }' +} + +page=0 +while true; do + echo -ne " Querying page $((page + 1)) ...\r" >&2 + content=$(query $page) + lines=$(echo "$content" | wc -l) + echo "$content" + + [ "$lines" -lt 100 ] && break + page=$((page + 1)) +done | jq -s -c 'sort_by(.id) | .[]' +echo >&2