🔖 1.3.3

Bumps [nonebot-adapter-discord](https://github.com/nonebot/adapter-discord) from 0.1.7 to 0.1.8.
- [Release notes](https://github.com/nonebot/adapter-discord/releases)
- [Commits](https://github.com/nonebot/adapter-discord/compare/v0.1.7...v0.1.8)

---
updated-dependencies:
- dependency-name: nonebot-adapter-discord
  dependency-type: direct:development
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

.github/dependabot.yml

@@ -7,6 +7,6 @@ version: 2
 updates:
   - package-ecosystem: "pip" # See documentation for possible values
     directory: "/" # Location of package manifests
-    target-branch: "dev"
+    target-branch: "main"
     schedule:
       interval: "daily"

.github/workflows/Release.yml

@@ -8,18 +8,42 @@ on:
 jobs:
   release:
     runs-on: ubuntu-latest
+    permissions:
+      id-token: write
+      contents: write
     steps:
       - uses: actions/checkout@v3
+
+      - name: Install poetry
+        run: pipx install poetry
+        shell: bash
+
       - uses: actions/setup-python@v4
         with:
-          python-version: '3.10'
-      - name: Install Poetry
+          python-version: '3.11'
+          cache: "poetry"
+
+      - run: poetry install
         shell: bash
+
+      - name: Get Version
+        id: version
         run: |
-          pip install poetry
-      - name: Build
-        shell: bash
-        run: |
-          poetry install
-          poetry env use python
-          poetry publish --build -u ${{ secrets.USERNAME }} -p ${{ secrets.PASSWORD }} -n
+          echo "VERSION=$(poetry version -s)" >> $GITHUB_OUTPUT
+          echo "TAG_VERSION=${GITHUB_REF#refs/tags/}" >> $GITHUB_OUTPUT
+          echo "TAG_NAME=${GITHUB_REF#refs/tags/}" >> $GITHUB_OUTPUT
+
+      - name: Check Version
+        if: steps.version.outputs.VERSION != steps.version.outputs.TAG_VERSION
+        run: exit 1
+
+      - name: Build Package
+        run: poetry build
+
+      - name: Publish Package to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+
+      - name: Publish Package to GitHub Release
+        run: gh release create ${{ steps.version.outputs.TAG_NAME }} dist/*.tar.gz dist/*.whl -t "🔖 ${{ steps.version.outputs.TAG_NAME }}" --generate-notes
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

.gitignore

@@ -1,6 +1,22 @@
 .idea
 dist
-test*
+test_*
 Untitled*
 *copy*
 .vscode
+*dev*
+*_cache*
+*backup*
+*.pyc
+node_modules
+.prettier*
+package.json
+pnpm-lock.yaml
+*.drawio.svg
+package-lock.json
+*Zone.Identifier
+.env*
+bot.py
+TODO
+*.fish
+extracted_skin_mino_*

.pre-commit-config.yaml

@@ -0,0 +1,22 @@
+default_install_hook_types: [pre-commit, prepare-commit-msg]
+ci:
+    autofix_commit_msg: ':rotating_light: auto fix by pre-commit hooks'
+    autofix_prs: true
+    autoupdate_branch: master
+    autoupdate_schedule: monthly
+    autoupdate_commit_msg: ':arrow_up: auto update by pre-commit hooks'
+repos:
+    - repo: https://github.com/astral-sh/ruff-pre-commit
+      rev: v0.4.10
+      hooks:
+          - id: ruff
+            args: [--fix, --exit-non-zero-on-fix]
+            stages: [commit]
+          - id: ruff-format
+            stages: [commit]
+
+    - repo: https://github.com/nonebot/nonemoji
+      rev: v0.1.4
+      hooks:
+          - id: nonemoji
+            stages: [prepare-commit-msg]

.pylintrc

@@ -1,31 +0,0 @@
-[MAIN]
-# A comma-separated list of package or module names from where C extensions may
-# be loaded. Extensions are loading into the active Python interpreter and may
-# run arbitrary code.
-extension-pkg-allow-list=lxml, pydantic
-
-# A comma-separated list of package or module names from where C extensions may
-# be loaded. Extensions are loading into the active Python interpreter and may
-# run arbitrary code. (This is an alternative name to extension-pkg-allow-list
-# for backward compatibility.)
-extension-pkg-whitelist=lxml, pydantic
-
-# List of module names for which member attributes should not be checked
-# (useful for modules/projects where namespaces are manipulated during runtime
-# and thus existing member attributes cannot be deduced by static analysis). It
-# supports qualified module names, as well as Unix pattern matching.
-ignored-modules=ujson
-
-disable=
-    C0114, # missing-module-docstring
-
-[MESSAGES CONTROL]
-# Enable the message, report, category or checker with the given id(s). You can
-# either give multiple identifier separated by comma (,) or put this option
-# multiple time (only on the command line, not in the configuration file where
-# it should appear only once). See also the "--disable" option for examples.
-enable=c-extension-no-member
-
-[FORMAT]
-# Maximum number of characters on a single line.
-max-line-length=120

LICENSE

@@ -1,21 +1,661 @@
-MIT License
+                    GNU AFFERO GENERAL PUBLIC LICENSE
+                       Version 3, 19 November 2007
 
-Copyright (c) 2022 scdhh
+ Copyright (C) 2007 Free Software Foundation, Inc. <https://fsf.org/>
+ Everyone is permitted to copy and distribute verbatim copies
+ of this license document, but changing it is not allowed.
 
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
+                            Preamble
 
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
+  The GNU Affero General Public License is a free, copyleft license for
+software and other kinds of works, specifically designed to ensure
+cooperation with the community in the case of network server software.
 
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.
+  The licenses for most software and other practical works are designed
+to take away your freedom to share and change the works.  By contrast,
+our General Public Licenses are intended to guarantee your freedom to
+share and change all versions of a program--to make sure it remains free
+software for all its users.
+
+  When we speak of free software, we are referring to freedom, not
+price.  Our General Public Licenses are designed to make sure that you
+have the freedom to distribute copies of free software (and charge for
+them if you wish), that you receive source code or can get it if you
+want it, that you can change the software or use pieces of it in new
+free programs, and that you know you can do these things.
+
+  Developers that use our General Public Licenses protect your rights
+with two steps: (1) assert copyright on the software, and (2) offer
+you this License which gives you legal permission to copy, distribute
+and/or modify the software.
+
+  A secondary benefit of defending all users' freedom is that
+improvements made in alternate versions of the program, if they
+receive widespread use, become available for other developers to
+incorporate.  Many developers of free software are heartened and
+encouraged by the resulting cooperation.  However, in the case of
+software used on network servers, this result may fail to come about.
+The GNU General Public License permits making a modified version and
+letting the public access it on a server without ever releasing its
+source code to the public.
+
+  The GNU Affero General Public License is designed specifically to
+ensure that, in such cases, the modified source code becomes available
+to the community.  It requires the operator of a network server to
+provide the source code of the modified version running there to the
+users of that server.  Therefore, public use of a modified version, on
+a publicly accessible server, gives the public access to the source
+code of the modified version.
+
+  An older license, called the Affero General Public License and
+published by Affero, was designed to accomplish similar goals.  This is
+a different license, not a version of the Affero GPL, but Affero has
+released a new version of the Affero GPL which permits relicensing under
+this license.
+
+  The precise terms and conditions for copying, distribution and
+modification follow.
+
+                       TERMS AND CONDITIONS
+
+  0. Definitions.
+
+  "This License" refers to version 3 of the GNU Affero General Public License.
+
+  "Copyright" also means copyright-like laws that apply to other kinds of
+works, such as semiconductor masks.
+
+  "The Program" refers to any copyrightable work licensed under this
+License.  Each licensee is addressed as "you".  "Licensees" and
+"recipients" may be individuals or organizations.
+
+  To "modify" a work means to copy from or adapt all or part of the work
+in a fashion requiring copyright permission, other than the making of an
+exact copy.  The resulting work is called a "modified version" of the
+earlier work or a work "based on" the earlier work.
+
+  A "covered work" means either the unmodified Program or a work based
+on the Program.
+
+  To "propagate" a work means to do anything with it that, without
+permission, would make you directly or secondarily liable for
+infringement under applicable copyright law, except executing it on a
+computer or modifying a private copy.  Propagation includes copying,
+distribution (with or without modification), making available to the
+public, and in some countries other activities as well.
+
+  To "convey" a work means any kind of propagation that enables other
+parties to make or receive copies.  Mere interaction with a user through
+a computer network, with no transfer of a copy, is not conveying.
+
+  An interactive user interface displays "Appropriate Legal Notices"
+to the extent that it includes a convenient and prominently visible
+feature that (1) displays an appropriate copyright notice, and (2)
+tells the user that there is no warranty for the work (except to the
+extent that warranties are provided), that licensees may convey the
+work under this License, and how to view a copy of this License.  If
+the interface presents a list of user commands or options, such as a
+menu, a prominent item in the list meets this criterion.
+
+  1. Source Code.
+
+  The "source code" for a work means the preferred form of the work
+for making modifications to it.  "Object code" means any non-source
+form of a work.
+
+  A "Standard Interface" means an interface that either is an official
+standard defined by a recognized standards body, or, in the case of
+interfaces specified for a particular programming language, one that
+is widely used among developers working in that language.
+
+  The "System Libraries" of an executable work include anything, other
+than the work as a whole, that (a) is included in the normal form of
+packaging a Major Component, but which is not part of that Major
+Component, and (b) serves only to enable use of the work with that
+Major Component, or to implement a Standard Interface for which an
+implementation is available to the public in source code form.  A
+"Major Component", in this context, means a major essential component
+(kernel, window system, and so on) of the specific operating system
+(if any) on which the executable work runs, or a compiler used to
+produce the work, or an object code interpreter used to run it.
+
+  The "Corresponding Source" for a work in object code form means all
+the source code needed to generate, install, and (for an executable
+work) run the object code and to modify the work, including scripts to
+control those activities.  However, it does not include the work's
+System Libraries, or general-purpose tools or generally available free
+programs which are used unmodified in performing those activities but
+which are not part of the work.  For example, Corresponding Source
+includes interface definition files associated with source files for
+the work, and the source code for shared libraries and dynamically
+linked subprograms that the work is specifically designed to require,
+such as by intimate data communication or control flow between those
+subprograms and other parts of the work.
+
+  The Corresponding Source need not include anything that users
+can regenerate automatically from other parts of the Corresponding
+Source.
+
+  The Corresponding Source for a work in source code form is that
+same work.
+
+  2. Basic Permissions.
+
+  All rights granted under this License are granted for the term of
+copyright on the Program, and are irrevocable provided the stated
+conditions are met.  This License explicitly affirms your unlimited
+permission to run the unmodified Program.  The output from running a
+covered work is covered by this License only if the output, given its
+content, constitutes a covered work.  This License acknowledges your
+rights of fair use or other equivalent, as provided by copyright law.
+
+  You may make, run and propagate covered works that you do not
+convey, without conditions so long as your license otherwise remains
+in force.  You may convey covered works to others for the sole purpose
+of having them make modifications exclusively for you, or provide you
+with facilities for running those works, provided that you comply with
+the terms of this License in conveying all material for which you do
+not control copyright.  Those thus making or running the covered works
+for you must do so exclusively on your behalf, under your direction
+and control, on terms that prohibit them from making any copies of
+your copyrighted material outside their relationship with you.
+
+  Conveying under any other circumstances is permitted solely under
+the conditions stated below.  Sublicensing is not allowed; section 10
+makes it unnecessary.
+
+  3. Protecting Users' Legal Rights From Anti-Circumvention Law.
+
+  No covered work shall be deemed part of an effective technological
+measure under any applicable law fulfilling obligations under article
+11 of the WIPO copyright treaty adopted on 20 December 1996, or
+similar laws prohibiting or restricting circumvention of such
+measures.
+
+  When you convey a covered work, you waive any legal power to forbid
+circumvention of technological measures to the extent such circumvention
+is effected by exercising rights under this License with respect to
+the covered work, and you disclaim any intention to limit operation or
+modification of the work as a means of enforcing, against the work's
+users, your or third parties' legal rights to forbid circumvention of
+technological measures.
+
+  4. Conveying Verbatim Copies.
+
+  You may convey verbatim copies of the Program's source code as you
+receive it, in any medium, provided that you conspicuously and
+appropriately publish on each copy an appropriate copyright notice;
+keep intact all notices stating that this License and any
+non-permissive terms added in accord with section 7 apply to the code;
+keep intact all notices of the absence of any warranty; and give all
+recipients a copy of this License along with the Program.
+
+  You may charge any price or no price for each copy that you convey,
+and you may offer support or warranty protection for a fee.
+
+  5. Conveying Modified Source Versions.
+
+  You may convey a work based on the Program, or the modifications to
+produce it from the Program, in the form of source code under the
+terms of section 4, provided that you also meet all of these conditions:
+
+    a) The work must carry prominent notices stating that you modified
+    it, and giving a relevant date.
+
+    b) The work must carry prominent notices stating that it is
+    released under this License and any conditions added under section
+    7.  This requirement modifies the requirement in section 4 to
+    "keep intact all notices".
+
+    c) You must license the entire work, as a whole, under this
+    License to anyone who comes into possession of a copy.  This
+    License will therefore apply, along with any applicable section 7
+    additional terms, to the whole of the work, and all its parts,
+    regardless of how they are packaged.  This License gives no
+    permission to license the work in any other way, but it does not
+    invalidate such permission if you have separately received it.
+
+    d) If the work has interactive user interfaces, each must display
+    Appropriate Legal Notices; however, if the Program has interactive
+    interfaces that do not display Appropriate Legal Notices, your
+    work need not make them do so.
+
+  A compilation of a covered work with other separate and independent
+works, which are not by their nature extensions of the covered work,
+and which are not combined with it such as to form a larger program,
+in or on a volume of a storage or distribution medium, is called an
+"aggregate" if the compilation and its resulting copyright are not
+used to limit the access or legal rights of the compilation's users
+beyond what the individual works permit.  Inclusion of a covered work
+in an aggregate does not cause this License to apply to the other
+parts of the aggregate.
+
+  6. Conveying Non-Source Forms.
+
+  You may convey a covered work in object code form under the terms
+of sections 4 and 5, provided that you also convey the
+machine-readable Corresponding Source under the terms of this License,
+in one of these ways:
+
+    a) Convey the object code in, or embodied in, a physical product
+    (including a physical distribution medium), accompanied by the
+    Corresponding Source fixed on a durable physical medium
+    customarily used for software interchange.
+
+    b) Convey the object code in, or embodied in, a physical product
+    (including a physical distribution medium), accompanied by a
+    written offer, valid for at least three years and valid for as
+    long as you offer spare parts or customer support for that product
+    model, to give anyone who possesses the object code either (1) a
+    copy of the Corresponding Source for all the software in the
+    product that is covered by this License, on a durable physical
+    medium customarily used for software interchange, for a price no
+    more than your reasonable cost of physically performing this
+    conveying of source, or (2) access to copy the
+    Corresponding Source from a network server at no charge.
+
+    c) Convey individual copies of the object code with a copy of the
+    written offer to provide the Corresponding Source.  This
+    alternative is allowed only occasionally and noncommercially, and
+    only if you received the object code with such an offer, in accord
+    with subsection 6b.
+
+    d) Convey the object code by offering access from a designated
+    place (gratis or for a charge), and offer equivalent access to the
+    Corresponding Source in the same way through the same place at no
+    further charge.  You need not require recipients to copy the
+    Corresponding Source along with the object code.  If the place to
+    copy the object code is a network server, the Corresponding Source
+    may be on a different server (operated by you or a third party)
+    that supports equivalent copying facilities, provided you maintain
+    clear directions next to the object code saying where to find the
+    Corresponding Source.  Regardless of what server hosts the
+    Corresponding Source, you remain obligated to ensure that it is
+    available for as long as needed to satisfy these requirements.
+
+    e) Convey the object code using peer-to-peer transmission, provided
+    you inform other peers where the object code and Corresponding
+    Source of the work are being offered to the general public at no
+    charge under subsection 6d.
+
+  A separable portion of the object code, whose source code is excluded
+from the Corresponding Source as a System Library, need not be
+included in conveying the object code work.
+
+  A "User Product" is either (1) a "consumer product", which means any
+tangible personal property which is normally used for personal, family,
+or household purposes, or (2) anything designed or sold for incorporation
+into a dwelling.  In determining whether a product is a consumer product,
+doubtful cases shall be resolved in favor of coverage.  For a particular
+product received by a particular user, "normally used" refers to a
+typical or common use of that class of product, regardless of the status
+of the particular user or of the way in which the particular user
+actually uses, or expects or is expected to use, the product.  A product
+is a consumer product regardless of whether the product has substantial
+commercial, industrial or non-consumer uses, unless such uses represent
+the only significant mode of use of the product.
+
+  "Installation Information" for a User Product means any methods,
+procedures, authorization keys, or other information required to install
+and execute modified versions of a covered work in that User Product from
+a modified version of its Corresponding Source.  The information must
+suffice to ensure that the continued functioning of the modified object
+code is in no case prevented or interfered with solely because
+modification has been made.
+
+  If you convey an object code work under this section in, or with, or
+specifically for use in, a User Product, and the conveying occurs as
+part of a transaction in which the right of possession and use of the
+User Product is transferred to the recipient in perpetuity or for a
+fixed term (regardless of how the transaction is characterized), the
+Corresponding Source conveyed under this section must be accompanied
+by the Installation Information.  But this requirement does not apply
+if neither you nor any third party retains the ability to install
+modified object code on the User Product (for example, the work has
+been installed in ROM).
+
+  The requirement to provide Installation Information does not include a
+requirement to continue to provide support service, warranty, or updates
+for a work that has been modified or installed by the recipient, or for
+the User Product in which it has been modified or installed.  Access to a
+network may be denied when the modification itself materially and
+adversely affects the operation of the network or violates the rules and
+protocols for communication across the network.
+
+  Corresponding Source conveyed, and Installation Information provided,
+in accord with this section must be in a format that is publicly
+documented (and with an implementation available to the public in
+source code form), and must require no special password or key for
+unpacking, reading or copying.
+
+  7. Additional Terms.
+
+  "Additional permissions" are terms that supplement the terms of this
+License by making exceptions from one or more of its conditions.
+Additional permissions that are applicable to the entire Program shall
+be treated as though they were included in this License, to the extent
+that they are valid under applicable law.  If additional permissions
+apply only to part of the Program, that part may be used separately
+under those permissions, but the entire Program remains governed by
+this License without regard to the additional permissions.
+
+  When you convey a copy of a covered work, you may at your option
+remove any additional permissions from that copy, or from any part of
+it.  (Additional permissions may be written to require their own
+removal in certain cases when you modify the work.)  You may place
+additional permissions on material, added by you to a covered work,
+for which you have or can give appropriate copyright permission.
+
+  Notwithstanding any other provision of this License, for material you
+add to a covered work, you may (if authorized by the copyright holders of
+that material) supplement the terms of this License with terms:
+
+    a) Disclaiming warranty or limiting liability differently from the
+    terms of sections 15 and 16 of this License; or
+
+    b) Requiring preservation of specified reasonable legal notices or
+    author attributions in that material or in the Appropriate Legal
+    Notices displayed by works containing it; or
+
+    c) Prohibiting misrepresentation of the origin of that material, or
+    requiring that modified versions of such material be marked in
+    reasonable ways as different from the original version; or
+
+    d) Limiting the use for publicity purposes of names of licensors or
+    authors of the material; or
+
+    e) Declining to grant rights under trademark law for use of some
+    trade names, trademarks, or service marks; or
+
+    f) Requiring indemnification of licensors and authors of that
+    material by anyone who conveys the material (or modified versions of
+    it) with contractual assumptions of liability to the recipient, for
+    any liability that these contractual assumptions directly impose on
+    those licensors and authors.
+
+  All other non-permissive additional terms are considered "further
+restrictions" within the meaning of section 10.  If the Program as you
+received it, or any part of it, contains a notice stating that it is
+governed by this License along with a term that is a further
+restriction, you may remove that term.  If a license document contains
+a further restriction but permits relicensing or conveying under this
+License, you may add to a covered work material governed by the terms
+of that license document, provided that the further restriction does
+not survive such relicensing or conveying.
+
+  If you add terms to a covered work in accord with this section, you
+must place, in the relevant source files, a statement of the
+additional terms that apply to those files, or a notice indicating
+where to find the applicable terms.
+
+  Additional terms, permissive or non-permissive, may be stated in the
+form of a separately written license, or stated as exceptions;
+the above requirements apply either way.
+
+  8. Termination.
+
+  You may not propagate or modify a covered work except as expressly
+provided under this License.  Any attempt otherwise to propagate or
+modify it is void, and will automatically terminate your rights under
+this License (including any patent licenses granted under the third
+paragraph of section 11).
+
+  However, if you cease all violation of this License, then your
+license from a particular copyright holder is reinstated (a)
+provisionally, unless and until the copyright holder explicitly and
+finally terminates your license, and (b) permanently, if the copyright
+holder fails to notify you of the violation by some reasonable means
+prior to 60 days after the cessation.
+
+  Moreover, your license from a particular copyright holder is
+reinstated permanently if the copyright holder notifies you of the
+violation by some reasonable means, this is the first time you have
+received notice of violation of this License (for any work) from that
+copyright holder, and you cure the violation prior to 30 days after
+your receipt of the notice.
+
+  Termination of your rights under this section does not terminate the
+licenses of parties who have received copies or rights from you under
+this License.  If your rights have been terminated and not permanently
+reinstated, you do not qualify to receive new licenses for the same
+material under section 10.
+
+  9. Acceptance Not Required for Having Copies.
+
+  You are not required to accept this License in order to receive or
+run a copy of the Program.  Ancillary propagation of a covered work
+occurring solely as a consequence of using peer-to-peer transmission
+to receive a copy likewise does not require acceptance.  However,
+nothing other than this License grants you permission to propagate or
+modify any covered work.  These actions infringe copyright if you do
+not accept this License.  Therefore, by modifying or propagating a
+covered work, you indicate your acceptance of this License to do so.
+
+  10. Automatic Licensing of Downstream Recipients.
+
+  Each time you convey a covered work, the recipient automatically
+receives a license from the original licensors, to run, modify and
+propagate that work, subject to this License.  You are not responsible
+for enforcing compliance by third parties with this License.
+
+  An "entity transaction" is a transaction transferring control of an
+organization, or substantially all assets of one, or subdividing an
+organization, or merging organizations.  If propagation of a covered
+work results from an entity transaction, each party to that
+transaction who receives a copy of the work also receives whatever
+licenses to the work the party's predecessor in interest had or could
+give under the previous paragraph, plus a right to possession of the
+Corresponding Source of the work from the predecessor in interest, if
+the predecessor has it or can get it with reasonable efforts.
+
+  You may not impose any further restrictions on the exercise of the
+rights granted or affirmed under this License.  For example, you may
+not impose a license fee, royalty, or other charge for exercise of
+rights granted under this License, and you may not initiate litigation
+(including a cross-claim or counterclaim in a lawsuit) alleging that
+any patent claim is infringed by making, using, selling, offering for
+sale, or importing the Program or any portion of it.
+
+  11. Patents.
+
+  A "contributor" is a copyright holder who authorizes use under this
+License of the Program or a work on which the Program is based.  The
+work thus licensed is called the contributor's "contributor version".
+
+  A contributor's "essential patent claims" are all patent claims
+owned or controlled by the contributor, whether already acquired or
+hereafter acquired, that would be infringed by some manner, permitted
+by this License, of making, using, or selling its contributor version,
+but do not include claims that would be infringed only as a
+consequence of further modification of the contributor version.  For
+purposes of this definition, "control" includes the right to grant
+patent sublicenses in a manner consistent with the requirements of
+this License.
+
+  Each contributor grants you a non-exclusive, worldwide, royalty-free
+patent license under the contributor's essential patent claims, to
+make, use, sell, offer for sale, import and otherwise run, modify and
+propagate the contents of its contributor version.
+
+  In the following three paragraphs, a "patent license" is any express
+agreement or commitment, however denominated, not to enforce a patent
+(such as an express permission to practice a patent or covenant not to
+sue for patent infringement).  To "grant" such a patent license to a
+party means to make such an agreement or commitment not to enforce a
+patent against the party.
+
+  If you convey a covered work, knowingly relying on a patent license,
+and the Corresponding Source of the work is not available for anyone
+to copy, free of charge and under the terms of this License, through a
+publicly available network server or other readily accessible means,
+then you must either (1) cause the Corresponding Source to be so
+available, or (2) arrange to deprive yourself of the benefit of the
+patent license for this particular work, or (3) arrange, in a manner
+consistent with the requirements of this License, to extend the patent
+license to downstream recipients.  "Knowingly relying" means you have
+actual knowledge that, but for the patent license, your conveying the
+covered work in a country, or your recipient's use of the covered work
+in a country, would infringe one or more identifiable patents in that
+country that you have reason to believe are valid.
+
+  If, pursuant to or in connection with a single transaction or
+arrangement, you convey, or propagate by procuring conveyance of, a
+covered work, and grant a patent license to some of the parties
+receiving the covered work authorizing them to use, propagate, modify
+or convey a specific copy of the covered work, then the patent license
+you grant is automatically extended to all recipients of the covered
+work and works based on it.
+
+  A patent license is "discriminatory" if it does not include within
+the scope of its coverage, prohibits the exercise of, or is
+conditioned on the non-exercise of one or more of the rights that are
+specifically granted under this License.  You may not convey a covered
+work if you are a party to an arrangement with a third party that is
+in the business of distributing software, under which you make payment
+to the third party based on the extent of your activity of conveying
+the work, and under which the third party grants, to any of the
+parties who would receive the covered work from you, a discriminatory
+patent license (a) in connection with copies of the covered work
+conveyed by you (or copies made from those copies), or (b) primarily
+for and in connection with specific products or compilations that
+contain the covered work, unless you entered into that arrangement,
+or that patent license was granted, prior to 28 March 2007.
+
+  Nothing in this License shall be construed as excluding or limiting
+any implied license or other defenses to infringement that may
+otherwise be available to you under applicable patent law.
+
+  12. No Surrender of Others' Freedom.
+
+  If conditions are imposed on you (whether by court order, agreement or
+otherwise) that contradict the conditions of this License, they do not
+excuse you from the conditions of this License.  If you cannot convey a
+covered work so as to satisfy simultaneously your obligations under this
+License and any other pertinent obligations, then as a consequence you may
+not convey it at all.  For example, if you agree to terms that obligate you
+to collect a royalty for further conveying from those to whom you convey
+the Program, the only way you could satisfy both those terms and this
+License would be to refrain entirely from conveying the Program.
+
+  13. Remote Network Interaction; Use with the GNU General Public License.
+
+  Notwithstanding any other provision of this License, if you modify the
+Program, your modified version must prominently offer all users
+interacting with it remotely through a computer network (if your version
+supports such interaction) an opportunity to receive the Corresponding
+Source of your version by providing access to the Corresponding Source
+from a network server at no charge, through some standard or customary
+means of facilitating copying of software.  This Corresponding Source
+shall include the Corresponding Source for any work covered by version 3
+of the GNU General Public License that is incorporated pursuant to the
+following paragraph.
+
+  Notwithstanding any other provision of this License, you have
+permission to link or combine any covered work with a work licensed
+under version 3 of the GNU General Public License into a single
+combined work, and to convey the resulting work.  The terms of this
+License will continue to apply to the part which is the covered work,
+but the work with which it is combined will remain governed by version
+3 of the GNU General Public License.
+
+  14. Revised Versions of this License.
+
+  The Free Software Foundation may publish revised and/or new versions of
+the GNU Affero General Public License from time to time.  Such new versions
+will be similar in spirit to the present version, but may differ in detail to
+address new problems or concerns.
+
+  Each version is given a distinguishing version number.  If the
+Program specifies that a certain numbered version of the GNU Affero General
+Public License "or any later version" applies to it, you have the
+option of following the terms and conditions either of that numbered
+version or of any later version published by the Free Software
+Foundation.  If the Program does not specify a version number of the
+GNU Affero General Public License, you may choose any version ever published
+by the Free Software Foundation.
+
+  If the Program specifies that a proxy can decide which future
+versions of the GNU Affero General Public License can be used, that proxy's
+public statement of acceptance of a version permanently authorizes you
+to choose that version for the Program.
+
+  Later license versions may give you additional or different
+permissions.  However, no additional obligations are imposed on any
+author or copyright holder as a result of your choosing to follow a
+later version.
+
+  15. Disclaimer of Warranty.
+
+  THERE IS NO WARRANTY FOR THE PROGRAM, TO THE EXTENT PERMITTED BY
+APPLICABLE LAW.  EXCEPT WHEN OTHERWISE STATED IN WRITING THE COPYRIGHT
+HOLDERS AND/OR OTHER PARTIES PROVIDE THE PROGRAM "AS IS" WITHOUT WARRANTY
+OF ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING, BUT NOT LIMITED TO,
+THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+PURPOSE.  THE ENTIRE RISK AS TO THE QUALITY AND PERFORMANCE OF THE PROGRAM
+IS WITH YOU.  SHOULD THE PROGRAM PROVE DEFECTIVE, YOU ASSUME THE COST OF
+ALL NECESSARY SERVICING, REPAIR OR CORRECTION.
+
+  16. Limitation of Liability.
+
+  IN NO EVENT UNLESS REQUIRED BY APPLICABLE LAW OR AGREED TO IN WRITING
+WILL ANY COPYRIGHT HOLDER, OR ANY OTHER PARTY WHO MODIFIES AND/OR CONVEYS
+THE PROGRAM AS PERMITTED ABOVE, BE LIABLE TO YOU FOR DAMAGES, INCLUDING ANY
+GENERAL, SPECIAL, INCIDENTAL OR CONSEQUENTIAL DAMAGES ARISING OUT OF THE
+USE OR INABILITY TO USE THE PROGRAM (INCLUDING BUT NOT LIMITED TO LOSS OF
+DATA OR DATA BEING RENDERED INACCURATE OR LOSSES SUSTAINED BY YOU OR THIRD
+PARTIES OR A FAILURE OF THE PROGRAM TO OPERATE WITH ANY OTHER PROGRAMS),
+EVEN IF SUCH HOLDER OR OTHER PARTY HAS BEEN ADVISED OF THE POSSIBILITY OF
+SUCH DAMAGES.
+
+  17. Interpretation of Sections 15 and 16.
+
+  If the disclaimer of warranty and limitation of liability provided
+above cannot be given local legal effect according to their terms,
+reviewing courts shall apply local law that most closely approximates
+an absolute waiver of all civil liability in connection with the
+Program, unless a warranty or assumption of liability accompanies a
+copy of the Program in return for a fee.
+
+                     END OF TERMS AND CONDITIONS
+
+            How to Apply These Terms to Your New Programs
+
+  If you develop a new program, and you want it to be of the greatest
+possible use to the public, the best way to achieve this is to make it
+free software which everyone can redistribute and change under these terms.
+
+  To do so, attach the following notices to the program.  It is safest
+to attach them to the start of each source file to most effectively
+state the exclusion of warranty; and each file should have at least
+the "copyright" line and a pointer to where the full notice is found.
+
+    <one line to give the program's name and a brief idea of what it does.>
+    Copyright (C) <year>  <name of author>
+
+    This program is free software: you can redistribute it and/or modify
+    it under the terms of the GNU Affero General Public License as published
+    by the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU Affero General Public License for more details.
+
+    You should have received a copy of the GNU Affero General Public License
+    along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+Also add information on how to contact you by electronic and paper mail.
+
+  If your software can interact with users remotely through a computer
+network, you should also make sure that it provides a way for users to
+get its source.  For example, if your program is a web application, its
+interface could display a "Source" link that leads users to an archive
+of the code.  There are many ways you could offer source, and different
+solutions will be better for different programs; see section 13 for the
+specific requirements.
+
+  You should also get your employer (if you work as a programmer) or school,
+if any, to sign a "copyright disclaimer" for the program, if necessary.
+For more information on this, and how to apply and follow the GNU AGPL, see
+<https://www.gnu.org/licenses/>.

README.md

@@ -1,54 +1,89 @@
-TETRIS Stats
-============
+<div align="center">
 
-一个基于nonebot2的用于查询TETRIS相关游戏玩家数据的插件  
-目前支持
-* [TETR.IO](https://tetr.io/)
-* [茶服](https://teatube.cn/tos/)
-* [TOP](http://tetrisonline.pl/)
+<p align="center">
+  <img src="img/logo.svg" width="200" height="200" alt="logo" title="Tetris Stats"></a>
+</p>
 
-安装
-----
+# Tetris Stats
 
-* 使用 nb-cli
-```
+✨ 一款基于 [NoneBot2](https://github.com/nonebot/nonebot2) 的用于查询 Tetris 相关游戏玩家数据的插件 ✨
+
+</div>
+
+<p align="center">
+<a href="https://github.com/shoucandanghehe/nonebot-plugin-tetris-stats/blob/main/LICENSE">
+  <img
+    src="https://img.shields.io/github/license/shoucandanghehe/nonebot-plugin-tetris-stats"
+    alt="License"
+  />
+</a>
+<a href="https://www.python.org/">
+  <img
+    src="https://img.shields.io/badge/Python-3.10+-blue"
+    alt="Python"
+  />
+</a>
+<a href="https://pypi.python.org/pypi/nonebot-plugin-tetris-stats">
+  <img
+    src="https://img.shields.io/pypi/v/nonebot-plugin-tetris-stats"
+    alt="PyPi"
+  />
+</a>
+<a href="https://github.com/charliermarsh/ruff">
+  <img
+    src="https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/charliermarsh/ruff/main/assets/badge/v1.json"
+    alt="Ruff"
+  />
+</a>
+<a href="https://gitmoji.dev">
+  <img
+    src="https://img.shields.io/badge/gitmoji-%20😜%20😍-FFDD67.svg?style=flat-square"
+    alt="Gitmoji"
+  />
+</a>
+<a href="https://wakatime.com/badge/user/138b2226-8e02-42be-b99d-35c05198836f/project/65f5bdf7-45ec-479a-8dd2-18c498c910ca">
+  <img
+    src="https://wakatime.com/badge/user/138b2226-8e02-42be-b99d-35c05198836f/project/65f5bdf7-45ec-479a-8dd2-18c498c910ca.svg"
+    alt="wakatime"
+    />
+</a>
+</p>
+
+## ✨ 目前支持的游戏
+
+- [TETR.IO](https://tetr.io/)
+- [茶服](https://teatube.cn/tos/)
+- [TOP](http://tetrisonline.pl/)
+
+## 🚀 安装
+
+- 使用 nb-cli
+
+```bash
 nb plugin install nonebot-plugin-tetris-stats
 ```
 
-* 使用 pip
+- 使用 poetry
+
+```bash
+poetry add nonebot-plugin-tetris-stats
 ```
+
+- 使用 pip ~~不推荐~~
+
+```bash
 pip install nonebot-plugin-tetris-stats
 ```
-* 对于 Windows
-```
-# CMD or PowerShell
-playwright install firefox
-```
-* 对于 Linux
-```
-# 似乎 playwright官方 只支持 Ubuntu, 如果你是其他系统请自行尝试解决依赖问题
-playwright install firefox
-playwright install-deps firefox
-```
 
-使用
-----
+## ♿️ 使用
 
-参考NoneBot2文档 [加载插件](https://v2.nonebot.dev/docs/tutorial/plugin/load-plugin/)
+- 参考 NoneBot2 文档 [加载插件](https://nonebot.dev/docs/tutorial/create-plugin#%E5%8A%A0%E8%BD%BD%E6%8F%92%E4%BB%B6)
 
-依赖
-----
+## 🎉 鸣谢
 
-目前只支持 `OneBot V11` 协议
+- [NoneBot2](https://v2.nonebot.dev/)
+- 所有为机器人生态做出贡献的人❤️
 
-鸣谢
-----
+## 📝 开源
 
-* [NoneBot2](https://v2.nonebot.dev/)
-* [OneBot](https://onebot.dev/)
-* [go-cqhttp](https://github.com/Mrs4s/go-cqhttp/)
-
-开源
-----
-
-本项目使用[MIT](https://github.com/shoucandanghehe/nonebot-plugin-tetris-stats/blob/main/LICENSE)许可证开源
+本项目使用 [AGPL-3.0](https://github.com/shoucandanghehe/nonebot-plugin-tetris-stats/blob/main/LICENSE) 许可证开源

img/logo.svg

@@ -0,0 +1,7 @@
+<svg width="71" height="71" viewBox="0 0 71 71" fill="none" xmlns="http://www.w3.org/2000/svg">
+<path d="M53.1133 39.2642L42.3935 49.9446C41.8086 50.5274 41.5161 50.8188 41.1788 50.928C40.8822 51.024 40.5626 51.024 40.2659 50.928C39.9287 50.8188 39.6362 50.5274 39.0512 49.9446L30.2776 41.2032L21.5041 49.9446C20.9191 50.5274 20.6266 50.8188 20.2893 50.928C19.9927 51.024 19.6731 51.024 19.3764 50.928C19.0392 50.8188 18.7467 50.5274 18.1617 49.9446L16.1772 47.9673L30.2776 33.9187L39.2299 42.8381C40.0952 43.7002 41.498 43.7002 42.3633 42.8381L53.1133 32.1276V39.2642Z" fill="#EA5252"/>
+<path d="M57.5446 34.8491L59.9407 32.4618C60.5256 31.879 60.8181 31.5876 60.9277 31.2516C61.0241 30.956 61.0241 30.6376 60.9277 30.342C60.8181 30.006 60.5256 29.7146 59.9407 29.1318L57.5439 26.7438C57.5444 26.7619 57.5446 26.78 57.5446 26.7982L57.5446 34.8491Z" fill="#EA5252"/>
+<path d="M55.3835 24.5913C55.3653 24.5908 55.3472 24.5906 55.3289 24.5906L46.9514 24.5906L49.4959 22.0554C50.0809 21.4726 50.3734 21.1812 50.7107 21.072C51.0073 20.976 51.3269 20.976 51.6236 21.072C51.9608 21.1812 52.2533 21.4726 52.8383 22.0554L55.3835 24.5913Z" fill="#EA5252"/>
+<path d="M42.5201 29.0057L40.7224 30.7968L31.9488 22.0554C31.3638 21.4726 31.0714 21.1812 30.7341 21.072C30.4374 20.976 30.1179 20.976 29.8212 21.072C29.4839 21.1812 29.1914 21.4726 28.6065 22.0554L11.0593 39.5382C10.4744 40.121 10.1819 40.4124 10.0723 40.7484C9.9759 41.044 9.9759 41.3624 10.0723 41.658C10.1819 41.994 10.4744 42.2854 11.0593 42.8682L13.0438 44.8454L28.7109 29.2358C29.5762 28.3737 30.9791 28.3737 31.8443 29.2358L40.7966 38.1552L49.9799 29.0057H42.5201Z" fill="#EA5252"/>
+<circle cx="35.5" cy="35.5" r="29.5" stroke="#EA5252" stroke-width="2"/>
+</svg>

nonebot_plugin_tetris_stats/__init__.py

@@ -1 +1,31 @@
-from . import game_data_processor
+from nonebot import require
+from nonebot.plugin import PluginMetadata
+
+require('nonebot_plugin_alconna')
+require('nonebot_plugin_apscheduler')
+require('nonebot_plugin_localstore')
+require('nonebot_plugin_orm')
+require('nonebot_plugin_session_orm')
+require('nonebot_plugin_session')
+require('nonebot_plugin_user')
+require('nonebot_plugin_userinfo')
+
+from nonebot_plugin_alconna import namespace  # noqa: E402
+
+with namespace('tetris_stats') as ns:
+    ns.enable_message_cache = False
+
+from .config import migrations  # noqa: E402
+
+__plugin_meta__ = PluginMetadata(
+    name='Tetris Stats',
+    description='一个用于查询 Tetris 相关游戏玩家数据的插件',
+    usage='发送 tstats --help 查询使用方法',
+    type='application',
+    homepage='https://github.com/A-minos/nonebot-plugin-tetris-stats',
+    extra={
+        'orm_version_location': migrations,
+    },
+)
+
+from . import games  # noqa: F401, E402

nonebot_plugin_tetris_stats/config/config.py

@@ -0,0 +1,12 @@
+from pathlib import Path
+
+from nonebot_plugin_localstore import get_cache_dir  # type: ignore[import-untyped]
+from pydantic import BaseModel
+
+CACHE_PATH: Path = get_cache_dir('nonebot_plugin_tetris_stats')
+
+
+class Config(BaseModel):
+    """配置类"""
+
+    tetris_req_timeout: float = 30.0

nonebot_plugin_tetris_stats/config/migrations/09d4bb60160d_rename_field.py

@@ -0,0 +1,54 @@
+"""Rename field
+
+迁移 ID: 09d4bb60160d
+父迁移: b9d65badc713
+创建时间: 2024-04-23 23:42:04.541672
+
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+import sqlalchemy as sa
+from alembic import op
+
+if TYPE_CHECKING:
+    from collections.abc import Sequence
+
+revision: str = '09d4bb60160d'
+down_revision: str | Sequence[str] | None = 'b9d65badc713'
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade(name: str = '') -> None:
+    if name:
+        return
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_iorank', schema=None) as batch_op:
+        batch_op.alter_column('create_time', new_column_name='update_time', existing_type=sa.DateTime())
+        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_iorank_create_time')
+    op.create_index(
+        batch_op.f('ix_nonebot_plugin_tetris_stats_iorank_update_time'),
+        'nonebot_plugin_tetris_stats_iorank',
+        ['update_time'],
+        unique=False,
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade(name: str = '') -> None:
+    if name:
+        return
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_iorank', schema=None) as batch_op:
+        batch_op.alter_column('update_time', new_column_name='create_time')
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_iorank_update_time'))
+    op.create_index(
+        'ix_nonebot_plugin_tetris_stats_iorank_create_time',
+        'nonebot_plugin_tetris_stats_iorank',
+        ['create_time'],
+        unique=False,
+    )
+    # ### end Alembic commands ###

nonebot_plugin_tetris_stats/config/migrations/0d50142b780f_add_field.py

@@ -0,0 +1,46 @@
+"""add field
+
+迁移 ID: 0d50142b780f
+父迁移: 09d4bb60160d
+创建时间: 2024-04-24 14:55:08.064098
+
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+import sqlalchemy as sa
+from alembic import op
+
+if TYPE_CHECKING:
+    from collections.abc import Sequence
+
+revision: str = '0d50142b780f'
+down_revision: str | Sequence[str] | None = '09d4bb60160d'
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade(name: str = '') -> None:
+    if name:
+        return
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_iorank', schema=None) as batch_op:
+        batch_op.add_column(sa.Column('file_hash', sa.String(length=128), nullable=True))
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_iorank_file_hash'), ['file_hash'], unique=False
+        )
+
+    # ### end Alembic commands ###
+
+
+def downgrade(name: str = '') -> None:
+    if name:
+        return
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_iorank', schema=None) as batch_op:
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_iorank_file_hash'))
+        batch_op.drop_column('file_hash')
+
+    # ### end Alembic commands ###

nonebot_plugin_tetris_stats/config/migrations/3c25a5a8c050_refactor_historical.py

@@ -0,0 +1,279 @@
+"""Refactor Historical
+
+迁移 ID: 3c25a5a8c050
+父迁移: b7fbdafc339a
+创建时间: 2024-05-14 09:16:35.193001
+
+"""
+
+from __future__ import annotations
+
+from datetime import datetime
+from typing import TYPE_CHECKING, Any
+
+import sqlalchemy as sa
+from alembic import op
+from nonebot.log import logger
+from rich.progress import BarColumn, MofNCompleteColumn, Progress, TaskProgressColumn, TextColumn, TimeRemainingColumn
+from sqlalchemy import desc, select
+from sqlalchemy.dialects import sqlite
+from sqlalchemy.ext.automap import automap_base
+from sqlalchemy.orm import Session
+from ujson import dumps, loads
+
+if TYPE_CHECKING:
+    from collections.abc import Sequence
+
+revision: str = '3c25a5a8c050'
+down_revision: str | Sequence[str] | None = 'b7fbdafc339a'
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def migrate_old_data() -> None:
+    Base = automap_base()  # noqa: N806
+    Base.prepare(autoload_with=op.get_bind())
+    OldHistoricalData = Base.classes.nonebot_plugin_tetris_stats_historicaldata  # noqa: N806
+    TETRIOHistoricalData = Base.classes.nonebot_plugin_tetris_stats_tetriohistoricaldata  # noqa: N806
+    TOSHistoricalData = Base.classes.nonebot_plugin_tetris_stats_toshistoricaldata  # noqa: N806
+    with (
+        Session(op.get_bind()) as session,
+        Progress(
+            TextColumn('[progress.description]{task.description}'),
+            BarColumn(),
+            MofNCompleteColumn(),
+            TaskProgressColumn(),
+            TimeRemainingColumn(),
+        ) as progress,
+    ):
+        task_id = progress.add_task('[cyan]Migrating:', total=session.query(OldHistoricalData).count())
+        pointer = 0
+        while pointer < session.query(OldHistoricalData).order_by(desc(OldHistoricalData.id)).limit(1).one().id:
+            result = session.scalars(
+                select(OldHistoricalData)
+                .where(OldHistoricalData.id > pointer)
+                .order_by(OldHistoricalData.id)
+                .limit(100)
+            ).all()
+            for j in result:
+                processed_data: dict[str, Any] = loads(j.processed_data)
+                if j.game_platform == 'IO':
+                    if (data := processed_data.get('user_info')) is not None:
+                        session.add(
+                            TETRIOHistoricalData(
+                                user_unique_identifier=j.user_unique_identifier,
+                                api_type='User Info',
+                                data=dumps(data),
+                                update_time=datetime.fromisoformat(data['cache']['cached_at']),
+                            )
+                        )
+                    if (data := processed_data.get('user_records')) is not None:
+                        session.add(
+                            TETRIOHistoricalData(
+                                user_unique_identifier=j.user_unique_identifier,
+                                api_type='User Records',
+                                data=dumps(data),
+                                update_time=datetime.fromisoformat(data['cache']['cached_at']),
+                            )
+                        )
+                if j.game_platform == 'TOS' and not j.user_unique_identifier.isdigit():
+                    if (data := processed_data.get('user_info')) is not None:
+                        session.add(
+                            TOSHistoricalData(
+                                user_unique_identifier=j.user_unique_identifier,
+                                api_type='User Info',
+                                data=dumps(data),
+                                update_time=j.finish_time,
+                            )
+                        )
+                    if (data := processed_data.get('user_profile')) is not None:
+                        for v in data.values():
+                            session.add(
+                                TOSHistoricalData(
+                                    user_unique_identifier=j.user_unique_identifier,
+                                    api_type='User Profile',
+                                    data=dumps(v),
+                                    update_time=j.finish_time,
+                                )
+                            )
+                progress.update(task_id, advance=1)
+            session.commit()
+            pointer = result[-1].id
+    logger.success('Migrate successfully')
+
+
+def upgrade(name: str = '') -> None:
+    if name:
+        return
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table(
+        'nonebot_plugin_tetris_stats_tetriohistoricaldata',
+        sa.Column('id', sa.Integer(), nullable=False),
+        sa.Column('user_unique_identifier', sa.String(length=24), nullable=False),
+        sa.Column('api_type', sa.String(length=16), nullable=False),
+        sa.Column('data', sa.JSON(), nullable=False),
+        sa.Column('update_time', sa.DateTime(), nullable=False),
+        sa.PrimaryKeyConstraint('id', name=op.f('pk_nonebot_plugin_tetris_stats_tetriohistoricaldata')),
+        info={'bind_key': 'nonebot_plugin_tetris_stats'},
+    )
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_tetriohistoricaldata', schema=None) as batch_op:
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_tetriohistoricaldata_api_type'), ['api_type'], unique=False
+        )
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_tetriohistoricaldata_update_time'), ['update_time'], unique=False
+        )
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_tetriohistoricaldata_user_unique_identifier'),
+            ['user_unique_identifier'],
+            unique=False,
+        )
+
+    op.create_table(
+        'nonebot_plugin_tetris_stats_tophistoricaldata',
+        sa.Column('id', sa.Integer(), nullable=False),
+        sa.Column('user_unique_identifier', sa.String(length=24), nullable=False),
+        sa.Column('api_type', sa.String(length=16), nullable=False),
+        sa.Column('data', sa.JSON(), nullable=False),
+        sa.Column('update_time', sa.DateTime(), nullable=False),
+        sa.PrimaryKeyConstraint('id', name=op.f('pk_nonebot_plugin_tetris_stats_tophistoricaldata')),
+        info={'bind_key': 'nonebot_plugin_tetris_stats'},
+    )
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_tophistoricaldata', schema=None) as batch_op:
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_tophistoricaldata_api_type'), ['api_type'], unique=False
+        )
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_tophistoricaldata_update_time'), ['update_time'], unique=False
+        )
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_tophistoricaldata_user_unique_identifier'),
+            ['user_unique_identifier'],
+            unique=False,
+        )
+
+    op.create_table(
+        'nonebot_plugin_tetris_stats_toshistoricaldata',
+        sa.Column('id', sa.Integer(), nullable=False),
+        sa.Column('user_unique_identifier', sa.String(length=24), nullable=False),
+        sa.Column('api_type', sa.String(length=16), nullable=False),
+        sa.Column('data', sa.JSON(), nullable=False),
+        sa.Column('update_time', sa.DateTime(), nullable=False),
+        sa.PrimaryKeyConstraint('id', name=op.f('pk_nonebot_plugin_tetris_stats_toshistoricaldata')),
+        info={'bind_key': 'nonebot_plugin_tetris_stats'},
+    )
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_toshistoricaldata', schema=None) as batch_op:
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_toshistoricaldata_api_type'), ['api_type'], unique=False
+        )
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_toshistoricaldata_update_time'), ['update_time'], unique=False
+        )
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_toshistoricaldata_user_unique_identifier'),
+            ['user_unique_identifier'],
+            unique=False,
+        )
+
+    op.create_table(
+        'nonebot_plugin_tetris_stats_triggerhistoricaldata',
+        sa.Column('id', sa.Integer(), nullable=False),
+        sa.Column('trigger_time', sa.DateTime(), nullable=False),
+        sa.Column('session_persist_id', sa.Integer(), nullable=False),
+        sa.Column('game_platform', sa.String(length=32), nullable=False),
+        sa.Column('command_type', sa.String(length=16), nullable=False),
+        sa.Column('command_args', sa.JSON(), nullable=False),
+        sa.Column('finish_time', sa.DateTime(), nullable=False),
+        sa.PrimaryKeyConstraint('id', name=op.f('pk_nonebot_plugin_tetris_stats_triggerhistoricaldata')),
+        info={'bind_key': 'nonebot_plugin_tetris_stats'},
+    )
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_triggerhistoricaldata', schema=None) as batch_op:
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_triggerhistoricaldata_command_type'),
+            ['command_type'],
+            unique=False,
+        )
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_triggerhistoricaldata_game_platform'),
+            ['game_platform'],
+            unique=False,
+        )
+
+    migrate_old_data()
+
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_historicaldata', schema=None) as batch_op:
+        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_historicaldata_command_type')
+        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_historicaldata_game_platform')
+        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_historicaldata_source_account')
+        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_historicaldata_source_type')
+        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_historicaldata_user_unique_identifier')
+
+    op.drop_table('nonebot_plugin_tetris_stats_historicaldata')
+    # ### end Alembic commands ###
+
+
+def downgrade(name: str = '') -> None:
+    if name:
+        return
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table(
+        'nonebot_plugin_tetris_stats_historicaldata',
+        sa.Column('id', sa.INTEGER(), nullable=False),
+        sa.Column('trigger_time', sa.DATETIME(), nullable=False),
+        sa.Column('bot_platform', sa.VARCHAR(length=32), nullable=True),
+        sa.Column('bot_account', sa.VARCHAR(), nullable=True),
+        sa.Column('source_type', sa.VARCHAR(length=32), nullable=True),
+        sa.Column('source_account', sa.VARCHAR(), nullable=True),
+        sa.Column('message', sa.BLOB(), nullable=True),
+        sa.Column('game_platform', sa.VARCHAR(length=32), nullable=False),
+        sa.Column('command_type', sa.VARCHAR(length=16), nullable=False),
+        sa.Column('command_args', sqlite.JSON(), nullable=False),
+        sa.Column('game_user', sqlite.JSON(), nullable=False),
+        sa.Column('processed_data', sqlite.JSON(), nullable=False),
+        sa.Column('finish_time', sa.DATETIME(), nullable=False),
+        sa.Column('user_unique_identifier', sa.VARCHAR(length=32), nullable=False),
+        sa.PrimaryKeyConstraint('id', name='pk_nonebot_plugin_tetris_stats_historicaldata'),
+    )
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_historicaldata', schema=None) as batch_op:
+        batch_op.create_index(
+            'ix_nonebot_plugin_tetris_stats_historicaldata_user_unique_identifier',
+            ['user_unique_identifier'],
+            unique=False,
+        )
+        batch_op.create_index(
+            'ix_nonebot_plugin_tetris_stats_historicaldata_source_type', ['source_type'], unique=False
+        )
+        batch_op.create_index(
+            'ix_nonebot_plugin_tetris_stats_historicaldata_source_account', ['source_account'], unique=False
+        )
+        batch_op.create_index(
+            'ix_nonebot_plugin_tetris_stats_historicaldata_game_platform', ['game_platform'], unique=False
+        )
+        batch_op.create_index(
+            'ix_nonebot_plugin_tetris_stats_historicaldata_command_type', ['command_type'], unique=False
+        )
+
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_triggerhistoricaldata', schema=None) as batch_op:
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_triggerhistoricaldata_game_platform'))
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_triggerhistoricaldata_command_type'))
+
+    op.drop_table('nonebot_plugin_tetris_stats_triggerhistoricaldata')
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_toshistoricaldata', schema=None) as batch_op:
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_toshistoricaldata_user_unique_identifier'))
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_toshistoricaldata_update_time'))
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_toshistoricaldata_api_type'))
+
+    op.drop_table('nonebot_plugin_tetris_stats_toshistoricaldata')
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_tophistoricaldata', schema=None) as batch_op:
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_tophistoricaldata_user_unique_identifier'))
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_tophistoricaldata_update_time'))
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_tophistoricaldata_api_type'))
+
+    op.drop_table('nonebot_plugin_tetris_stats_tophistoricaldata')
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_tetriohistoricaldata', schema=None) as batch_op:
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_tetriohistoricaldata_user_unique_identifier'))
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_tetriohistoricaldata_update_time'))
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_tetriohistoricaldata_api_type'))
+
+    op.drop_table('nonebot_plugin_tetris_stats_tetriohistoricaldata')
+    # ### end Alembic commands ###

nonebot_plugin_tetris_stats/config/migrations/6c3206f90cc3_add_field.py

@@ -0,0 +1,69 @@
+"""Add redundant platform field
+
+迁移 ID: 6c3206f90cc3
+父迁移: 9f6582279ce2
+创建时间: 2023-11-26 20:15:56.033892
+
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from alembic import op
+from sqlalchemy.ext.automap import automap_base
+from sqlalchemy.orm import Session
+from ujson import dumps, loads
+
+if TYPE_CHECKING:
+    from collections.abc import Sequence
+
+revision: str = '6c3206f90cc3'
+down_revision: str | Sequence[str] | None = '9f6582279ce2'
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade(name: str = '') -> None:
+    if name:
+        return
+
+    Base = automap_base()  # noqa: N806
+    connection = op.get_bind()
+    Base.prepare(autoload_with=connection)
+
+    HistoricalData = Base.classes.nonebot_plugin_tetris_stats_historicaldata  # noqa: N806
+
+    with Session(connection) as session:
+        for row in session.query(HistoricalData):
+            platform = row.game_platform
+            game_user = loads(row.game_user)
+            processed_data = loads(row.processed_data)
+            game_user['platform'] = platform
+            processed_data['platform'] = platform
+            row.game_user = dumps(game_user)
+            row.processed_data = dumps(processed_data)
+            session.add(row)
+        session.commit()
+
+
+def downgrade(name: str = '') -> None:
+    if name:
+        return
+
+    Base = automap_base()  # noqa: N806
+    connection = op.get_bind()
+    Base.prepare(autoload_with=connection)
+
+    HistoricalData = Base.classes.nonebot_plugin_tetris_stats_historicaldata  # noqa: N806
+
+    with Session(connection) as session:
+        for row in session.query(HistoricalData):
+            game_user = loads(row.game_user)
+            processed_data = loads(row.processed_data)
+            game_user.pop('platform', None)
+            processed_data.pop('platform', None)
+            row.game_user = dumps(game_user)
+            row.processed_data = dumps(processed_data)
+            session.add(row)
+        session.commit()

nonebot_plugin_tetris_stats/config/migrations/8a91210ce14d_correct_the_data_in_historicaldata.py

@@ -0,0 +1,98 @@
+"""Correct the data in HistoricalData
+
+迁移 ID: 8a91210ce14d
+父迁移: 0d50142b780f
+创建时间: 2024-05-06 08:16:38.487214
+
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from alembic import op
+from nonebot.log import logger
+from sqlalchemy import select
+from sqlalchemy.ext.automap import automap_base
+from sqlalchemy.orm import Session
+
+if TYPE_CHECKING:
+    from collections.abc import Sequence
+
+revision: str = '8a91210ce14d'
+down_revision: str | Sequence[str] | None = '0d50142b780f'
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade(name: str = '') -> None:  # noqa: C901
+    if name:
+        return
+    from nonebot_plugin_tetris_stats.version import __version__
+
+    if __version__ != '1.0.3':
+        msg = '本迁移需要1.0.3版本, 请先锁定版本至1.0.3版本再执行本迁移'
+        logger.critical(msg)
+        raise RuntimeError(msg)
+
+    from nonebot.compat import PYDANTIC_V2, type_validate_json
+    from pydantic import BaseModel, ValidationError
+    from rich.progress import (
+        BarColumn,
+        MofNCompleteColumn,
+        Progress,
+        TaskProgressColumn,
+        TextColumn,
+        TimeRemainingColumn,
+    )
+
+    from nonebot_plugin_tetris_stats.game_data_processor.schemas import (  # type: ignore[import-untyped]
+        BaseProcessedData,
+    )
+
+    Base = automap_base()  # noqa: N806
+    Base.prepare(autoload_with=op.get_bind())
+    HistoricalData = Base.classes.nonebot_plugin_tetris_stats_historicaldata  # noqa: N806
+    if PYDANTIC_V2:
+
+        def model_to_json(value: BaseModel) -> str:
+            return value.model_dump_json(by_alias=True)
+    else:
+
+        def model_to_json(value: BaseModel) -> str:
+            return value.json(by_alias=True)
+
+    models = BaseProcessedData.__subclasses__()
+
+    def json_to_model(value: str) -> BaseModel:
+        for i in models:
+            try:
+                return type_validate_json(i, value)
+            except ValidationError:  # noqa: PERF203
+                ...
+        raise ValueError
+
+    with Session(op.get_bind()) as session:
+        count = session.query(HistoricalData).count()
+        with Progress(
+            TextColumn('[progress.description]{task.description}'),
+            BarColumn(),
+            MofNCompleteColumn(),
+            TaskProgressColumn(),
+            TimeRemainingColumn(),
+        ) as progress:
+            task_id = progress.add_task('[cyan]Updateing:', total=count)
+            for i in range(0, count, 100):
+                for j in session.scalars(
+                    select(HistoricalData).where(HistoricalData.id > i).order_by(HistoricalData.id).limit(100)
+                ):
+                    model = json_to_model(j.processed_data)
+                    j.processed_data = model_to_json(model)
+                    progress.update(task_id, advance=1)
+                session.commit()
+    logger.success('Corrected HistoricalData')
+
+
+def downgrade(name: str = '') -> None:
+    if name:
+        return

nonebot_plugin_tetris_stats/config/migrations/9866f53ce44f_init_db.py

@@ -0,0 +1,143 @@
+"""init db
+
+迁移 ID: 9866f53ce44f
+父迁移:
+创建时间: 2023-11-11 16:24:11.826667
+
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+import sqlalchemy as sa
+from alembic import op
+
+if TYPE_CHECKING:
+    from collections.abc import Sequence
+
+revision: str = '9866f53ce44f'
+down_revision: str | Sequence[str] | None = None
+branch_labels: str | Sequence[str] | None = ('nonebot_plugin_tetris_stats',)
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade(name: str = '') -> None:
+    if name:
+        return
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table(
+        'nonebot_plugin_tetris_stats_bind',
+        sa.Column('id', sa.Integer(), nullable=False),
+        sa.Column('chat_platform', sa.String(length=32), nullable=False),
+        sa.Column('chat_account', sa.String(), nullable=False),
+        sa.Column('game_platform', sa.String(length=32), nullable=False),
+        sa.Column('game_account', sa.String(), nullable=False),
+        sa.PrimaryKeyConstraint('id', name=op.f('pk_nonebot_plugin_tetris_stats_bind')),
+    )
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_bind', schema=None) as batch_op:
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_bind_chat_account'),
+            ['chat_account'],
+            unique=False,
+        )
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_bind_chat_platform'),
+            ['chat_platform'],
+            unique=False,
+        )
+
+    op.create_table(
+        'nonebot_plugin_tetris_stats_historicaldata',
+        sa.Column('id', sa.Integer(), nullable=False),
+        sa.Column('trigger_time', sa.DateTime(), nullable=False),
+        sa.Column('bot_platform', sa.String(length=32), nullable=True),
+        sa.Column('bot_account', sa.String(), nullable=True),
+        sa.Column('source_type', sa.String(length=32), nullable=True),
+        sa.Column('source_account', sa.String(), nullable=True),
+        sa.Column('message', sa.PickleType(), nullable=True),
+        sa.Column('game_platform', sa.String(length=32), nullable=False),
+        sa.Column('command_type', sa.String(length=16), nullable=False),
+        sa.Column('command_args', sa.JSON(), nullable=False),
+        sa.Column('game_user', sa.PickleType(), nullable=False),
+        sa.Column('processed_data', sa.PickleType(), nullable=False),
+        sa.Column('finish_time', sa.DateTime(), nullable=False),
+        sa.PrimaryKeyConstraint('id', name=op.f('pk_nonebot_plugin_tetris_stats_historicaldata')),
+    )
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_historicaldata', schema=None) as batch_op:
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_command_type'),
+            ['command_type'],
+            unique=False,
+        )
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_game_platform'),
+            ['game_platform'],
+            unique=False,
+        )
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_source_account'),
+            ['source_account'],
+            unique=False,
+        )
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_source_type'),
+            ['source_type'],
+            unique=False,
+        )
+
+    op.create_table(
+        'nonebot_plugin_tetris_stats_iorank',
+        sa.Column('id', sa.Integer(), nullable=False),
+        sa.Column('rank', sa.String(length=2), nullable=False),
+        sa.Column('tr_line', sa.Float(), nullable=False),
+        sa.Column('player_count', sa.Integer(), nullable=False),
+        sa.Column('low_pps', sa.JSON(), nullable=False),
+        sa.Column('low_apm', sa.JSON(), nullable=False),
+        sa.Column('low_vs', sa.JSON(), nullable=False),
+        sa.Column('avg_pps', sa.Float(), nullable=False),
+        sa.Column('avg_apm', sa.Float(), nullable=False),
+        sa.Column('avg_vs', sa.Float(), nullable=False),
+        sa.Column('high_pps', sa.JSON(), nullable=False),
+        sa.Column('high_apm', sa.JSON(), nullable=False),
+        sa.Column('high_vs', sa.JSON(), nullable=False),
+        sa.Column('create_time', sa.DateTime(), nullable=False),
+        sa.PrimaryKeyConstraint('id', name=op.f('pk_nonebot_plugin_tetris_stats_iorank')),
+    )
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_iorank', schema=None) as batch_op:
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_iorank_create_time'),
+            ['create_time'],
+            unique=False,
+        )
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_iorank_rank'),
+            ['rank'],
+            unique=False,
+        )
+
+    # ### end Alembic commands ###
+
+
+def downgrade(name: str = '') -> None:
+    if name:
+        return
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_iorank', schema=None) as batch_op:
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_iorank_rank'))
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_iorank_create_time'))
+
+    op.drop_table('nonebot_plugin_tetris_stats_iorank')
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_historicaldata', schema=None) as batch_op:
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_source_type'))
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_source_account'))
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_game_platform'))
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_command_type'))
+
+    op.drop_table('nonebot_plugin_tetris_stats_historicaldata')
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_bind', schema=None) as batch_op:
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_bind_chat_platform'))
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_bind_chat_account'))
+
+    op.drop_table('nonebot_plugin_tetris_stats_bind')
+    # ### end Alembic commands ###

nonebot_plugin_tetris_stats/config/migrations/9cd1647db502_merge_old_db.py

@@ -0,0 +1,97 @@
+"""Merge old db
+
+迁移 ID: 9cd1647db502
+父迁移: 9866f53ce44f
+创建时间: 2023-11-11 16:51:30.718277
+
+"""
+
+from __future__ import annotations
+
+from pathlib import Path
+from shutil import copyfile
+from typing import TYPE_CHECKING
+
+from alembic import op
+from nonebot import get_driver
+from nonebot.log import logger
+from sqlalchemy import Connection, create_engine, inspect, text
+from sqlalchemy.ext.automap import automap_base
+from sqlalchemy.orm import Session
+
+if TYPE_CHECKING:
+    from collections.abc import Sequence
+
+revision: str = '9cd1647db502'
+down_revision: str | Sequence[str] | None = '9866f53ce44f'
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+driver = get_driver()
+config = driver.config
+
+
+def migrate_old_data(connection: Connection) -> None:
+    Base = automap_base()  # noqa: N806
+    Base.prepare(autoload_with=op.get_bind())
+    Bind = Base.classes.nonebot_plugin_tetris_stats_bind  # noqa: N806
+
+    def non_empty(obj: str) -> bool:
+        if obj != '' and not obj.isspace():
+            return True
+        return False
+
+    def is_int(obj: int | str) -> bool:
+        if isinstance(obj, int) or obj.isdigit():
+            return True
+        return False
+
+    bind_list = [
+        Bind(chat_platform='OneBot V11', chat_account=int(row.QQ), game_platform='IO', game_account=row.USER)
+        for row in connection.execute(text('select QQ, USER from IOBIND;'))
+        if is_int(row.QQ) and non_empty(row.USER)
+    ]
+    bind_list.extend(
+        [
+            Bind(chat_platform='OneBot V11', chat_account=int(row.QQ), game_platform='TOP', game_account=row.USER)
+            for row in connection.execute(text('select QQ, USER from TOPBIND;'))
+            if is_int(row.QQ) and non_empty(row.USER)
+        ]
+    )
+    with Session(op.get_bind()) as session:
+        session.add_all(bind_list)
+        session.commit()
+    logger.success('nonebot_plugin_tetris_stats: 迁移完成')
+
+
+def upgrade(name: str = '') -> None:
+    if name:
+        return
+    try:
+        db_path = Path(config.db_url)
+    except AttributeError:
+        db_path = Path('data/nonebot_plugin_tetris_stats/data.db')
+    if db_path.exists() is False:
+        logger.warning('nonebot_plugin_tetris_stats: 未发现老版本的数据')
+        logger.success('nonebot_plugin_tetris_stats: 跳过迁移')
+        return
+    copyfile(db_path, db_path.parent / 'data.db.bak')
+    engine = create_engine(f'sqlite:///{db_path.absolute()!s}')
+    with engine.connect() as connection:
+        tables = inspect(connection).get_table_names()
+        if 'IOBIND' not in tables or 'TOPBIND' not in tables:
+            logger.warning('nonebot_plugin_tetris_stats: 未发现老版本的数据')
+            logger.success('nonebot_plugin_tetris_stats: 跳过迁移')
+            return
+        if 'IORANK' not in tables:
+            msg = 'nonebot_plugin_tetris_stats: 请先安装 0.4.4 版本完成迁移之后再升级'
+            logger.warning(msg)
+            raise RuntimeError(msg)
+        logger.info('nonebot_plugin_tetris_stats: 发现来自老版本的数据, 正在迁移...')
+        migrate_old_data(connection)
+    db_path.unlink()
+
+
+def downgrade(name: str = '') -> None:
+    if name:
+        return

nonebot_plugin_tetris_stats/config/migrations/9f6582279ce2_recreate_historical_data.py

@@ -0,0 +1,114 @@
+"""Recreate HistoricalData
+
+迁移 ID: 9f6582279ce2
+父迁移: 9cd1647db502
+创建时间: 2023-11-21 08:35:50.393246
+
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+import sqlalchemy as sa
+from alembic import op
+from sqlalchemy.dialects import sqlite
+
+if TYPE_CHECKING:
+    from collections.abc import Sequence
+
+revision: str = '9f6582279ce2'
+down_revision: str | Sequence[str] | None = '9cd1647db502'
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade(name: str = '') -> None:
+    if name:
+        return
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_historicaldata', schema=None) as batch_op:
+        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_historicaldata_command_type')
+        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_historicaldata_game_platform')
+        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_historicaldata_source_account')
+        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_historicaldata_source_type')
+
+    op.drop_table('nonebot_plugin_tetris_stats_historicaldata')
+
+    op.create_table(
+        'nonebot_plugin_tetris_stats_historicaldata',
+        sa.Column('id', sa.Integer(), nullable=False),
+        sa.Column('trigger_time', sa.DateTime(), nullable=False),
+        sa.Column('bot_platform', sa.String(length=32), nullable=True),
+        sa.Column('bot_account', sa.String(), nullable=True),
+        sa.Column('source_type', sa.String(length=32), nullable=True),
+        sa.Column('source_account', sa.String(), nullable=True),
+        sa.Column('message', sa.PickleType(), nullable=True),
+        sa.Column('game_platform', sa.String(length=32), nullable=False),
+        sa.Column('command_type', sa.String(length=16), nullable=False),
+        sa.Column('command_args', sa.JSON(), nullable=False),
+        sa.Column('game_user', sa.JSON(), nullable=False),
+        sa.Column('processed_data', sa.JSON(), nullable=False),
+        sa.Column('finish_time', sa.DateTime(), nullable=False),
+        sa.PrimaryKeyConstraint('id', name=op.f('pk_nonebot_plugin_tetris_stats_historicaldata')),
+    )
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_historicaldata', schema=None) as batch_op:
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_command_type'), ['command_type'], unique=False
+        )
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_game_platform'), ['game_platform'], unique=False
+        )
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_source_account'), ['source_account'], unique=False
+        )
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_source_type'), ['source_type'], unique=False
+        )
+    # ### end Alembic commands ###
+
+
+def downgrade(name: str = '') -> None:
+    if name:
+        return
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_historicaldata', schema=None) as batch_op:
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_source_type'))
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_source_account'))
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_game_platform'))
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_command_type'))
+
+    op.drop_table('nonebot_plugin_tetris_stats_historicaldata')
+
+    op.create_table(
+        'nonebot_plugin_tetris_stats_historicaldata',
+        sa.Column('id', sa.INTEGER(), nullable=False),
+        sa.Column('trigger_time', sa.DATETIME(), nullable=False),
+        sa.Column('bot_platform', sa.VARCHAR(length=32), nullable=True),
+        sa.Column('bot_account', sa.VARCHAR(), nullable=True),
+        sa.Column('source_type', sa.VARCHAR(length=32), nullable=True),
+        sa.Column('source_account', sa.VARCHAR(), nullable=True),
+        sa.Column('message', sa.BLOB(), nullable=True),
+        sa.Column('game_platform', sa.VARCHAR(length=32), nullable=False),
+        sa.Column('command_type', sa.VARCHAR(length=16), nullable=False),
+        sa.Column('command_args', sqlite.JSON(), nullable=False),
+        sa.Column('game_user', sa.BLOB(), nullable=False),
+        sa.Column('processed_data', sa.BLOB(), nullable=False),
+        sa.Column('finish_time', sa.DATETIME(), nullable=False),
+        sa.PrimaryKeyConstraint('id', name='pk_nonebot_plugin_tetris_stats_historicaldata'),
+    )
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_historicaldata', schema=None) as batch_op:
+        batch_op.create_index(
+            'ix_nonebot_plugin_tetris_stats_historicaldata_source_type', ['source_type'], unique=False
+        )
+        batch_op.create_index(
+            'ix_nonebot_plugin_tetris_stats_historicaldata_source_account', ['source_account'], unique=False
+        )
+        batch_op.create_index(
+            'ix_nonebot_plugin_tetris_stats_historicaldata_game_platform', ['game_platform'], unique=False
+        )
+        batch_op.create_index(
+            'ix_nonebot_plugin_tetris_stats_historicaldata_command_type', ['command_type'], unique=False
+        )
+
+    # ### end Alembic commands ###

nonebot_plugin_tetris_stats/config/migrations/a1195e989cc6_add_tetrio_user_configuration.py

@@ -0,0 +1,44 @@
+"""Add TETRIO user configuration
+
+迁移 ID: a1195e989cc6
+父迁移: b15844837693
+创建时间: 2024-06-09 04:20:07.819194
+
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+import sqlalchemy as sa
+from alembic import op
+
+if TYPE_CHECKING:
+    from collections.abc import Sequence
+
+revision: str = 'a1195e989cc6'
+down_revision: str | Sequence[str] | None = 'b15844837693'
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade(name: str = '') -> None:
+    if name:
+        return
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table(
+        'nonebot_plugin_tetris_stats_tetriouserconfig',
+        sa.Column('id', sa.Integer(), nullable=False),
+        sa.Column('query_template', sa.String(length=2), nullable=False),
+        sa.PrimaryKeyConstraint('id', name=op.f('pk_nonebot_plugin_tetris_stats_tetriouserconfig')),
+        info={'bind_key': 'nonebot_plugin_tetris_stats'},
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade(name: str = '') -> None:
+    if name:
+        return
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table('nonebot_plugin_tetris_stats_tetriouserconfig')
+    # ### end Alembic commands ###

nonebot_plugin_tetris_stats/config/migrations/b15844837693_migrate_to_nonobot_plugin_user.py

@@ -0,0 +1,71 @@
+"""Migrate to nonobot-plugin-user
+
+迁移 ID: b15844837693
+父迁移: 3c25a5a8c050
+创建时间: 2024-06-08 02:27:35.227596
+
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+import sqlalchemy as sa
+from alembic import op
+
+if TYPE_CHECKING:
+    from collections.abc import Sequence
+
+revision: str = 'b15844837693'
+down_revision: str | Sequence[str] | None = '3c25a5a8c050'
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade(name: str = '') -> None:
+    if name:
+        return
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_bind', schema=None) as batch_op:
+        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_bind_chat_account')
+        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_bind_chat_platform')
+
+    op.drop_table('nonebot_plugin_tetris_stats_bind')
+
+    op.create_table(
+        'nonebot_plugin_tetris_stats_bind',
+        sa.Column('id', sa.Integer(), nullable=False),
+        sa.Column('user_id', sa.Integer(), nullable=False),
+        sa.Column('game_platform', sa.String(length=32), nullable=False),
+        sa.Column('game_account', sa.String(), nullable=False),
+        sa.PrimaryKeyConstraint('id', name=op.f('pk_nonebot_plugin_tetris_stats_bind')),
+        info={'bind_key': 'nonebot_plugin_tetris_stats'},
+    )
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_bind', schema=None) as batch_op:
+        batch_op.create_index(batch_op.f('ix_nonebot_plugin_tetris_stats_bind_user_id'), ['user_id'], unique=False)
+    # ### end Alembic commands ###
+
+
+def downgrade(name: str = '') -> None:
+    if name:
+        return
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_bind', schema=None) as batch_op:
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_bind_user_id'))
+
+    op.drop_table('nonebot_plugin_tetris_stats_bind')
+
+    op.create_table(
+        'nonebot_plugin_tetris_stats_bind',
+        sa.Column('id', sa.INTEGER(), nullable=False),
+        sa.Column('chat_platform', sa.VARCHAR(length=32), nullable=False),
+        sa.Column('chat_account', sa.VARCHAR(), nullable=False),
+        sa.Column('game_platform', sa.VARCHAR(length=32), nullable=False),
+        sa.Column('game_account', sa.VARCHAR(), nullable=False),
+        sa.PrimaryKeyConstraint('id', name='pk_nonebot_plugin_tetris_stats_bind'),
+    )
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_bind', schema=None) as batch_op:
+        batch_op.create_index('ix_nonebot_plugin_tetris_stats_bind_chat_platform', ['chat_platform'], unique=False)
+        batch_op.create_index('ix_nonebot_plugin_tetris_stats_bind_chat_account', ['chat_account'], unique=False)
+
+    # ### end Alembic commands ###

nonebot_plugin_tetris_stats/config/migrations/b7fbdafc339a_add_user_unique_identifier_field_to_.py

@@ -0,0 +1,107 @@
+"""Add user_unique_identifier field to HistoricalData
+
+迁移 ID: b7fbdafc339a
+父迁移: 8a91210ce14d
+创建时间: 2024-05-07 16:55:29.527215
+
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+import sqlalchemy as sa
+from alembic import op
+from nonebot.log import logger
+
+if TYPE_CHECKING:
+    from collections.abc import Sequence
+
+revision: str = 'b7fbdafc339a'
+down_revision: str | Sequence[str] | None = '8a91210ce14d'
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade(name: str = '') -> None:
+    if name:
+        return
+    from nonebot_plugin_tetris_stats.version import __version__
+
+    if __version__ != '1.0.4':
+        msg = '本迁移需要1.0.4版本, 请先锁定版本至1.0.4版本再执行本迁移'
+        logger.critical(msg)
+        raise RuntimeError(msg)
+    from nonebot.compat import type_validate_json
+    from pydantic import ValidationError
+    from rich.progress import (
+        BarColumn,
+        MofNCompleteColumn,
+        Progress,
+        TaskProgressColumn,
+        TextColumn,
+        TimeRemainingColumn,
+    )
+    from sqlalchemy import select
+    from sqlalchemy.ext.automap import automap_base
+    from sqlalchemy.orm import Session
+
+    from nonebot_plugin_tetris_stats.game_data_processor.schemas import BaseUser  # type: ignore[import-untyped]
+
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_historicaldata', schema=None) as batch_op:
+        batch_op.add_column(sa.Column('user_unique_identifier', sa.String(length=32), nullable=True))
+        batch_op.create_index(
+            batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_user_unique_identifier'),
+            ['user_unique_identifier'],
+            unique=False,
+        )
+    Base = automap_base()  # noqa: N806
+    connection = op.get_bind()
+    Base.prepare(autoload_with=connection)
+    HistoricalData = Base.classes.nonebot_plugin_tetris_stats_historicaldata  # noqa: N806
+
+    models: list[type[BaseUser]] = BaseUser.__subclasses__()
+
+    def json_to_model(value: str) -> BaseUser:
+        for i in models:
+            try:
+                return type_validate_json(i, value)
+            except ValidationError:  # noqa: PERF203
+                ...
+        raise ValueError
+
+    with Session(op.get_bind()) as session:
+        count = session.query(HistoricalData).count()
+        with Progress(
+            TextColumn('[progress.description]{task.description}'),
+            BarColumn(),
+            MofNCompleteColumn(),
+            TaskProgressColumn(),
+            TimeRemainingColumn(),
+        ) as progress:
+            task_id = progress.add_task('[cyan]Updateing:', total=count)
+            for i in range(0, count, 100):
+                for j in session.scalars(
+                    select(HistoricalData).where(HistoricalData.id > i).order_by(HistoricalData.id).limit(100)
+                ):
+                    model = json_to_model(j.game_user)
+                    try:
+                        j.user_unique_identifier = model.unique_identifier
+                    except ValueError:
+                        session.delete(j)
+                    progress.update(task_id, advance=1)
+                session.commit()
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_historicaldata', schema=None) as batch_op:
+        batch_op.alter_column('user_unique_identifier', existing_type=sa.VARCHAR(length=32), nullable=False)
+    logger.success('database upgrade success')
+
+
+def downgrade(name: str = '') -> None:
+    if name:
+        return
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('nonebot_plugin_tetris_stats_historicaldata', schema=None) as batch_op:
+        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_user_unique_identifier'))
+        batch_op.drop_column('user_unique_identifier')
+
+    # ### end Alembic commands ###

nonebot_plugin_tetris_stats/config/migrations/b9d65badc713_del_old_tos_bind_data.py

@@ -0,0 +1,42 @@
+"""Del old TOS bind data
+
+迁移 ID: b9d65badc713
+父迁移: 6c3206f90cc3
+创建时间: 2023-12-30 00:27:40.991704
+
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from alembic import op
+from sqlalchemy.ext.automap import automap_base
+from sqlalchemy.orm import Session
+
+if TYPE_CHECKING:
+    from collections.abc import Sequence
+
+revision: str = 'b9d65badc713'
+down_revision: str | Sequence[str] | None = '6c3206f90cc3'
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade(name: str = '') -> None:
+    if name:
+        return
+
+    Base = automap_base()  # noqa: N806
+    connection = op.get_bind()
+    Base.prepare(autoload_with=connection)
+
+    Bind = Base.classes.nonebot_plugin_tetris_stats_bind  # noqa: N806
+    with Session(connection) as session:
+        session.query(Bind).filter(Bind.game_platform == 'TOS').delete()
+        session.commit()
+
+
+def downgrade(name: str = '') -> None:
+    if name:
+        return

nonebot_plugin_tetris_stats/db/__init__.py

@@ -0,0 +1,135 @@
+from asyncio import Lock
+from collections.abc import AsyncGenerator
+from contextlib import asynccontextmanager
+from datetime import datetime, timezone
+from enum import Enum, auto
+from typing import TYPE_CHECKING, Literal, TypeVar, overload
+
+from nonebot.exception import FinishedException
+from nonebot.log import logger
+from nonebot_plugin_orm import AsyncSession, get_session
+from nonebot_plugin_user import User  # type: ignore[import-untyped]
+from sqlalchemy import select
+
+from ..utils.typing import CommandType, GameType
+from .models import Bind, TriggerHistoricalData
+
+UTC = timezone.utc
+
+if TYPE_CHECKING:
+    from ..games.tetrio.api.models import TETRIOHistoricalData
+    from ..games.top.api.models import TOPHistoricalData
+    from ..games.tos.api.models import TOSHistoricalData
+
+
+class BindStatus(Enum):
+    SUCCESS = auto()
+    UPDATE = auto()
+
+
+async def query_bind_info(
+    session: AsyncSession,
+    user: User,
+    game_platform: GameType,
+) -> Bind | None:
+    return (
+        await session.scalars(select(Bind).where(Bind.user_id == user.id).where(Bind.game_platform == game_platform))
+    ).one_or_none()
+
+
+async def create_or_update_bind(
+    session: AsyncSession,
+    user: User,
+    game_platform: GameType,
+    game_account: str,
+) -> BindStatus:
+    bind = await query_bind_info(
+        session=session,
+        user=user,
+        game_platform=game_platform,
+    )
+    if bind is None:
+        bind = Bind(
+            user_id=user.id,
+            game_platform=game_platform,
+            game_account=game_account,
+        )
+        session.add(bind)
+        message = BindStatus.SUCCESS
+    else:
+        bind.game_account = game_account
+        message = BindStatus.UPDATE
+    await session.commit()
+    return message
+
+
+T = TypeVar('T', 'TETRIOHistoricalData', 'TOPHistoricalData', 'TOSHistoricalData')
+
+lock = Lock()
+
+
+async def anti_duplicate_add(cls: type[T], model: T) -> None:
+    async with lock, get_session() as session:
+        result = (
+            await session.scalars(
+                select(cls)
+                .where(cls.update_time == model.update_time)
+                .where(cls.user_unique_identifier == model.user_unique_identifier)
+                .where(cls.api_type == model.api_type)
+            )
+        ).all()
+        if result:
+            for i in result:
+                if i.data == model.data:
+                    logger.debug('Anti duplicate successfully')
+                    return
+        session.add(model)
+        await session.commit()
+
+
+@asynccontextmanager
+@overload
+async def trigger(
+    session_persist_id: int,
+    game_platform: Literal['IO'],
+    command_type: CommandType | Literal['rank', 'config', 'record'],
+    command_args: list[str],
+) -> AsyncGenerator:
+    yield
+
+
+@asynccontextmanager
+@overload
+async def trigger(
+    session_persist_id: int,
+    game_platform: GameType,
+    command_type: CommandType,
+    command_args: list[str],
+) -> AsyncGenerator:
+    yield
+
+
+@asynccontextmanager
+async def trigger(
+    session_persist_id: int,
+    game_platform: GameType,
+    command_type: CommandType | Literal['rank', 'config', 'record'],
+    command_args: list[str],
+) -> AsyncGenerator:
+    trigger_time = datetime.now(UTC)
+    try:
+        yield
+    except FinishedException:
+        async with get_session() as session:
+            session.add(
+                TriggerHistoricalData(
+                    trigger_time=trigger_time,
+                    session_persist_id=session_persist_id,
+                    game_platform=game_platform,
+                    command_type=command_type,
+                    command_args=command_args,
+                    finish_time=datetime.now(UTC),
+                )
+            )
+            await session.commit()
+        raise

nonebot_plugin_tetris_stats/db/models.py

@@ -0,0 +1,81 @@
+from collections.abc import Callable, Sequence
+from datetime import datetime
+from typing import Any, Literal
+
+from nonebot.compat import PYDANTIC_V2, type_validate_json
+from nonebot_plugin_orm import Model
+from pydantic import BaseModel, ValidationError
+from sqlalchemy import JSON, DateTime, Dialect, String, TypeDecorator
+from sqlalchemy.orm import Mapped, MappedAsDataclass, mapped_column
+from typing_extensions import override
+
+from ..utils.typing import CommandType, GameType
+
+
+class PydanticType(TypeDecorator):
+    impl = JSON
+
+    @override
+    def __init__(
+        self,
+        get_model: Sequence[Callable[[], Sequence[type[BaseModel]]]],
+        models: set[type[BaseModel]],
+        *args: Any,
+        **kwargs: Any,
+    ):
+        self.get_model = get_model
+        self._models = models
+        super().__init__(*args, **kwargs)
+
+    if PYDANTIC_V2:
+
+        @override
+        def process_bind_param(self, value: Any | None, dialect: Dialect) -> str:
+            # 将 Pydantic 模型实例转换为 JSON
+            if isinstance(value, tuple(self.models)):
+                return value.model_dump_json(by_alias=True)  # type: ignore[union-attr]
+            raise TypeError
+    else:
+
+        @override
+        def process_bind_param(self, value: Any | None, dialect: Dialect) -> str:
+            # 将 Pydantic 模型实例转换为 JSON
+            if isinstance(value, tuple(self.models)):
+                return value.json(by_alias=True)  # type: ignore[union-attr]
+            raise TypeError
+
+    @override
+    def process_result_value(self, value: Any | None, dialect: Dialect) -> BaseModel:
+        # 将 JSON 转换回 Pydantic 模型实例
+        if isinstance(value, str | bytes):
+            for i in self.models:
+                try:
+                    return type_validate_json(i, value)
+                except ValidationError:  # noqa: PERF203
+                    ...
+        raise ValueError
+
+    @property
+    def models(self) -> tuple[type[BaseModel], ...]:
+        models: set[type[BaseModel]] = set()
+        for i in self.get_model:
+            models.update(i())
+        models.update(self._models)
+        return tuple(models)
+
+
+class Bind(MappedAsDataclass, Model):
+    id: Mapped[int] = mapped_column(init=False, primary_key=True)
+    user_id: Mapped[int] = mapped_column(index=True)
+    game_platform: Mapped[GameType] = mapped_column(String(32))
+    game_account: Mapped[str]
+
+
+class TriggerHistoricalData(MappedAsDataclass, Model):
+    id: Mapped[int] = mapped_column(init=False, primary_key=True)
+    trigger_time: Mapped[datetime] = mapped_column(DateTime)
+    session_persist_id: Mapped[int]
+    game_platform: Mapped[GameType] = mapped_column(String(32), index=True)
+    command_type: Mapped[CommandType | Literal['rank', 'config', 'record']] = mapped_column(String(16), index=True)
+    command_args: Mapped[list[str]] = mapped_column(JSON)
+    finish_time: Mapped[datetime] = mapped_column(DateTime)

nonebot_plugin_tetris_stats/game_data_processor/__init__.py

@@ -1 +0,0 @@
-from . import io_data_processor, top_data_processor, tos_data_processor

nonebot_plugin_tetris_stats/game_data_processor/io_data_processor/__init__.py

@@ -1,40 +0,0 @@
-from re import I
-
-from nonebot import on_regex
-from nonebot.adapters.onebot.v11 import GROUP, MessageEvent
-from nonebot.matcher import Matcher
-
-from .processor import Processor
-
-IOBind = on_regex(pattern=r'^io绑定|^iobind', flags=I, permission=GROUP)
-IOStats = on_regex(pattern=r'^io查|^iostats', flags=I, permission=GROUP)
-IORank = on_regex(pattern=r'^io段位|^iorank', flags=I, permission=GROUP)
-
-@IOBind.handle()
-async def _(event: MessageEvent, matcher: Matcher):
-    await matcher.finish(
-        await Processor.handle_bind(
-            message=event.raw_message,
-            qq_number=event.sender.user_id
-        )
-    )
-
-
-@IOStats.handle()
-async def _(event: MessageEvent, matcher: Matcher):
-    if event.is_tome():
-        await matcher.finish('不能查询bot的信息')
-    await matcher.finish(
-        await Processor.handle_query(
-            message=event.raw_message,
-            qq_number=event.sender.user_id
-        )
-    )
-
-@IORank.handle()
-async def _(event: MessageEvent, matcher: Matcher):
-    await matcher.finish(
-        await Processor.handle_rank(
-            message=event.raw_message
-        )
-    )

nonebot_plugin_tetris_stats/game_data_processor/io_data_processor/processor.py

@@ -1,289 +0,0 @@
-import math
-from asyncio import gather
-from typing import Any
-
-from nonebot.log import logger
-
-from ...utils.database import DataBase
-from ...utils.message_analyzer import (
-    handle_bind_message,
-    handle_rank_message,
-    handle_stats_query_message,
-)
-from .request import Request
-
-
-class Processor:
-    @classmethod
-    async def handle_bind(cls, message: str, qq_number: int | None) -> str:
-        '''处理绑定消息'''
-        decoded_message = await handle_bind_message(message=message, game_type='IO')
-        if decoded_message[0] is None:
-            return decoded_message[1][0]
-        if decoded_message[0] == 'ID':
-            user_id_stats = await cls.check_user_id(decoded_message[1][1])
-            if user_id_stats[0] is False:
-                return user_id_stats[1]
-            user_id = decoded_message[1][1]
-        elif decoded_message[0] == 'Name':
-            user_data = await cls.get_user_data(user_name=decoded_message[1][1])
-            if user_data[0] is False:
-                return '用户信息请求失败'
-            if user_data[1] is False:
-                return f'用户信息请求错误:\n{user_data[2]["error"]}'
-            user_id = await cls.get_user_id(user_data[2])
-            if qq_number is None:  # 理论上是不会有None出现的, ide快乐行属于是（
-                logger.error('获取QQ号失败')
-                return '获取QQ号失败'
-            return (
-                await DataBase.write_bind_info(
-                    qq_number=qq_number,
-                    user=user_id,
-                    game_type='IO'
-                )
-            )
-        logger.error('预期外行为, 请上报GitHub')
-        return '出现预期外行为，请查看后台信息'
-
-    @classmethod
-    async def handle_rank(cls, message: str):
-        query_rank = await handle_rank_message(message)
-        rank_info = await DataBase.query_rank_info_today(rank=query_rank.lower())
-
-        if rank_info is None:
-            ranks_percentiles = {
-                'x': 1,
-                'u': 5,
-                'ss': 11,
-                's+': 17,
-                's': 23,
-                's-': 30,
-                'a+': 38,
-                'a': 46,
-                'a-': 54,
-                'b+': 62,
-                'b': 70,
-                'b-': 78,
-                'c+': 84,
-                'c': 90,
-                'c-': 95,
-                'd+': 97.5,
-                'd': 100,
-            }
-
-            if query_rank.lower() not in (i for i in ranks_percentiles.keys()):
-                return '未知段位'
-
-            result = await Request.request('https://ch.tetr.io/api/users/lists/league/all')
-            users: list = result[2]['data']['users']
-
-            def avg(rank_users: list, column: str, playercount: int | None = None) -> float:
-                return sum(i['league'][column] for i in rank_users) / (playercount or len(rank_users))
-
-            for rank, percentile in ranks_percentiles.items():
-                offset = math.floor((percentile / 100) * len(users)) - 1
-                tr = users[offset]['league']['rating']
-
-                rank_users = list(filter(lambda x: x['league']['rank'] == rank, users))
-                playercount = len(rank_users)
-
-                avg_apm = avg(rank_users, 'apm', playercount)
-                avg_pps = avg(rank_users, 'pps', playercount)
-                avg_vs = avg(rank_users, 'vs', playercount)
-
-                await DataBase.write_rank_info_today(
-                    rank=rank,
-                    trline=tr,
-                    playercount=playercount,
-                    avgapm=avg_apm,
-                    avgpps=avg_pps,
-                    avgvs=avg_vs
-                    )
-
-            return await Processor.handle_rank(message=message)
-        else:
-            avg_apm = round(rank_info[3], 2)
-            avg_pps = round(rank_info[4], 2)
-            avg_vs = round(rank_info[5], 2)
-            avg_lpm = round((avg_pps * 24), 2)
-            avg_apl = round((avg_apm / avg_lpm), 2)
-            avg_adpm = round((avg_vs * 0.6), 2)
-            avg_adpl = round((avg_adpm / avg_lpm), 2)
-
-            message = f'{query_rank.upper()} 段, 分数线 {round(rank_info[1], 2)} TR, {rank_info[2]} 名玩家'
-            message += f'\n对比昨日趋势: {rank_info[0]}'
-            message += '\n平均数据: '
-            message += f"\nL'PM: {avg_lpm} ( {avg_pps} pps )"
-            message += f'\nAPM: {avg_apm} ( x{avg_apl} )'
-            message += f'\nADPM: {avg_adpm} ( x{avg_adpl} ) ( {avg_vs}vs )'
-            message += '\n'
-            message += f'\n数据更新时间: {rank_info[6]}'
-
-        return message
-
-    @classmethod
-    async def handle_query(cls, message: str, qq_number: int | None):
-        '''处理查询消息'''
-        decoded_message = await handle_stats_query_message(message=message, game_type='IO')
-        if decoded_message[0] is None:
-            return decoded_message[1][0]
-        if decoded_message[0] == 'AT':  # 在入口处判断是否@bot本身
-            bind_info = await DataBase.query_bind_info(qq_number=decoded_message[1][1], game_type='IO')
-            if bind_info is None:
-                return '未查询到绑定信息'
-            return f'* 由于无法验证绑定信息, 不能保证查询到的用户为本人\n{await Processor.generate_message(user_id=bind_info)}'
-        if decoded_message[0] == 'ME':
-            if qq_number is None:
-                logger.error('获取QQ号失败')
-                return '获取QQ号失败, 请联系bot主人'
-            bind_info = await DataBase.query_bind_info(qq_number=qq_number, game_type='IO')
-            if bind_info is None:
-                return '未查询到绑定信息'
-            return f'* 由于无法验证绑定信息, 不能保证查询到的用户为本人\n{await Processor.generate_message(user_id=bind_info)}'
-        if decoded_message[0] == 'ID':
-            return await Processor.generate_message(user_id=decoded_message[1][1])
-        if decoded_message[0] == 'Name':
-            return await Processor.generate_message(user_name=decoded_message[1][1])
-
-    @classmethod
-    async def get_user_data(
-            cls,
-            user_name: str | None = None,
-            user_id: str | None = None
-    ) -> tuple[bool, bool, dict[str, Any]]:
-        '''获取用户数据'''
-        if user_name is not None and user_id is None:
-            user_data_url = f'https://ch.tetr.io/api/users/{user_name}'
-        elif user_name is None and user_id is not None:
-            user_data_url = f'https://ch.tetr.io/api/users/{user_id}'
-        else:
-            raise ValueError('预期外行为, 请上报GitHub')
-        return await Request.request(user_data_url)
-
-    @classmethod
-    async def get_solo_data(
-            cls,
-            user_name: str | None = None,
-            user_id: str | None = None
-    ) -> tuple[bool, bool, dict[str, Any]]:
-        '''获取Solo数据'''
-        if user_name is not None and user_id is None:
-            user_solo_url = f'https://ch.tetr.io/api/users/{user_name}/records'
-        elif user_name is None and user_id is not None:
-            user_solo_url = f'https://ch.tetr.io/api/users/{user_id}/records'
-        else:
-            raise ValueError('预期外行为, 请上报GitHub')
-        return await Request.request(user_solo_url)
-
-    @classmethod
-    async def get_user_id(cls, user_data: dict) -> str:
-        '''获取用户ID'''
-        return user_data['data']['user']['_id']
-
-    @classmethod
-    async def check_user_id(cls, user_id: str) -> tuple[bool, str]:
-        '''检查用户ID是否有效 返回值为tuple[bool, message]'''
-        user_data = await cls.get_user_data(user_id=user_id)
-        if user_data[0] is False:
-            return False, '用户信息请求失败'
-        if user_data[1] is False:
-            return False, f'用户信息请求错误:\n{user_data[2]["error"]}'
-        if user_id == user_data[2]['data']['user']['_id']:
-            return True, ''
-        raise ValueError('服务器返回的userID和用户提供的不一致, 这种情况理论上不应该发生, 以防万一还是写一下（x')
-
-    @classmethod
-    async def get_league_stats(cls, user_data: dict) -> dict[str, Any]:
-        '''获取排位统计数据'''
-        league = user_data['data']['user']['league']
-        league_stats: dict[str, Any] = {}
-        if league['gamesplayed'] != 0:
-            league_stats['PPS'] = league['pps']
-            league_stats['APM'] = league['apm']
-            league_stats['VS'] = 0 if league['vs'] is None else league['vs']
-            league_stats['Rank'] = 'Z' if league['rank'] == 'z' else league['rank'].upper(
-            )
-            if league['rating'] == -1:
-                league_stats['Rank'] = None
-            else:
-                league_stats['Rating'] = round(league['rating'], 2)
-                league_stats['Glicko'] = round(league['glicko'], 2)
-                league_stats['RD'] = round(league['rd'], 2)
-            league_stats['Standing'] = league['standing']
-            league_stats['LPM'] = round((league['pps'] * 24), 2)
-            league_stats['APL'] = round(
-                (league_stats['APM'] / league_stats['LPM']), 2)
-            league_stats['ADPM'] = round((league_stats['VS'] * 0.6), 2)
-            league_stats['ADPL'] = round(
-                (league_stats['ADPM'] / league_stats['LPM']), 2)
-        return league_stats
-
-    @classmethod
-    async def get_sprint_stats(cls, solo_data: dict) -> dict[str, Any]:
-        '''获取40L统计数据'''
-        sprint_stats = {}
-        solo = solo_data['data']['records']['40l']
-        if solo['record'] is not None:
-            sprint_stats['Time'] = round(
-                solo['record']['endcontext']['finalTime'] / 1000, 2)
-            if solo['rank'] is not None:
-                sprint_stats['Rank'] = solo['rank']
-        return sprint_stats
-
-    @classmethod
-    async def get_blitz_stats(cls, solo_data: dict) -> dict[str, Any]:
-        '''获取Blitz统计数据'''
-        blitz_stats = {}
-        blitz = solo_data['data']['records']['blitz']
-        if blitz['record'] is not None:
-            blitz_stats['Score'] = blitz['record']['endcontext']['score']
-            if blitz['rank'] is not None:
-                blitz_stats['Rank'] = blitz['rank']
-        return blitz_stats
-
-    @classmethod
-    async def generate_message(
-            cls,
-            user_name: str | None = None,
-            user_id: str | None = None
-    ) -> str:
-        '''生成消息'''
-        user_data, solo_data = await gather(
-            cls.get_user_data(user_name=user_name, user_id=user_id),
-            cls.get_solo_data(user_name=user_name, user_id=user_id)
-        )
-        if user_data[0] is False:
-            return '用户信息请求失败'
-        if user_data[1] is False:
-            return f'用户信息请求错误:\n{user_data[2]["error"]}'
-        user_name = user_data[2]['data']['user']['username'].upper()
-        league_stats = await cls.get_league_stats(user_data[2])
-        message = ''
-        if not league_stats:
-            message += f'用户 {user_name} 没有排位统计数据'
-        else:
-            if league_stats['Rank'] is None:
-                message += f'用户 {user_name} 暂未完成定级赛, 最近十场的数据:'
-            else:
-                if league_stats['Rank'] == 'Z':
-                    message += f'用户 {user_name} 暂无段位, {league_stats["Rating"]} TR'
-                else:
-                    message += f'{league_stats["Rank"]} 段用户 {user_name} {league_stats["Rating"]} TR (#{league_stats["Standing"]})'
-                message += f', 段位分 {league_stats["Glicko"]}±{league_stats["RD"]}, 最近十场的数据:'
-            message += f'\nL\'PM: {league_stats["LPM"]} ( {league_stats["PPS"]} pps )'
-            message += f'\nAPM: {league_stats["APM"]} ( x{league_stats["APL"]} )'
-            if league_stats["VS"] != 0:
-                message += f'\nADPM: {league_stats["ADPM"]} ( x{league_stats["ADPL"]} ) ( {league_stats["VS"]}vs )'
-        if solo_data[0] is False:
-            return f'{message}\nSolo统计数据请求失败'
-        if solo_data[1] is False:
-            return f'{message}\nSolo统计数据请求错误:\n{solo_data[2]["error"]}'
-        sprint_stats, blitz_stats = await gather(
-            cls.get_sprint_stats(solo_data[2]),
-            cls.get_blitz_stats(solo_data[2])
-        )
-        message += f'\n40L: {sprint_stats["Time"]}s' if 'Time' in sprint_stats else ''
-        message += f' ( #{sprint_stats["Rank"]} )' if 'Rank' in sprint_stats else ''
-        message += f'\nBlitz: {blitz_stats["Score"]}' if 'Score' in blitz_stats else ''
-        message += f' ( #{blitz_stats["Rank"]} )' if 'Rank' in blitz_stats else ''
-        return message

nonebot_plugin_tetris_stats/game_data_processor/io_data_processor/request.py

@@ -1,155 +0,0 @@
-import os
-from typing import Any
-
-import aiohttp
-from nonebot import get_driver
-from nonebot.log import logger
-from playwright.async_api import Browser, Response, async_playwright
-from ujson import JSONDecodeError, dumps, loads
-
-from ...utils.config import Config
-
-driver = get_driver()
-
-config = Config.parse_obj(get_driver().config)
-
-
-@driver.on_startup
-async def _():
-    await Request.init_cache()
-    await Request.read_cache()
-
-
-@driver.on_shutdown
-async def _():
-    await Request.close_browser()
-    await Request.write_cache()
-
-
-class Request:
-    '''网络请求相关类'''
-    _browser: Browser | None = None
-    _headers: dict | None = None
-    _cookies: dict | None = None
-
-    @classmethod
-    async def _init_playwright(cls) -> Browser:
-        '''初始化playwright'''
-        playwright = await async_playwright().start()
-        cls._browser = await playwright.firefox.launch()
-        return cls._browser
-
-    @classmethod
-    async def _get_browser(cls) -> Browser:
-        '''获取浏览器对象'''
-        return cls._browser or await cls._init_playwright()
-
-    @classmethod
-    async def _anti_cloudflare(cls, url: str) -> tuple[bool, bool, dict[str, Any]]:
-        '''用firefox硬穿五秒盾'''
-        browser = await cls._get_browser()
-        context = await browser.new_context()
-        page = await context.new_page()
-        response = await page.goto(url)
-        attempts = 0
-        while attempts < 60:
-            attempts += 1
-            text = await page.locator("body").text_content()
-            if text is None:
-                await page.wait_for_timeout(1000)
-                continue
-            if await page.title() == 'Please Wait... | Cloudflare':
-                # TODO 有无人来做一个过验证码（
-                break
-            try:
-                data = loads(text)
-            except JSONDecodeError:
-                await page.wait_for_timeout(1000)
-            else:
-                assert isinstance(response, Response)
-                cls._headers = await response.request.all_headers()
-                try:
-                    cls._cookies = {i['name']: i['value'] for i in await context.cookies()}
-                except KeyError:
-                    cls._cookies = None
-                await page.close()
-                await context.close()
-                return True, data['success'], data
-        await page.close()
-        await context.close()
-        return True, False, {'error': '绕过五秒盾失败'}
-
-    @classmethod
-    async def init_cache(cls) -> None:
-        '''初始化缓存文件'''
-        if not os.path.exists(os.path.dirname(config.cache_path)):
-            os.makedirs(os.path.dirname(config.cache_path))
-        if not os.path.exists(config.cache_path):
-            with open(file=config.cache_path, mode='w', encoding='UTF-8') as file:
-                file.write(
-                    dumps(
-                        {
-                            'headers': cls._headers,
-                            'cookies': cls._cookies
-                        }
-                    )
-                )
-
-    @classmethod
-    async def read_cache(cls) -> None:
-        '''读取缓存文件'''
-        try:
-            with open(file=config.cache_path, mode='r', encoding='UTF-8') as file:
-                json = loads(file.read())
-                cls._headers = json['headers']
-                cls._cookies = json['cookies']
-        except FileNotFoundError:
-            await cls.init_cache()
-        except PermissionError:
-            os.remove(config.cache_path)
-            await cls.init_cache()
-        except JSONDecodeError:
-            os.remove(config.cache_path)
-            await cls.init_cache()
-
-    @classmethod
-    async def write_cache(cls) -> None:
-        '''写入缓存文件'''
-        try:
-            with open(file=config.cache_path, mode='r+', encoding='UTF-8') as file:
-                file.write(
-                    dumps(
-                        {
-                            'headers': cls._headers,
-                            'cookies': cls._cookies
-                        }
-                    )
-                )
-        except FileNotFoundError:
-            await cls.init_cache()
-        except PermissionError:
-            os.remove(config.cache_path)
-            await cls.init_cache()
-        except JSONDecodeError:
-            os.remove(config.cache_path)
-            await cls.init_cache()
-
-    @classmethod
-    async def request(cls, url: str) -> tuple[bool, bool, dict[str, Any]]:
-        '''请求api'''
-        try:
-            async with aiohttp.ClientSession(cookies=cls._cookies) as session:
-                async with session.get(url, headers=cls._headers) as resp:
-                    data = await resp.json()
-                    return True, data['success'], data
-        except aiohttp.client_exceptions.ClientConnectorError as error: # type: ignore
-            logger.error(f'请求错误\n{error}')
-            return False, False, {}
-        except aiohttp.client_exceptions.ContentTypeError:  # type: ignore
-            return await cls._anti_cloudflare(url)
-
-    @classmethod
-    async def close_browser(cls) -> None:
-        '''关闭浏览器对象'''
-        if isinstance(cls._browser, Browser):
-            await cls._browser.close()

nonebot_plugin_tetris_stats/game_data_processor/top_data_processor.py

@@ -1,196 +0,0 @@
-from asyncio import gather
-from re import I
-from typing import Any
-
-import aiohttp
-from lxml import etree
-from nonebot import on_regex
-from nonebot.adapters.onebot.v11 import GROUP, MessageEvent
-from nonebot.log import logger
-from nonebot.matcher import Matcher
-from pandas import read_html
-
-from ..utils.database import DataBase
-from ..utils.message_analyzer import (
-    handle_bind_message,
-    handle_stats_query_message
-)
-
-TOPBind = on_regex(pattern=r'^top绑定|^topbind', flags=I, permission=GROUP)
-TopStats = on_regex(pattern=r'^top查|^topstats', flags=I, permission=GROUP)
-
-
-@TOPBind.handle()
-async def _(event: MessageEvent, matcher: Matcher):
-    decoded_message = await handle_bind_message(message=event.raw_message, game_type='TOP')
-    if decoded_message[0] is None:
-        await matcher.finish(decoded_message[1][0])
-    elif decoded_message[0] == 'Name':
-        user_data = await get_user_data(decoded_message[1][1])
-        if user_data[0] is False:
-            await matcher.finish('用户信息请求失败')
-        else:
-            if await check_user(user_data[1]) is False:
-                await matcher.finish('用户不存在')
-            user_name = await get_user_name(user_data[1])
-            if event.sender.user_id is None:  # 理论上是不会有None出现的, ide快乐行属于是（
-                logger.error('获取QQ号失败')
-                await matcher.finish('获取QQ号失败')
-            await matcher.finish(
-                await DataBase.write_bind_info(
-                    qq_number=event.sender.user_id,
-                    user=user_name,
-                    game_type='TOP'
-                )
-            )
-
-
-@TopStats.handle()
-async def _(event: MessageEvent, matcher: Matcher):
-    decoded_message = await handle_stats_query_message(message=event.raw_message, game_type='TOP')
-    if decoded_message[0] is None:
-        await matcher.finish(decoded_message[1][0])
-    elif decoded_message[0] == 'AT':
-        if event.is_tome() is True:
-            await matcher.finish('不能查询bot的信息')
-        bind_info = await DataBase.query_bind_info(qq_number=decoded_message[1][1], game_type='TOP')
-        if bind_info is None:
-            message = '未查询到绑定信息'
-        else:
-            message = (f'* 由于无法验证绑定信息, 不能保证查询到的用户为本人\n{await generate_message(bind_info)}')
-    elif decoded_message[0] == 'ME':
-        if event.sender.user_id is None:
-            logger.error('获取QQ号失败')
-            await matcher.finish('获取QQ号失败, 请联系bot主人')
-        bind_info = await DataBase.query_bind_info(qq_number=event.sender.user_id, game_type='TOP')
-        if bind_info is None:
-            message = '未查询到绑定信息'
-        else:
-            message = (f'* 由于无法验证绑定信息, 不能保证查询到的用户为本人\n{await generate_message(bind_info)}')
-    elif decoded_message[0] == 'Name':
-        message = await generate_message(decoded_message[1][1])
-    else:
-        raise ValueError('预期外行为, 请上报GitHub')
-    await matcher.finish(message)
-
-
-async def get_user_data(user_name: str) -> tuple[bool, str]:
-    '''获取用户信息'''
-    url = f'http://tetrisonline.pl/top/profile.php?user={user_name}'
-    try:
-        async with aiohttp.ClientSession() as session:
-            async with session.get(url) as resp:
-                return True, await resp.text()
-    except aiohttp.client_exceptions.ClientConnectorError as error: # type: ignore
-        logger.error(error)
-        return False, ''
-
-
-async def check_user(user_data: str) -> bool:
-    '''如果用户存在返回True, 如果用户不存在返回False'''
-    return user_data.find('user not found!') == -1
-
-
-async def get_user_name(user_data: str) -> str:
-    '''获取用户名'''
-    data = etree.HTML(user_data).xpath('//div[@class="mycontent"]/h1/text()')
-    if isinstance(data, list):
-        return str(data[0]).replace('\'s profile', '')
-    raise TypeError('预期外行为, 请上报GitHub')
-
-
-async def get_game_stats(user_data: str) -> dict[str, dict[str, Any]]:
-    '''获取游戏统计数据'''
-    game_stats: dict[str, Any] = {'24H': {}, 'All': {}}
-    html = etree.HTML(user_data)
-    data = html.xpath('//div[@class="mycontent"]//text()')
-    if isinstance(data, list):
-        for i in data:
-            if not isinstance(i, str):
-                i = str(i)
-            i = i.strip()
-            if i.startswith('lpm:'):
-                game_stats['24H']['LPM'] = i.replace('lpm:', '').strip()
-            if i.startswith('apm:'):
-                game_stats['24H']['APM'] = i.replace('apm:', '').strip()
-            if 'LPM' in game_stats['24H'] and 'APM' in game_stats['24H']:
-                break
-    else:
-        raise TypeError('预期外行为, 请上报GitHub')
-    # 如果没有24H统计数据
-    if game_stats['24H'].get('LPM') in [None, ''] or game_stats['24H'].get('APM') in [None, '']:
-        game_stats['24H'].pop('LPM', None)
-        game_stats['24H'].pop('APM', None)
-    else:
-        game_stats['24H']['PPS'] = round(
-            float(game_stats['24H']['LPM']) / 24, 2)
-        game_stats['24H']['APL'] = round(
-            float(game_stats['24H']['APM']) / float(game_stats['24H']['LPM']), 2)
-        game_stats['24H']['LPM'] = round(float(game_stats['24H']['LPM']), 2)
-        game_stats['24H']['APM'] = round(float(game_stats['24H']['APM']), 2)
-    table = html.xpath('//table')
-    if isinstance(table, list):
-        if isinstance(table[0], etree._Element):
-            stats_table = etree.tostring(table[0], encoding='utf-8').decode()
-            df = read_html(stats_table, encoding='utf-8', header=0)[0]
-            results = df.T.to_dict().values()
-            if results:
-                game_stats['All']['LPM'] = 0
-                game_stats['All']['APM'] = 0
-                for i in results:
-                    if isinstance(i, dict):
-                        game_stats['All']['LPM'] += i['lpm']
-                        game_stats['All']['APM'] += i['apm']
-                game_stats['All']['LPM'] = game_stats['All']['LPM'] / \
-                    len(results)
-                game_stats['All']['APM'] = game_stats['All']['APM'] / \
-                    len(results)
-                game_stats['All']['PPS'] = round(
-                    game_stats['All']['LPM'] / 24, 2)
-                game_stats['All']['APL'] = round(
-                    float(game_stats['All']['APM']) / float(game_stats['All']['LPM']), 2)
-                game_stats['All']['LPM'] = round(
-                    float(game_stats['All']['LPM']), 2)
-                game_stats['All']['APM'] = round(
-                    float(game_stats['All']['APM']), 2)
-    else:
-        raise TypeError('预期外行为, 请上报GitHub')
-    return game_stats
-
-
-async def generate_message(user_name: str) -> str:
-    '''生成消息'''
-    user_data = await get_user_data(user_name)
-    if user_data[0] is False:
-        return '用户信息请求失败'
-    if await check_user(user_data[1]) is False:
-        return '用户不存在'
-    user_name, game_stats = await gather(
-        get_user_name(user_data[1]),
-        get_game_stats(user_data[1])
-    )
-    message = ''
-    if game_stats['24H'] and game_stats['All']:
-        message += f'用户 {user_name} 24小时内统计数据为: '
-        message += f'\nL\'PM: {game_stats["24H"]["LPM"]} ( {game_stats["24H"]["PPS"]} pps )'
-        message += f'\nAPM: {game_stats["24H"]["APM"]} ( x{game_stats["24H"]["APL"]} )'
-        message += '\n历史统计数据为: '
-        message += f'\nL\'PM: {game_stats["All"]["LPM"]} ( {game_stats["All"]["PPS"]} pps )'
-        message += f'\nAPM: {game_stats["All"]["APM"]} ( x{game_stats["All"]["APL"]} )'
-    elif game_stats['24H'] and not game_stats['All']:
-        message += f'用户 {user_name} 24小时内统计数据为: '
-        message += f'\nL\'PM: {game_stats["24H"]["LPM"]} ( {game_stats["24H"]["PPS"]} pps )'
-        message += f'\nAPM: {game_stats["24H"]["APM"]} ( x{game_stats["24H"]["APL"]} )'
-        message += '\n暂无历史统计数据'
-        message += '\n( 这理论上不该存在, 如果你看到了, 请联系bot主人查看后台'
-        logger.error(f'老实说这个不算Error, 但是理论上不应该有, 如果你看到了这条日志, 我希望你能来Github发个issue（\
-user_name: {user_name}\
-user_data: {user_data}\
-game_stats: {game_stats}')
-    elif not game_stats['24H'] and game_stats['All']:
-        message += f'用户 {user_name} 暂无24小时内统计数据, 历史统计数据为: '
-        message += f'\nL\'PM: {game_stats["All"]["LPM"]} ( {game_stats["All"]["PPS"]} pps )'
-        message += f'\nAPM: {game_stats["All"]["APM"]} ( x{game_stats["All"]["APL"]} )'
-    else:
-        message += f'用户 {user_name} 暂无24小时内统计数据, 暂无历史统计数据'
-    return message

nonebot_plugin_tetris_stats/game_data_processor/tos_data_processor.py

@@ -1,179 +0,0 @@
-from asyncio import gather
-from re import I
-from typing import Any
-
-import aiohttp
-from nonebot import on_regex
-from nonebot.adapters.onebot.v11 import GROUP, MessageEvent
-from nonebot.log import logger
-from nonebot.matcher import Matcher
-
-from ..utils.message_analyzer import handle_stats_query_message
-
-TOSStats = on_regex(
-    pattern=r'^tos查|^tostats|^tosstats|^茶服查|^茶服stats',
-    flags=I,
-    permission=GROUP
-)
-
-
-@TOSStats.handle()
-async def _(event: MessageEvent, matcher: Matcher):
-    decoded_message = await handle_stats_query_message(message=event.raw_message, game_type='TOS')
-    if decoded_message[0] is None:
-        await matcher.finish(decoded_message[1][0])
-    elif decoded_message[0] == 'AT' or decoded_message[0] == 'QQ':
-        if decoded_message[1][1] == event.self_id:
-            await matcher.finish('不能查询bot的信息')
-        message = await generate_message(tea_id=decoded_message[1][1])
-    elif decoded_message[0] == 'ME':
-        message = await generate_message(tea_id=event.sender.user_id)
-    elif decoded_message[0] == 'Name':
-        message = await generate_message(user_name=decoded_message[1][1])
-    else:
-        raise ValueError('预期外行为, 请上报GitHub')
-    await matcher.finish(message)
-
-
-async def request(url: str) -> tuple[bool, bool, dict[str, Any]]:
-    '''请求api'''
-    try:
-        async with aiohttp.ClientSession() as session:
-            async with session.get(url) as resp:
-                data = await resp.json()
-                return True, data['success'], data
-    except aiohttp.client_exceptions.ClientConnectorError as error: # type: ignore
-        logger.error(f'请求错误\n{error}')
-        return False, False, {}
-
-
-async def get_user_info(
-    user_name: str | None = None,
-    tea_id: int | None = None
-) -> tuple[bool, bool, dict[str, Any]]:
-    '''获取用户信息'''
-    if user_name is not None and tea_id is None:
-        user_data_url = f'https://teatube.cn:8888/getUsernameInfo?username={user_name}'
-    elif user_name is None and tea_id is not None:
-        user_data_url = f'https://teatube.cn:8888/getTeaIdInfo?teaId={tea_id}'
-    else:
-        raise ValueError('预期外行为, 请上报GitHub')
-    return await request(user_data_url)
-
-
-async def get_user_data(
-    user_name: str | None = None,
-    tea_id: int | None = None,
-    other_parameter: str = ''
-) -> tuple[bool, bool, dict[str, Any]]:
-    '''获取用户数据'''
-    if user_name is not None and tea_id is None:
-        user_data_url = f'https://teatube.cn:8888/getProfile?id={user_name}{other_parameter}'
-    elif user_name is None and tea_id is not None:
-        user_data_url = f'https://teatube.cn:8888/getProfile?id={tea_id}{other_parameter}'
-    else:
-        raise ValueError('预期外行为, 请上报GitHub')
-    return await request(user_data_url)
-
-
-async def get_rank_stats(user_info: dict) -> dict[str, float]:
-    '''获取Rank数据'''
-    data = user_info['data']
-    rank_stats = {}
-    if int(data['rankedGames']) != 0:
-        rank_stats['Rating'] = round(float(data['ratingNow']), 2)
-        rank_stats['RD'] = round(float(data['rdNow']), 2)
-        rank_stats['Vol'] = round(float(data['volNow']), 3)
-    return rank_stats
-
-
-async def get_game_data(user_data: dict) -> dict[str, int | float]:
-    '''获取游戏数据'''
-    game_data: dict[str, int | float] = {}
-    if user_data['data'] != []:
-        weighted_total_lpm = weighted_total_apm = weighted_total_adpm = total_time = num = 0
-        for i in user_data['data']:
-            # 排除单人局和时间为0的游戏
-            if i['num_players'] == 1 or i['time'] == 0:
-                continue
-            # 茶：不计算没挖掘的局, 即使apm和lpm也如此
-            if i['dig'] is None:
-                continue
-            # 加权计算
-            time = i['time'] / 1000
-            lpm = 24 * (i['pieces'] / time)
-            apm = (i['attack'] / time) * 60
-            adpm = ((i['attack'] + i['dig']) / time) * 60
-            weighted_total_lpm += lpm * time
-            weighted_total_apm += apm * time
-            weighted_total_adpm += adpm * time
-            total_time += time
-            num += 1
-            if num == 50:
-                break
-        if num > 0:
-            game_data['NUM'] = num
-            game_data['LPM'] = round((weighted_total_lpm / total_time), 2)
-            game_data['APM'] = round((weighted_total_apm / total_time), 2)
-            game_data['ADPM'] = round((weighted_total_adpm / total_time), 2)
-            game_data['PPS'] = round((game_data['LPM'] / 24), 2)
-            game_data['APL'] = round((game_data['APM'] / game_data['LPM']), 2)
-            game_data['ADPL'] = round(
-                (game_data['ADPM'] / game_data['LPM']), 2)
-            game_data['VS'] = round((game_data['ADPM'] / 60 * 100), 2)
-        # TODO: 如果有效局数不满50, 没有无dig信息的局, 且userData['data']内有50个局, 则继续往前获取信息
-    return game_data
-
-
-async def get_pb_data(user_info: dict) -> dict[str, float | str]:
-    '''获取PB数据'''
-    pb_data: dict[str, float | str] = {}
-    data = user_info['data']
-    if int(data['PBSprint']) != 2147483647:
-        pb_data['Sprint'] = round(float(data['PBSprint']) / 1000, 2)
-    if int(data['PBMarathon']) != 0:
-        pb_data['Marathon'] = data['PBMarathon']
-    if int(data['PBChallenge']) != 0:
-        pb_data['Challenge'] = data['PBChallenge']
-    return pb_data
-
-
-async def generate_message(
-    user_name: str | None = None,
-    tea_id: int | None = None
-) -> str:
-    '''生成消息'''
-    user_info, user_data = await gather(
-        get_user_info(user_name=user_name, tea_id=tea_id),
-        get_user_data(user_name=user_name, tea_id=tea_id)
-    )
-    if user_info[0] is False:
-        return '用户信息请求失败'
-    if user_info[1] is False:
-        return f'用户信息请求错误:\n{user_info[2]["error"]}'
-    rank_stats, pb_data = await gather(
-        get_rank_stats(user_info[2]),
-        get_pb_data(user_info[2])
-    )
-    message = f'用户 {user_info[2]["data"]["name"]} ({user_info[2]["data"]["teaId"]}) '
-    if not rank_stats:
-        message += '暂无段位统计数据'
-    else:
-        message += f', 段位分 {rank_stats["Rating"]}±{rank_stats["RD"]} ({rank_stats["Vol"]}) '
-    if user_data[0] is False:
-        message = f'{message.rstrip()}\n游戏数据请求失败'
-    elif user_data[1] is False:
-        message = f'{message.rstrip()}\n游戏数据请求错误:\n{user_data[2]["error"]}'
-    else:
-        game_data = await get_game_data(user_data[2])
-        if not game_data:
-            message += ', 暂无游戏数据'
-        else:
-            message += f', 最近 {game_data["NUM"]} 局数据'
-            message += f'\nL\'PM: {game_data["LPM"]} ( {game_data["PPS"]} pps )'
-            message += f'\nAPM：{game_data["APM"]} ( x{game_data["APL"]} )'
-            message += f'\nADPM：{game_data["ADPM"]} ( x{game_data["ADPL"]} ) ( {game_data["VS"]}vs )'
-    message += f'\n40L: {pb_data["Sprint"]}s' if 'Sprint' in pb_data else ''
-    message += f'\nMarathon: {pb_data["Marathon"]}' if 'Marathon' in pb_data else ''
-    message += f'\nChallenge: {pb_data["Challenge"]}' if 'Challenge' in pb_data else ''
-    return message

nonebot_plugin_tetris_stats/games/__init__.py

@@ -0,0 +1,55 @@
+from collections.abc import Callable
+
+from nonebot.adapters import Bot
+from nonebot.matcher import Matcher
+from nonebot.message import run_postprocessor
+from nonebot.typing import T_Handler
+from nonebot_plugin_alconna import AlcMatches, Alconna, At, CommandMeta, on_alconna
+
+from .. import ns
+from ..utils.exception import MessageFormatError, NeedCatchError
+
+command: Alconna = Alconna(
+    ['tetris-stats', 'tstats'],
+    namespace=ns,
+    meta=CommandMeta(
+        description='俄罗斯方块相关游戏数据查询',
+        fuzzy_match=True,
+    ),
+)
+
+alc = on_alconna(
+    command=command,
+    skip_for_unmatch=False,
+    auto_send_output=True,
+    use_origin=True,
+)
+
+
+def add_block_handlers(handler: Callable[[T_Handler], T_Handler]) -> None:
+    @handler
+    async def _(bot: Bot, matcher: Matcher, target: At):
+        if isinstance(target, At) and target.target == bot.self_id:
+            await matcher.finish('不能查询bot的信息')
+
+
+from . import tetrio, top, tos  # noqa: F401, E402
+
+
+@alc.handle()
+async def _(matcher: Matcher, account: MessageFormatError):
+    await matcher.finish(str(account))
+
+
+@alc.handle()
+async def _(matcher: Matcher, matches: AlcMatches):
+    if matches.head_matched and matches.options != {} or matches.main_args == {}:
+        await matcher.finish(
+            (f'{matches.error_info!r}\n' if matches.error_info is not None else '')
+            + f'输入"{matches.header_result} --help"查看帮助'
+        )
+
+
+@run_postprocessor
+async def _(matcher: Matcher, exception: NeedCatchError):
+    await matcher.send(str(exception))

nonebot_plugin_tetris_stats/games/constant.py

@@ -0,0 +1,3 @@
+BIND_COMMAND: list[str] = ['绑定', 'bind']
+QUERY_COMMAND: list[str] = ['查', '查询', 'query', 'stats']
+CANT_VERIFY_MESSAGE = '* 由于无法验证绑定信息, 不能保证查询到的用户为本人\n'

nonebot_plugin_tetris_stats/games/schemas.py

@@ -0,0 +1,23 @@
+from abc import ABC, abstractmethod
+
+from pydantic import BaseModel
+
+from ..utils.typing import GameType
+
+
+class Base(BaseModel):
+    platform: GameType
+
+
+class BaseUser(ABC, Base):
+    """游戏用户"""
+
+    def __eq__(self, __value: object) -> bool:
+        if isinstance(__value, BaseUser):
+            return self.unique_identifier == __value.unique_identifier
+        return False
+
+    @property
+    @abstractmethod
+    def unique_identifier(self) -> str:
+        raise NotImplementedError

nonebot_plugin_tetris_stats/games/tetrio/__init__.py

@@ -0,0 +1,166 @@
+from arclet.alconna import Arg, ArgFlag, Args, Option, Subcommand
+from nonebot_plugin_alconna import At
+
+from ...utils.exception import MessageFormatError
+from ...utils.typing import Me
+from .. import add_block_handlers, alc, command
+from .api import Player
+from .api.typing import ValidRank
+from .constant import USER_ID, USER_NAME
+from .typing import Template
+
+
+def get_player(user_id_or_name: str) -> Player | MessageFormatError:
+    if USER_ID.match(user_id_or_name):
+        return Player(user_id=user_id_or_name, trust=True)
+    if USER_NAME.match(user_id_or_name):
+        return Player(user_name=user_id_or_name, trust=True)
+    return MessageFormatError('用户名/ID不合法')
+
+
+command.add(
+    Subcommand(
+        'TETR.IO',
+        Subcommand(
+            'bind',
+            Args(
+                Arg(
+                    'account',
+                    get_player,
+                    notice='TETR.IO 用户名 / ID',
+                    flags=[ArgFlag.HIDDEN],
+                )
+            ),
+            help_text='绑定 TETR.IO 账号',
+        ),
+        Subcommand(
+            'query',
+            Args(
+                Arg(
+                    'target',
+                    At | Me,
+                    notice='@想要查询的人 / 自己',
+                    flags=[ArgFlag.HIDDEN, ArgFlag.OPTIONAL],
+                ),
+                Arg(
+                    'account',
+                    get_player,
+                    notice='TETR.IO 用户名 / ID',
+                    flags=[ArgFlag.HIDDEN, ArgFlag.OPTIONAL],
+                ),
+            ),
+            Option(
+                '--template',
+                Arg('template', Template),
+                alias=['-T'],
+                help_text='要使用的查询模板',
+            ),
+            help_text='查询 TETR.IO 游戏信息',
+        ),
+        Subcommand(
+            'record',
+            Option(
+                '--40l',
+                dest='sprint',
+            ),
+            Option(
+                '--blitz',
+                dest='blitz',
+            ),
+            Args(
+                Arg(
+                    'target',
+                    At | Me,
+                    notice='@想要查询的人 / 自己',
+                    flags=[ArgFlag.HIDDEN, ArgFlag.OPTIONAL],
+                ),
+                Arg(
+                    'account',
+                    get_player,
+                    notice='TETR.IO 用户名 / ID',
+                    flags=[ArgFlag.HIDDEN, ArgFlag.OPTIONAL],
+                ),
+            ),
+        ),
+        Subcommand(
+            'rank',
+            Args(Arg('rank', ValidRank, notice='TETR.IO 段位')),
+            help_text='查询 TETR.IO 段位信息',
+        ),
+        Subcommand(
+            'config',
+            Option(
+                '--default-template',
+                Arg('template', Template),
+                alias=['-DT', 'DefaultTemplate'],
+            ),
+        ),
+        dest='TETRIO',
+        help_text='TETR.IO 游戏相关指令',
+    )
+)
+
+alc.shortcut(
+    '(?i:io)(?i:绑定|绑|bind)',
+    {
+        'command': 'tstats TETR.IO bind',
+        'humanized': 'io绑定',
+    },
+)
+alc.shortcut(
+    '(?i:io)(?i:查询|查|query|stats)',
+    {
+        'command': 'tstats TETR.IO query',
+        'humanized': 'io查',
+    },
+)
+alc.shortcut(
+    '(?i:io)(?i:记录|record)(?i:40l)',
+    {
+        'command': 'tstats TETR.IO record --40l',
+        'humanized': 'io记录40l',
+    },
+)
+alc.shortcut(
+    '(?i:io)(?i:记录|record)(?i:blitz)',
+    {
+        'command': 'tstats TETR.IO record --blitz',
+        'humanized': 'io记录blitz',
+    },
+)
+alc.shortcut(
+    '(?i:io)(?i:段位|段|rank)',
+    {
+        'command': 'tstats TETR.IO rank',
+        'humanized': 'iorank',
+    },
+)
+alc.shortcut(
+    '(?i:io)(?i:配置|配|config)',
+    {
+        'command': 'tstats TETR.IO config',
+        'humanized': 'io配置',
+    },
+)
+
+alc.shortcut(
+    'fkosk',
+    {
+        'command': 'tstats TETR.IO query',
+        'args': ['我'],
+        'fuzzy': False,
+        'humanized': 'An Easter egg!',
+    },
+)
+
+add_block_handlers(alc.assign('TETRIO.query'))
+
+from . import bind, config, query, rank, record  # noqa: E402
+
+__all__ = [
+    'bind',
+    'config',
+    'query',
+    'rank',
+    'record',
+]

nonebot_plugin_tetris_stats/games/tetrio/api/__init__.py

@@ -0,0 +1,7 @@
+from .player import Player
+from .schemas.user import User
+from .schemas.user_info import UserInfoSuccess
+from .schemas.user_records import UserRecordsSuccess
+from .tetra_league import full_export as tetra_league_full_export
+
+__all__ = ['Player', 'User', 'UserInfoSuccess', 'UserRecordsSuccess', 'tetra_league_full_export']

nonebot_plugin_tetris_stats/games/tetrio/api/cache.py

@@ -0,0 +1,35 @@
+from asyncio import Lock
+from datetime import datetime, timezone
+from typing import ClassVar
+from weakref import WeakValueDictionary
+
+from aiocache import Cache as ACache  # type: ignore[import-untyped]
+from nonebot.compat import type_validate_json
+from nonebot.log import logger
+
+from ....utils.request import Request
+from .schemas.base import FailedModel, SuccessModel
+
+UTC = timezone.utc
+
+
+class Cache:
+    cache = ACache(ACache.MEMORY)
+    task: ClassVar[WeakValueDictionary[str, Lock]] = WeakValueDictionary()
+
+    @classmethod
+    async def get(cls, url: str) -> bytes:
+        lock = cls.task.setdefault(url, Lock())
+        async with lock:
+            if (cached_data := await cls.cache.get(url)) is not None:
+                logger.debug(f'{url}: Cache hit!')
+                return cached_data
+            response_data = await Request.request(url)
+            parsed_data: SuccessModel | FailedModel = type_validate_json(SuccessModel | FailedModel, response_data)  # type: ignore[arg-type]
+            if isinstance(parsed_data, SuccessModel):
+                await cls.cache.add(
+                    url,
+                    response_data,
+                    (parsed_data.cache.cached_until - datetime.now(UTC)).total_seconds(),
+                )
+            return response_data

nonebot_plugin_tetris_stats/games/tetrio/api/models.py

@@ -0,0 +1,17 @@
+from datetime import datetime
+from typing import Literal
+
+from nonebot_plugin_orm import Model
+from sqlalchemy import DateTime, String
+from sqlalchemy.orm import Mapped, MappedAsDataclass, mapped_column
+
+from ....db.models import PydanticType
+from .schemas.base import SuccessModel
+
+
+class TETRIOHistoricalData(MappedAsDataclass, Model):
+    id: Mapped[int] = mapped_column(init=False, primary_key=True)
+    user_unique_identifier: Mapped[str] = mapped_column(String(24), index=True)
+    api_type: Mapped[Literal['User Info', 'User Records']] = mapped_column(String(16), index=True)
+    data: Mapped[SuccessModel] = mapped_column(PydanticType(get_model=[SuccessModel.__subclasses__], models=set()))
+    update_time: Mapped[datetime] = mapped_column(DateTime, index=True)

nonebot_plugin_tetris_stats/games/tetrio/api/player.py

@@ -0,0 +1,114 @@
+from typing import overload
+
+from nonebot.compat import type_validate_json
+
+from ....db import anti_duplicate_add
+from ....utils.exception import RequestError
+from ....utils.request import splice_url
+from ..constant import BASE_URL, USER_ID, USER_NAME
+from .cache import Cache
+from .models import TETRIOHistoricalData
+from .schemas.base import FailedModel
+from .schemas.user import User
+from .schemas.user_info import UserInfo, UserInfoSuccess
+from .schemas.user_records import SoloModeRecord, UserRecords, UserRecordsSuccess, Zen
+
+
+class Player:
+    @overload
+    def __init__(self, *, user_id: str, trust: bool = False): ...
+    @overload
+    def __init__(self, *, user_name: str, trust: bool = False): ...
+    def __init__(self, *, user_id: str | None = None, user_name: str | None = None, trust: bool = False):
+        self.user_id = user_id
+        self.user_name = user_name
+        if not trust:
+            if self.user_id is not None:
+                if not USER_ID.match(self.user_id):
+                    msg = 'Invalid user id'
+                    raise ValueError(msg)
+            elif self.user_name is not None:
+                if not USER_NAME.match(self.user_name):
+                    msg = 'Invalid user name'
+                    raise ValueError(msg)
+            else:
+                msg = 'Invalid user'
+                raise ValueError(msg)
+        self.__user: User | None = None
+        self._user_info: UserInfoSuccess | None = None
+        self._user_records: UserRecordsSuccess | None = None
+
+    @property
+    def _request_user_parameter(self) -> str:
+        if self.user_id is not None:
+            return self.user_id
+        if self.user_name is not None:
+            return self.user_name.lower()
+        msg = 'Invalid user'
+        raise ValueError(msg)
+
+    @property
+    async def user(self) -> User:
+        if self.__user is None:
+            user_info = await self.get_info()
+            self.__user = User(
+                ID=user_info.data.user.id,
+                name=user_info.data.user.username,
+            )
+            self.user_id = user_info.data.user.id
+            self.user_name = user_info.data.user.username
+        return self.__user
+
+    async def get_info(self) -> UserInfoSuccess:
+        """Get User Info"""
+        if self._user_info is None:
+            raw_user_info = await Cache.get(splice_url([BASE_URL, 'users/', f'{self._request_user_parameter}']))
+            user_info: UserInfo = type_validate_json(UserInfo, raw_user_info)  # type: ignore[arg-type]
+            if isinstance(user_info, FailedModel):
+                msg = f'用户信息请求错误:\n{user_info.error}'
+                raise RequestError(msg)
+            self._user_info = user_info
+            await anti_duplicate_add(
+                TETRIOHistoricalData,
+                TETRIOHistoricalData(
+                    user_unique_identifier=(await self.user).unique_identifier,
+                    api_type='User Info',
+                    data=user_info,
+                    update_time=user_info.cache.cached_at,
+                ),
+            )
+        return self._user_info
+
+    async def get_records(self) -> UserRecordsSuccess:
+        """Get User Records"""
+        if self._user_records is None:
+            raw_user_records = await Cache.get(
+                splice_url([BASE_URL, 'users/', f'{self._request_user_parameter}/', 'records'])
+            )
+            user_records: UserRecords = type_validate_json(UserRecords, raw_user_records)  # type: ignore[arg-type]
+            if isinstance(user_records, FailedModel):
+                msg = f'用户Solo数据请求错误:\n{user_records.error}'
+                raise RequestError(msg)
+            self._user_records = user_records
+            await anti_duplicate_add(
+                TETRIOHistoricalData,
+                TETRIOHistoricalData(
+                    user_unique_identifier=(await self.user).unique_identifier,
+                    api_type='User Records',
+                    data=user_records,
+                    update_time=user_records.cache.cached_at,
+                ),
+            )
+        return self._user_records
+
+    @property
+    async def sprint(self) -> SoloModeRecord:
+        return (await self.get_records()).data.records.sprint
+
+    @property
+    async def blitz(self) -> SoloModeRecord:
+        return (await self.get_records()).data.records.blitz
+
+    @property
+    async def zen(self) -> Zen:
+        return (await self.get_records()).data.zen

nonebot_plugin_tetris_stats/games/tetrio/api/schemas/base.py

@@ -0,0 +1,20 @@
+from datetime import datetime
+from typing import Literal
+
+from pydantic import BaseModel
+
+
+class Cache(BaseModel):
+    status: str
+    cached_at: datetime
+    cached_until: datetime
+
+
+class SuccessModel(BaseModel):
+    success: Literal[True]
+    cache: Cache
+
+
+class FailedModel(BaseModel):
+    success: Literal[False]
+    error: str

nonebot_plugin_tetris_stats/games/tetrio/api/schemas/tetra_league.py

@@ -0,0 +1,59 @@
+from pydantic import BaseModel, Field
+
+from ..typing import Rank
+from .base import FailedModel
+from .base import SuccessModel as BaseSuccessModel
+
+
+class _User(BaseModel):
+    id: str = Field(..., alias='_id')
+    username: str
+    role: str
+    xp: float
+    supporter: bool
+    verified: bool
+    country: str | None = None
+
+
+class _League(BaseModel):
+    gamesplayed: int
+    gameswon: int
+    rating: float
+    rank: Rank
+    bestrank: Rank
+    decaying: bool
+
+
+class ValidLeague(_League):
+    glicko: float
+    rd: float
+    apm: float
+    pps: float
+    vs: float
+
+
+class ValidUser(_User):
+    league: ValidLeague
+
+
+class InvalidLeague(_League):
+    glicko: float | None = None
+    rd: float | None = None
+    apm: float | None = None
+    pps: float | None = None
+    vs: float | None = None
+
+
+class InvalidUser(_User):
+    league: InvalidLeague
+
+
+class Data(BaseModel):
+    users: list[ValidUser | InvalidUser]
+
+
+class TetraLeagueSuccess(BaseSuccessModel):
+    data: Data
+
+
+TetraLeague = TetraLeagueSuccess | FailedModel

nonebot_plugin_tetris_stats/games/tetrio/api/schemas/user.py

@@ -0,0 +1,18 @@
+from typing import Literal
+
+from typing_extensions import override
+
+from ....schemas import BaseUser
+from ...constant import GAME_TYPE
+
+
+class User(BaseUser):
+    platform: Literal['IO'] = GAME_TYPE
+
+    ID: str
+    name: str
+
+    @property
+    @override
+    def unique_identifier(self) -> str:
+        return self.ID

nonebot_plugin_tetris_stats/games/tetrio/api/schemas/user_info.py

@@ -0,0 +1,133 @@
+from datetime import datetime
+from typing import Literal
+
+from pydantic import BaseModel, Field
+
+from ..typing import Rank
+from .base import FailedModel
+from .base import SuccessModel as BaseSuccessModel
+
+
+class Badge(BaseModel):
+    id: str
+    label: str
+    group: str | None = None
+    ts: datetime | Literal[False] | None = None
+
+
+class MetaLeague(BaseModel):
+    decaying: bool
+
+
+class NeverPlayedLeague(MetaLeague):
+    gamesplayed: Literal[0]
+    gameswon: Literal[0]
+    rating: Literal[-1]
+    rank: Literal['z']
+    standing: Literal[-1]
+    standing_local: Literal[-1]
+    next_rank: None
+    prev_rank: None
+    next_at: Literal[-1]
+    prev_at: Literal[-1]
+    percentile: Literal[-1]
+    percentile_rank: Literal['z']
+    apm: None = None
+    pps: None = None
+    vs: None = None
+
+
+class NeverRatedLeague(MetaLeague):
+    gamesplayed: Literal[1, 2, 3, 4, 5, 6, 7, 8, 9]
+    gameswon: int
+    rating: Literal[-1]
+    rank: Literal['z']
+    standing: Literal[-1]
+    standing_local: Literal[-1]
+    next_rank: None
+    prev_rank: None
+    next_at: Literal[-1]
+    prev_at: Literal[-1]
+    percentile: Literal[-1]
+    percentile_rank: Literal['z']
+    apm: float
+    pps: float
+    vs: float | None = None
+
+
+class RatedLeague(MetaLeague):
+    gamesplayed: int
+    gameswon: int
+    rating: float
+    rank: Rank
+    bestrank: Rank
+    standing: int
+    standing_local: int
+    next_rank: Rank | None = None
+    prev_rank: Rank | None = None
+    next_at: int
+    prev_at: int
+    percentile: float
+    percentile_rank: str
+    glicko: float
+    rd: float
+    apm: float
+    pps: float
+    vs: float | None = None
+
+
+class Discord(BaseModel):
+    id: str
+    username: str
+
+
+class Connections(BaseModel):
+    discord: Discord | None = None
+
+
+class Distinguishment(BaseModel):
+    type: str
+
+
+class User(BaseModel):
+    id: str = Field(..., alias='_id')
+    username: str
+    role: Literal['anon', 'user', 'bot', 'halfmod', 'mod', 'admin', 'sysop', 'banned']
+    ts: datetime | None = None
+    botmaster: str | None = None
+    badges: list[Badge]
+    xp: float
+    gamesplayed: int
+    gameswon: int
+    gametime: float
+    country: str | None = None
+    badstanding: bool | None = None
+    supporter: bool | None = None  # osk说是必有, 但实际上不是 fkosk
+    supporter_tier: int
+    verified: bool
+    league: NeverPlayedLeague | NeverRatedLeague | RatedLeague
+    avatar_revision: int | None = None
+    """This user's avatar ID. Get their avatar at
+
+    https://tetr.io/user-content/avatars/{ USERID }.jpg?rv={ AVATAR_REVISION }"""
+    banner_revision: int | None = None
+    """This user's banner ID. Get their banner at
+
+    https://tetr.io/user-content/banners/{ USERID }.jpg?rv={ BANNER_REVISION }
+
+    Ignore this field if the user is not a supporter."""
+    bio: str | None = None
+    connections: Connections
+    friend_count: int | None = None
+    distinguishment: Distinguishment | None = None
+
+
+class Data(BaseModel):
+    user: User
+
+
+class UserInfoSuccess(BaseSuccessModel):
+    data: Data
+
+
+UserInfo = UserInfoSuccess | FailedModel

nonebot_plugin_tetris_stats/games/tetrio/api/schemas/user_records.py

@@ -0,0 +1,122 @@
+from datetime import datetime
+
+from pydantic import BaseModel, Field
+
+from .....utils.typing import Number
+from .base import FailedModel
+from .base import SuccessModel as BaseSuccessModel
+
+
+class Time(BaseModel):
+    start: int
+    zero: bool
+    locked: bool
+    prev: int
+    frameoffset: int | None = None
+
+
+class Clears(BaseModel):
+    singles: int
+    doubles: int
+    triples: int
+    quads: int
+    pentas: int | None = None
+    realtspins: int
+    minitspins: int
+    minitspinsingles: int
+    tspinsingles: int
+    minitspindoubles: int
+    tspindoubles: int
+    tspintriples: int
+    tspinquads: int
+    allclear: int
+
+
+class Garbage(BaseModel):
+    sent: int
+    received: int
+    attack: int | None = None
+    cleared: int | None = None
+
+
+class Finesse(BaseModel):
+    combo: int
+    faults: int
+    perfectpieces: int
+
+
+class EndContext(BaseModel):
+    seed: Number
+    lines: int
+    level_lines: int
+    level_lines_needed: int
+    inputs: int
+    holds: int | None = None
+    time: Time
+    score: int
+    zenlevel: int | None = None
+    zenprogress: int | None = None
+    level: int
+    combo: int
+    currentcombopower: int | None = None  # WTF
+    topcombo: int
+    btb: int
+    topbtb: int
+    currentbtbchainpower: int | None = None  # WTF * 2
+    tspins: int
+    piecesplaced: int
+    clears: Clears
+    garbage: Garbage
+    kills: int
+    finesse: Finesse
+    final_time: float = Field(..., alias='finalTime')
+    gametype: str
+
+
+class _User(BaseModel):
+    id: str = Field(..., alias='_id')
+    username: str
+
+
+class _Record(BaseModel):
+    id: str = Field(..., alias='_id')
+    stream: str
+    replayid: str
+    user: _User
+    ts: datetime
+    ismulti: bool | None = None
+
+
+class SoloRecord(_Record):
+    endcontext: EndContext
+
+
+class MultiRecord(_Record):
+    endcontext: list[EndContext]
+
+
+class SoloModeRecord(BaseModel):
+    record: SoloRecord | None = None
+    rank: int | None = None
+
+
+class Records(BaseModel):
+    sprint: SoloModeRecord = Field(..., alias='40l')
+    blitz: SoloModeRecord
+
+
+class Zen(BaseModel):
+    level: int
+    score: int
+
+
+class Data(BaseModel):
+    records: Records
+    zen: Zen
+
+
+class UserRecordsSuccess(BaseSuccessModel):
+    data: Data
+
+
+UserRecords = UserRecordsSuccess | FailedModel

nonebot_plugin_tetris_stats/games/tetrio/api/tetra_league.py

@@ -0,0 +1,36 @@
+from typing import Literal, NamedTuple, overload
+
+from nonebot.compat import type_validate_json
+
+from ....utils.exception import RequestError
+from ....utils.request import splice_url
+from ..constant import BASE_URL
+from .cache import Cache
+from .schemas.base import FailedModel
+from .schemas.tetra_league import TetraLeague, TetraLeagueSuccess
+
+
+class FullExport(NamedTuple):
+    model: TetraLeagueSuccess
+    original: bytes
+
+
+@overload
+async def full_export(*, with_original: Literal[False]) -> TetraLeagueSuccess: ...
+
+
+@overload
+async def full_export(*, with_original: Literal[True]) -> FullExport: ...
+
+
+async def full_export(*, with_original: bool) -> TetraLeagueSuccess | FullExport:
+    full: TetraLeague = type_validate_json(
+        TetraLeague,  # type: ignore[arg-type]
+        (data := await Cache.get(splice_url([BASE_URL, 'users/lists/league/all']))),
+    )
+    if isinstance(full, FailedModel):
+        msg = f'排行榜数据请求错误:\n{full.error}'
+        raise RequestError(msg)
+    if with_original:
+        return FullExport(full, data)
+    return full

nonebot_plugin_tetris_stats/games/tetrio/api/typing.py

@@ -0,0 +1,23 @@
+from typing import Literal
+
+ValidRank = Literal[
+    'x',
+    'u',
+    'ss',
+    's+',
+    's',
+    's-',
+    'a+',
+    'a',
+    'a-',
+    'b+',
+    'b',
+    'b-',
+    'c+',
+    'c',
+    'c-',
+    'd+',
+    'd',
+]
+
+Rank = ValidRank | Literal['z']  # 未定级

nonebot_plugin_tetris_stats/games/tetrio/bind.py

@@ -0,0 +1,62 @@
+from asyncio import gather
+from hashlib import md5
+from urllib.parse import urlencode
+
+from nonebot_plugin_alconna.uniseg import UniMessage
+from nonebot_plugin_orm import get_session
+from nonebot_plugin_session import EventSession  # type: ignore[import-untyped]
+from nonebot_plugin_session_orm import get_session_persist_id  # type: ignore[import-untyped]
+from nonebot_plugin_user import User  # type: ignore[import-untyped]
+from nonebot_plugin_userinfo import BotUserInfo, UserInfo  # type: ignore[import-untyped]
+
+from ...db import BindStatus, create_or_update_bind, trigger
+from ...utils.host import HostPage, get_self_netloc
+from ...utils.image import get_avatar
+from ...utils.render import Bind, render
+from ...utils.render.schemas.base import Avatar, People
+from ...utils.screenshot import screenshot
+from . import alc
+from .api import Player
+from .constant import GAME_TYPE
+
+
+@alc.assign('TETRIO.bind')
+async def _(nb_user: User, account: Player, event_session: EventSession, bot_info: UserInfo = BotUserInfo()):  # noqa: B008
+    async with trigger(
+        session_persist_id=await get_session_persist_id(event_session),
+        game_platform=GAME_TYPE,
+        command_type='bind',
+        command_args=[],
+    ):
+        user, user_info = await gather(account.user, account.get_info())
+        async with get_session() as session:
+            bind_status = await create_or_update_bind(
+                session=session,
+                user=nb_user,
+                game_platform=GAME_TYPE,
+                game_account=user.unique_identifier,
+            )
+        if bind_status in (BindStatus.SUCCESS, BindStatus.UPDATE):
+            netloc = get_self_netloc()
+            async with HostPage(
+                await render(
+                    'v1/binding',
+                    Bind(
+                        platform='TETR.IO',
+                        status='unknown',
+                        user=People(
+                            avatar=f'http://{netloc}/host/resource/tetrio/avatars/{user.ID}?{urlencode({"revision": user_info.data.user.avatar_revision})}'
+                            if user_info.data.user.avatar_revision is not None
+                            and user_info.data.user.avatar_revision != 0
+                            else Avatar(type='identicon', hash=md5(user_info.data.user.id.encode()).hexdigest()),  # noqa: S324
+                            name=user_info.data.user.username.upper(),
+                        ),
+                        bot=People(
+                            avatar=await get_avatar(bot_info, 'Data URI', '../../static/logo/logo.svg'),
+                            name=bot_info.user_name,
+                        ),
+                        command='io查我',
+                    ),
+                )
+            ) as page_hash:
+                await UniMessage.image(raw=await screenshot(f'http://{netloc}/host/{page_hash}.html')).finish()

nonebot_plugin_tetris_stats/games/tetrio/config.py

@@ -0,0 +1,20 @@
+from nonebot_plugin_alconna.uniseg import UniMessage
+from nonebot_plugin_orm import async_scoped_session
+from nonebot_plugin_user import User  # type: ignore[import-untyped]
+from sqlalchemy import select
+
+from . import alc
+from .models import TETRIOUserConfig
+from .typing import Template
+
+
+@alc.assign('TETRIO.config')
+async def _(user: User, session: async_scoped_session, template: Template):
+    config = (await session.scalars(select(TETRIOUserConfig).where(TETRIOUserConfig.id == user.id))).one_or_none()
+    if config is None:
+        config = TETRIOUserConfig(id=user.id, query_template=template)
+        session.add(config)
+    else:
+        config.query_template = template
+    await session.commit()
+    await UniMessage('配置成功').finish()

nonebot_plugin_tetris_stats/games/tetrio/constant.py

@@ -0,0 +1,34 @@
+from re import compile
+from typing import Literal
+
+from .api.typing import Rank
+
+GAME_TYPE: Literal['IO'] = 'IO'
+
+BASE_URL = 'https://ch.tetr.io/api/'
+
+RANK_PERCENTILE: dict[Rank, float] = {
+    'x': 1,
+    'u': 5,
+    'ss': 11,
+    's+': 17,
+    's': 23,
+    's-': 30,
+    'a+': 38,
+    'a': 46,
+    'a-': 54,
+    'b+': 62,
+    'b': 70,
+    'b-': 78,
+    'c+': 84,
+    'c': 90,
+    'c-': 95,
+    'd+': 97.5,
+    'd': 100,
+}
+
+TR_MIN = 0
+TR_MAX = 25000
+
+USER_ID = compile(r'^[a-f0-9]{24}$')
+USER_NAME = compile(r'^[a-zA-Z0-9_-]{3,16}$')

nonebot_plugin_tetris_stats/games/tetrio/models.py

@@ -0,0 +1,34 @@
+from datetime import datetime
+
+from nonebot_plugin_orm import Model
+from sqlalchemy import JSON, DateTime, String
+from sqlalchemy.orm import Mapped, MappedAsDataclass, mapped_column
+
+from .api.typing import Rank
+from .typing import Template
+
+
+class IORank(MappedAsDataclass, Model):
+    id: Mapped[int] = mapped_column(init=False, primary_key=True)
+    rank: Mapped[Rank] = mapped_column(String(2), index=True)
+    tr_line: Mapped[float]
+    player_count: Mapped[int]
+    low_pps: Mapped[tuple[dict[str, str], float]] = mapped_column(JSON)
+    low_apm: Mapped[tuple[dict[str, str], float]] = mapped_column(JSON)
+    low_vs: Mapped[tuple[dict[str, str], float]] = mapped_column(JSON)
+    avg_pps: Mapped[float]
+    avg_apm: Mapped[float]
+    avg_vs: Mapped[float]
+    high_pps: Mapped[tuple[dict[str, str], float]] = mapped_column(JSON)
+    high_apm: Mapped[tuple[dict[str, str], float]] = mapped_column(JSON)
+    high_vs: Mapped[tuple[dict[str, str], float]] = mapped_column(JSON)
+    update_time: Mapped[datetime] = mapped_column(
+        DateTime,
+        index=True,
+    )
+    file_hash: Mapped[str | None] = mapped_column(String(128), index=True)
+
+
+class TETRIOUserConfig(MappedAsDataclass, Model):
+    id: Mapped[int] = mapped_column(primary_key=True)
+    query_template: Mapped[Template] = mapped_column(String(2))

nonebot_plugin_tetris_stats/games/tetrio/query.py

@@ -0,0 +1,553 @@
+from asyncio import gather
+from collections import defaultdict
+from datetime import date, datetime, timedelta, timezone
+from hashlib import md5
+from math import ceil, floor
+from typing import ClassVar, TypeVar, overload
+from urllib.parse import urlencode
+from zoneinfo import ZoneInfo
+
+from aiofiles import open
+from nonebot import get_driver
+from nonebot.adapters import Event
+from nonebot.compat import type_validate_json
+from nonebot.matcher import Matcher
+from nonebot_plugin_alconna import At
+from nonebot_plugin_alconna.uniseg import UniMessage
+from nonebot_plugin_apscheduler import scheduler  # type: ignore[import-untyped]
+from nonebot_plugin_localstore import get_data_file  # type: ignore[import-untyped]
+from nonebot_plugin_orm import get_session
+from nonebot_plugin_session import EventSession  # type: ignore[import-untyped]
+from nonebot_plugin_session_orm import get_session_persist_id  # type: ignore[import-untyped]
+from nonebot_plugin_user import User as NBUser  # type: ignore[import-untyped]
+from nonebot_plugin_user import get_user  # type: ignore[import-untyped]
+from sqlalchemy import select
+from zstandard import ZstdDecompressor
+
+from ...db import query_bind_info, trigger
+from ...utils.exception import FallbackError
+from ...utils.host import HostPage, get_self_netloc
+from ...utils.metrics import TetrisMetricsProWithPPSVS, get_metrics
+from ...utils.render import render
+from ...utils.render.schemas.base import Avatar, Ranking
+from ...utils.render.schemas.tetrio.tetrio_info import Info as V1TemplateInfo
+from ...utils.render.schemas.tetrio.tetrio_info import Radar, TetraLeague, TetraLeagueHistory, TetraLeagueHistoryData
+from ...utils.render.schemas.tetrio.tetrio_info import User as V1TemplateUser
+from ...utils.render.schemas.tetrio.tetrio_user_info_v2 import Badge, Blitz, Sprint, Statistic, TetraLeagueStatistic
+from ...utils.render.schemas.tetrio.tetrio_user_info_v2 import Info as V2TemplateInfo
+from ...utils.render.schemas.tetrio.tetrio_user_info_v2 import TetraLeague as V2TemplateTetraLeague
+from ...utils.render.schemas.tetrio.tetrio_user_info_v2 import User as V2TemplateUser
+from ...utils.screenshot import screenshot
+from ...utils.typing import Me, Number
+from ..constant import CANT_VERIFY_MESSAGE
+from . import alc
+from .api import Player, User, UserInfoSuccess
+from .api.models import TETRIOHistoricalData
+from .api.schemas.tetra_league import TetraLeagueSuccess
+from .api.schemas.user_info import NeverPlayedLeague, NeverRatedLeague, RatedLeague
+from .constant import GAME_TYPE, TR_MAX, TR_MIN
+from .models import IORank, TETRIOUserConfig
+from .typing import Template
+
+UTC = timezone.utc
+
+driver = get_driver()
+
+
+@alc.assign('TETRIO.query')
+async def _(  # noqa: PLR0913
+    user: NBUser,
+    event: Event,
+    matcher: Matcher,
+    target: At | Me,
+    event_session: EventSession,
+    template: Template | None = None,
+):
+    async with trigger(
+        session_persist_id=await get_session_persist_id(event_session),
+        game_platform=GAME_TYPE,
+        command_type='query',
+        command_args=[],
+    ):
+        async with get_session() as session:
+            bind = await query_bind_info(
+                session=session,
+                user=await get_user(
+                    event_session.platform, target.target if isinstance(target, At) else event.get_user_id()
+                ),
+                game_platform=GAME_TYPE,
+            )
+            if template is None:
+                template = await session.scalar(
+                    select(TETRIOUserConfig.query_template).where(TETRIOUserConfig.id == user.id)
+                )
+        if bind is None:
+            await matcher.finish('未查询到绑定信息')
+        message = UniMessage(CANT_VERIFY_MESSAGE)
+        player = Player(user_id=bind.game_account, trust=True)
+        await (message + (await make_query_result(player, template or 'v1'))).finish()
+
+
+@alc.assign('TETRIO.query')
+async def _(user: NBUser, account: Player, event_session: EventSession, template: Template | None = None):
+    async with trigger(
+        session_persist_id=await get_session_persist_id(event_session),
+        game_platform=GAME_TYPE,
+        command_type='query',
+        command_args=[],
+    ):
+        async with get_session() as session:
+            if template is None:
+                template = await session.scalar(
+                    select(TETRIOUserConfig.query_template).where(TETRIOUserConfig.id == user.id)
+                )
+        await (await make_query_result(account, template or 'v1')).finish()
+
+
+def get_value_bounds(values: list[int | float]) -> tuple[int, int]:
+    value_max = 10 * ceil(max(values) / 10)
+    value_min = 10 * floor(min(values) / 10)
+    return value_max, value_min
+
+
+def get_split(value_max: int, value_min: int) -> tuple[int, int]:
+    offset = 0
+    overflow = 0
+
+    while True:
+        if (new_max_value := value_max + offset + overflow) > TR_MAX:
+            overflow -= 1
+            continue
+        if (new_min_value := value_min - offset + overflow) < TR_MIN:
+            overflow += 1
+            continue
+        if ((new_max_value - new_min_value) / 40).is_integer():
+            split_value = int((value_max + offset - (value_min - offset)) / 4)
+            break
+        offset += 1
+    return split_value, offset + overflow
+
+
+def get_specified_point(
+    previous_point: TetraLeagueHistoryData,
+    behind_point: TetraLeagueHistoryData,
+    point_time: datetime,
+) -> TetraLeagueHistoryData:
+    """根据给出的 previous_point 和 behind_point, 推算 point_time 点处的数据
+
+    Args:
+        previous_point (Data): 前面的数据点
+        behind_point (Data): 后面的数据点
+        point_time (datetime): 要推算的点的位置
+
+    Returns:
+        Data: 要推算的点的数据
+    """
+    # 求两个点的斜率
+    slope = (behind_point.tr - previous_point.tr) / (
+        datetime.timestamp(behind_point.record_at) - datetime.timestamp(previous_point.record_at)
+    )
+    return TetraLeagueHistoryData(
+        record_at=point_time,
+        tr=previous_point.tr + slope * (datetime.timestamp(point_time) - datetime.timestamp(previous_point.record_at)),
+    )
+
+
+async def query_historical_data(user: User, user_info: UserInfoSuccess) -> list[TetraLeagueHistoryData]:
+    today = datetime.now(ZoneInfo('Asia/Shanghai')).replace(hour=0, minute=0, second=0, microsecond=0)
+    forward = timedelta(days=9)
+    start_time = (today - forward).astimezone(UTC)
+    async with get_session() as session:
+        historical_data = (
+            await session.scalars(
+                select(TETRIOHistoricalData)
+                .where(TETRIOHistoricalData.update_time >= start_time)
+                .where(TETRIOHistoricalData.user_unique_identifier == user.unique_identifier)
+                .where(TETRIOHistoricalData.api_type == 'User Info')
+            )
+        ).all()
+        if historical_data:
+            extra = (
+                await session.scalars(
+                    select(TETRIOHistoricalData)
+                    .where(TETRIOHistoricalData.user_unique_identifier == user.unique_identifier)
+                    .where(TETRIOHistoricalData.api_type == 'User Info')
+                    .order_by(TETRIOHistoricalData.id.desc())
+                    .where(TETRIOHistoricalData.id < min([i.id for i in historical_data]))
+                    .limit(1)
+                )
+            ).one_or_none()
+            if extra is not None:
+                historical_data = list(historical_data)
+                historical_data.append(extra)
+    full_export_data = FullExport.get_data(user.unique_identifier)
+    if not historical_data and not full_export_data:
+        return [
+            TetraLeagueHistoryData(record_at=today - forward, tr=user_info.data.user.league.rating),
+            TetraLeagueHistoryData(record_at=today.replace(microsecond=1000), tr=user_info.data.user.league.rating),
+        ]
+    histories = [
+        TetraLeagueHistoryData(
+            record_at=i.update_time.astimezone(ZoneInfo('Asia/Shanghai')),
+            tr=i.data.data.user.league.rating,
+        )
+        for i in historical_data
+        if isinstance(i.data, UserInfoSuccess) and isinstance(i.data.data.user.league, RatedLeague)
+    ] + full_export_data
+
+    # 按照时间排序
+    histories = sorted(histories, key=lambda x: x.record_at)
+    for index, value in enumerate(histories):
+        # 在历史记录里找有没有今天0点后的数据, 并且至少要有两个数据点
+        if value.record_at > today and len(histories) >= 2:  # noqa: PLR2004
+            histories = histories[:index] + [
+                get_specified_point(histories[index - 1], histories[index], today.replace(microsecond=1000))
+            ]
+            break
+    else:
+        histories.append(
+            get_specified_point(
+                histories[-1],
+                TetraLeagueHistoryData(record_at=user_info.cache.cached_at, tr=user_info.data.user.league.rating),
+                today.replace(microsecond=1000),
+            )
+        )
+    if histories[0].record_at < (today - forward):
+        histories[0] = get_specified_point(
+            histories[0],
+            histories[1],
+            today - forward,
+        )
+    else:
+        histories.insert(0, TetraLeagueHistoryData(record_at=today - forward, tr=histories[0].tr))
+    return histories
+
+
+L = TypeVar('L', NeverPlayedLeague, NeverRatedLeague, RatedLeague)
+
+
+@overload
+def get_league(user_info: UserInfoSuccess, league_type: type[L]) -> L: ...
+@overload
+def get_league(
+    user_info: UserInfoSuccess, league_type: None = None
+) -> NeverPlayedLeague | NeverRatedLeague | RatedLeague: ...
+def get_league(
+    user_info: UserInfoSuccess, league_type: type[L] | None = None
+) -> L | NeverPlayedLeague | NeverRatedLeague | RatedLeague:
+    league = user_info.data.user.league
+    if league_type is None:
+        return league
+    if isinstance(league, league_type):
+        return league
+    raise FallbackError
+
+
+async def make_query_image_v1(player: Player) -> bytes:
+    user, user_info, sprint, blitz = await gather(player.user, player.get_info(), player.sprint, player.blitz)
+    league = get_league(user_info, RatedLeague)
+    if league.vs is None:
+        raise FallbackError
+    histories = await query_historical_data(user, user_info)
+    value_max, value_min = get_value_bounds([i.tr for i in histories])
+    split_value, offset = get_split(value_max, value_min)
+    if sprint.record is not None:
+        duration = timedelta(milliseconds=sprint.record.endcontext.final_time).total_seconds()
+        sprint_value = f'{duration:.3f}s' if duration < 60 else f'{duration // 60:.0f}m {duration % 60:.3f}s'  # noqa: PLR2004
+    else:
+        sprint_value = 'N/A'
+    blitz_value = f'{blitz.record.endcontext.score:,}' if blitz.record is not None else 'N/A'
+    netloc = get_self_netloc()
+    async with HostPage(
+        page=await render(
+            'v1/tetrio/info',
+            V1TemplateInfo(
+                user=V1TemplateUser(
+                    avatar=f'http://{netloc}/host/resource/tetrio/avatars/{user.ID}?{urlencode({"revision": user_info.data.user.avatar_revision})}'
+                    if user_info.data.user.avatar_revision is not None and user_info.data.user.avatar_revision != 0
+                    else Avatar(
+                        type='identicon',
+                        hash=md5(user_info.data.user.id.encode()).hexdigest(),  # noqa: S324
+                    ),
+                    name=user.name.upper(),
+                    bio=user_info.data.user.bio,
+                ),
+                ranking=Ranking(
+                    rating=round(league.glicko, 2),
+                    rd=round(league.rd, 2),
+                ),
+                tetra_league=TetraLeague(
+                    rank=league.rank,
+                    tr=round(league.rating, 2),
+                    global_rank=league.standing,
+                    pps=league.pps,
+                    lpm=round(lpm := (league.pps * 24), 2),
+                    apm=league.apm,
+                    apl=round(league.apm / lpm, 2),
+                    vs=league.vs,
+                    adpm=round(adpm := (league.vs * 0.6), 2),
+                    adpl=round(adpm / lpm, 2),
+                ),
+                tetra_league_history=TetraLeagueHistory(
+                    data=histories,
+                    split_interval=split_value,
+                    min_tr=value_min,
+                    max_tr=value_max,
+                    offset=offset,
+                ),
+                radar=Radar(
+                    app=(app := (league.apm / (60 * league.pps))),
+                    dsps=(dsps := ((league.vs / 100) - (league.apm / 60))),
+                    dspp=(dspp := (dsps / league.pps)),
+                    ci=150 * dspp - 125 * app + 50 * (league.vs / league.apm) - 25,
+                    ge=2 * ((app * dsps) / league.pps),
+                ),
+                sprint=sprint_value,
+                blitz=blitz_value,
+            ),
+        )
+    ) as page_hash:
+        return await screenshot(f'http://{netloc}/host/{page_hash}.html')
+
+
+N = TypeVar('N', int, float)
+
+
+def handling_special_value(value: N) -> N | None:
+    return value if value != -1 else None
+
+
+async def make_query_image_v2(player: Player) -> bytes:
+    user, user_info, sprint, blitz, zen = await gather(
+        player.user, player.get_info(), player.sprint, player.blitz, player.zen
+    )
+    league = get_league(user_info)
+    histories = await query_historical_data(user, user_info)
+
+    if sprint.record is not None:
+        duration = timedelta(milliseconds=sprint.record.endcontext.final_time).total_seconds()
+        sprint_value = f'{duration:.3f}s' if duration < 60 else f'{duration // 60:.0f}m {duration % 60:.3f}s'  # noqa: PLR2004
+    else:
+        sprint_value = 'N/A'
+
+    play_time: str | None
+    if (game_time := handling_special_value(user_info.data.user.gametime)) is not None:
+        if game_time // 3600 > 0:
+            play_time = f'{game_time//3600:.0f}h {game_time % 3600 // 60:.0f}m {game_time % 60:.0f}s'
+        elif game_time // 60 > 0:
+            play_time = f'{game_time//60:.0f}m {game_time % 60:.0f}s'
+        else:
+            play_time = f'{game_time:.0f}s'
+    else:
+        play_time = game_time
+    netloc = get_self_netloc()
+    async with HostPage(
+        await render(
+            'v2/tetrio/user/info',
+            V2TemplateInfo(
+                user=V2TemplateUser(
+                    id=user.ID,
+                    name=user.name.upper(),
+                    bio=user_info.data.user.bio,
+                    banner=f'http://{netloc}/host/resource/tetrio/banners/{user.ID}?{urlencode({"revision": user_info.data.user.banner_revision})}'
+                    if user_info.data.user.banner_revision is not None and user_info.data.user.banner_revision != 0
+                    else None,
+                    avatar=f'http://{netloc}/host/resource/tetrio/avatars/{user.ID}?{urlencode({"revision": user_info.data.user.avatar_revision})}'
+                    if user_info.data.user.avatar_revision is not None and user_info.data.user.avatar_revision != 0
+                    else Avatar(
+                        type='identicon',
+                        hash=md5(user_info.data.user.id.encode()).hexdigest(),  # noqa: S324
+                    ),
+                    badges=[
+                        Badge(
+                            id=i.id,
+                            description=i.label,
+                            group=i.group,
+                            receive_at=i.ts if isinstance(i.ts, datetime) else None,
+                        )
+                        for i in user_info.data.user.badges
+                    ],
+                    country=user_info.data.user.country,
+                    xp=user_info.data.user.xp,
+                    friend_count=user_info.data.user.friend_count or 0,
+                    supporter_tier=user_info.data.user.supporter_tier,
+                    bad_standing=user_info.data.user.badstanding or False,
+                    verified=user_info.data.user.verified,
+                    playtime=play_time,
+                    join_at=user_info.data.user.ts,
+                ),
+                tetra_league=V2TemplateTetraLeague(
+                    rank=league.rank,
+                    highest_rank=league.bestrank,
+                    tr=round(league.rating, 2),
+                    glicko=round(league.glicko, 2),
+                    rd=round(league.rd, 2),
+                    global_rank=handling_special_value(league.standing),
+                    country_rank=handling_special_value(league.standing_local),
+                    pps=(
+                        metrics := get_metrics(pps=league.pps, apm=league.apm, vs=league.vs)
+                        if league.vs is not None
+                        else get_metrics(pps=league.pps, apm=league.apm)
+                    ).pps,
+                    apm=metrics.apm,
+                    apl=metrics.apl,
+                    vs=metrics.vs if isinstance(metrics, TetrisMetricsProWithPPSVS) else None,
+                    adpl=metrics.adpl if isinstance(metrics, TetrisMetricsProWithPPSVS) else None,
+                    statistic=TetraLeagueStatistic(
+                        total=league.gamesplayed,
+                        wins=league.gameswon,
+                    ),
+                    decaying=league.decaying,
+                    history=histories,
+                )
+                if isinstance(league, RatedLeague)
+                else None,
+                statistic=Statistic(
+                    total=handling_special_value(user_info.data.user.gamesplayed),
+                    wins=handling_special_value(user_info.data.user.gameswon),
+                ),
+                sprint=Sprint(
+                    time=sprint_value,
+                    global_rank=sprint.rank,
+                    play_at=sprint.record.ts,
+                )
+                if sprint.record is not None
+                else None,
+                blitz=Blitz(
+                    score=blitz.record.endcontext.score,
+                    global_rank=blitz.rank,
+                    play_at=blitz.record.ts,
+                )
+                if blitz.record is not None
+                else None,
+                zen=zen,
+            ),
+        ),
+    ) as page_hash:
+        return await screenshot(f'http://{netloc}/host/{page_hash}.html')
+
+
+async def make_query_text(player: Player) -> UniMessage:
+    user, user_info, sprint, blitz = await gather(player.user, player.get_info(), player.sprint, player.blitz)
+    league = get_league(user_info)
+
+    user_name = user.name.upper()
+
+    message = ''
+    if isinstance(league, NeverPlayedLeague):
+        message += f'用户 {user_name} 没有排位统计数据'
+    else:
+        if isinstance(league, NeverRatedLeague):
+            message += f'用户 {user_name} 暂未完成定级赛, 最近十场的数据:'
+        else:
+            if league.rank == 'z':
+                message += f'用户 {user_name} 暂无段位, {round(league.rating,2)} TR'
+            else:
+                message += f'{league.rank.upper()} 段用户 {user_name} {round(league.rating,2)} TR (#{league.standing})'
+            message += f', 段位分 {round(league.glicko,2)}±{round(league.rd,2)}, 最近十场的数据:'
+        metrics = (
+            get_metrics(pps=league.pps, apm=league.apm, vs=league.vs)
+            if league.vs is not None
+            else get_metrics(pps=league.pps, apm=league.apm)
+        )
+        message += f"\nL'PM: {metrics.lpm} ( {metrics.pps} pps )"
+        message += f'\nAPM: {metrics.apm} ( x{metrics.apl} )'
+        if isinstance(metrics, TetrisMetricsProWithPPSVS):
+            message += f'\nADPM: {metrics.adpm} ( x{metrics.adpl} ) ( {metrics.vs}vs )'
+    if sprint.record is not None:
+        message += f'\n40L: {round(sprint.record.endcontext.final_time/1000,2)}s'
+        message += f' ( #{sprint.rank} )' if sprint.rank is not None else ''
+    if blitz.record is not None:
+        message += f'\nBlitz: {blitz.record.endcontext.score}'
+        message += f' ( #{blitz.rank} )' if blitz.rank is not None else ''
+    return UniMessage(message)
+
+
+async def make_query_result(player: Player, template: Template) -> UniMessage:
+    try:
+        if template == 'v1':
+            return UniMessage.image(raw=await make_query_image_v1(player))
+        if template == 'v2':
+            return UniMessage.image(raw=await make_query_image_v2(player))
+    except FallbackError:
+        ...
+    return await make_query_text(player)
+
+
+class FullExport:
+    cache: ClassVar[defaultdict[str, set[tuple[datetime, Number]]]] = defaultdict(set)
+    latest_update: ClassVar[date | None] = None
+
+    @classmethod
+    async def init(cls) -> None:
+        async with get_session() as session:
+            full_exports = (await session.scalars(select(IORank).where(IORank.update_time >= cls.start_time()))).all()
+        await gather(
+            *[
+                cls._load(update_time, file_hash)
+                for file_hash, update_time in {
+                    i.file_hash: i.update_time for i in full_exports if i.file_hash is not None
+                }.items()
+            ]
+        )
+
+    @classmethod
+    async def update(cls) -> None:
+        if cls.latest_update == datetime.now(tz=ZoneInfo('Asia/Shanghai')).date():
+            return
+        start_time = cls.start_time()
+        for i in cls.cache:
+            cls.cache[i] = {j for j in cls.cache[i] if j[0] >= start_time}
+        latest_time = max(cls.cache)
+        async with get_session() as session:
+            full_exports = (await session.scalars(select(IORank).where(IORank.update_time > latest_time))).all()
+        await gather(
+            *[
+                cls._load(update_time, file_hash)
+                for file_hash, update_time in {
+                    i.file_hash: i.update_time for i in full_exports if i.file_hash is not None
+                }.items()
+            ]
+        )
+        cls.latest_update = datetime.now(tz=ZoneInfo('Asia/Shanghai')).date()
+
+    @classmethod
+    def get_data(cls, unique_identifier: str) -> list[TetraLeagueHistoryData]:
+        return [TetraLeagueHistoryData(record_at=i[0], tr=i[1]) for i in cls.cache[unique_identifier]]
+
+    @classmethod
+    def start_time(cls) -> datetime:
+        return (
+            datetime.now(ZoneInfo('Asia/Shanghai')).replace(hour=0, minute=0, second=0, microsecond=0)
+            - timedelta(days=9)
+        ).astimezone(UTC)
+
+    @classmethod
+    async def _load(cls, update_time: datetime, file_hash: str) -> None:
+        try:
+            users = type_validate_json(TetraLeagueSuccess, await cls.decompress(file_hash)).data.users
+        except FileNotFoundError:
+            await cls.clear_invalid(file_hash)
+            return
+        update_time = update_time.astimezone(ZoneInfo('Asia/Shanghai'))
+        for i in users:
+            cls.cache[i.id].add((update_time, i.league.rating))
+
+    @classmethod
+    async def decompress(cls, file_hash: str) -> bytes:
+        async with open(get_data_file('nonebot_plugin_tetris_stats', f'{file_hash}.json.zst'), mode='rb') as file:
+            return ZstdDecompressor().decompress(await file.read())
+
+    @classmethod
+    async def clear_invalid(cls, file_hash: str) -> None:
+        async with get_session() as session:
+            full_exports = (await session.scalars(select(IORank).where(IORank.file_hash == file_hash))).all()
+            for i in full_exports:
+                i.file_hash = None
+            await session.commit()
+
+
+@driver.on_startup
+async def _():
+    await FullExport.init()
+    scheduler.add_job(FullExport.update, 'interval', hours=1)

nonebot_plugin_tetris_stats/games/tetrio/rank.py

@@ -0,0 +1,174 @@
+from collections import defaultdict
+from collections.abc import Callable
+from datetime import datetime, timedelta, timezone
+from hashlib import sha512
+from math import floor
+from statistics import mean
+from zoneinfo import ZoneInfo
+
+from aiofiles import open
+from nonebot import get_driver
+from nonebot.compat import model_dump
+from nonebot.matcher import Matcher
+from nonebot.utils import run_sync
+from nonebot_plugin_apscheduler import scheduler  # type: ignore[import-untyped]
+from nonebot_plugin_localstore import get_data_file  # type: ignore[import-untyped]
+from nonebot_plugin_orm import get_session
+from nonebot_plugin_session import EventSession  # type: ignore[import-untyped]
+from nonebot_plugin_session_orm import get_session_persist_id  # type: ignore[import-untyped]
+from sqlalchemy import func, select
+from zstandard import ZstdCompressor
+
+from ...db import trigger
+from ...utils.exception import RequestError
+from ...utils.metrics import get_metrics
+from ...utils.retry import retry
+from . import alc
+from .api.schemas.base import FailedModel
+from .api.schemas.tetra_league import ValidUser
+from .api.schemas.user import User
+from .api.tetra_league import full_export
+from .api.typing import Rank
+from .constant import GAME_TYPE, RANK_PERCENTILE
+from .models import IORank
+
+UTC = timezone.utc
+
+driver = get_driver()
+
+
+@alc.assign('TETRIO.rank')
+async def _(matcher: Matcher, rank: Rank, event_session: EventSession):
+    async with trigger(
+        session_persist_id=await get_session_persist_id(event_session),
+        game_platform=GAME_TYPE,
+        command_type='rank',
+        command_args=[],
+    ):
+        if rank == 'z':
+            await matcher.finish('暂不支持查询未知段位')
+        async with get_session() as session:
+            latest_data = (
+                await session.scalars(select(IORank).where(IORank.rank == rank).order_by(IORank.id.desc()).limit(1))
+            ).one()
+            compare_data = (
+                await session.scalars(
+                    select(IORank)
+                    .where(IORank.rank == rank)
+                    .order_by(
+                        func.abs(
+                            func.julianday(IORank.update_time)
+                            - func.julianday(latest_data.update_time - timedelta(hours=24))
+                        )
+                    )
+                    .limit(1)
+                )
+            ).one()
+        message = ''
+        if (datetime.now(UTC) - latest_data.update_time.replace(tzinfo=UTC)) > timedelta(hours=7):
+            message += 'Warning: 数据超过7小时未更新, 请联系Bot主人查看后台\n'
+        message += f'{rank.upper()} 段 分数线 {latest_data.tr_line:.2f} TR, {latest_data.player_count} 名玩家\n'
+        if compare_data.id != latest_data.id:
+            message += f'对比 {(latest_data.update_time-compare_data.update_time).total_seconds()/3600:.2f} 小时前趋势: {f"↑{difference:.2f}" if (difference:=latest_data.tr_line-compare_data.tr_line) > 0 else f"↓{-difference:.2f}" if difference < 0 else "→"}'
+        else:
+            message += '暂无对比数据'
+        avg = get_metrics(pps=latest_data.avg_pps, apm=latest_data.avg_apm, vs=latest_data.avg_vs)
+        low_pps = get_metrics(pps=latest_data.low_pps[1])
+        low_vs = get_metrics(vs=latest_data.low_vs[1])
+        max_pps = get_metrics(pps=latest_data.high_pps[1])
+        max_vs = get_metrics(vs=latest_data.high_vs[1])
+        message += (
+            '\n'
+            '平均数据:\n'
+            f"L'PM: {avg.lpm} ( {avg.pps} pps )\n"
+            f'APM: {avg.apm} ( x{avg.apl} )\n'
+            f'ADPM: {avg.adpm} ( x{avg.adpl} ) ( {avg.vs}vs )\n'
+            '\n'
+            '最低数据:\n'
+            f"L'PM: {low_pps.lpm} ( {low_pps.pps} pps ) By: {latest_data.low_pps[0]['name'].upper()}\n"
+            f'APM: {latest_data.low_apm[1]} By: {latest_data.low_apm[0]["name"].upper()}\n'
+            f'ADPM: {low_vs.adpm} ( {low_vs.vs}vs ) By: {latest_data.low_vs[0]["name"].upper()}\n'
+            '\n'
+            '最高数据:\n'
+            f"L'PM: {max_pps.lpm} ( {max_pps.pps} pps ) By: {latest_data.high_pps[0]['name'].upper()}\n"
+            f'APM: {latest_data.high_apm[1]} By: {latest_data.high_apm[0]["name"].upper()}\n'
+            f'ADPM: {max_vs.adpm} ( {max_vs.vs}vs ) By: {latest_data.high_vs[0]["name"].upper()}\n'
+            '\n'
+            f'数据更新时间: {latest_data.update_time.replace(tzinfo=UTC).astimezone(ZoneInfo("Asia/Shanghai")).strftime("%Y-%m-%d %H:%M:%S")}'
+        )
+        await matcher.finish(message)
+
+
+@scheduler.scheduled_job('cron', hour='0,6,12,18', minute=0)
+@retry(exception_type=RequestError, delay=timedelta(minutes=15))
+async def get_tetra_league_data() -> None:
+    league, original = await full_export(with_original=True)
+    if isinstance(league, FailedModel):
+        msg = f'排行榜数据请求错误:\n{league.error}'
+        raise RequestError(msg)
+
+    def pps(user: ValidUser) -> float:
+        return user.league.pps
+
+    def apm(user: ValidUser) -> float:
+        return user.league.apm
+
+    def vs(user: ValidUser) -> float:
+        return user.league.vs
+
+    def _min(users: list[ValidUser], field: Callable[[ValidUser], float]) -> ValidUser:
+        return min(users, key=field)
+
+    def _max(users: list[ValidUser], field: Callable[[ValidUser], float]) -> ValidUser:
+        return max(users, key=field)
+
+    def build_extremes_data(
+        users: list[ValidUser],
+        field: Callable[[ValidUser], float],
+        sort: Callable[[list[ValidUser], Callable[[ValidUser], float]], ValidUser],
+    ) -> tuple[dict[str, str], float]:
+        user = sort(users, field)
+        return model_dump(User(ID=user.id, name=user.username)), field(user)
+
+    data_hash: str | None = await run_sync((await run_sync(sha512)(original)).hexdigest)()
+    async with open(get_data_file('nonebot_plugin_tetris_stats', f'{data_hash}.json.zst'), mode='wb') as file:
+        await file.write(await run_sync(ZstdCompressor(level=12, threads=-1).compress)(original))
+
+    users = [i for i in league.data.users if isinstance(i, ValidUser)]
+    rank_to_users: defaultdict[Rank, list[ValidUser]] = defaultdict(list)
+    for i in users:
+        rank_to_users[i.league.rank].append(i)
+    rank_info: list[IORank] = []
+    for rank, percentile in RANK_PERCENTILE.items():
+        offset = floor((percentile / 100) * len(users)) - 1
+        tr_line = users[offset].league.rating
+        rank_users = rank_to_users[rank]
+        rank_info.append(
+            IORank(
+                rank=rank,
+                tr_line=tr_line,
+                player_count=len(rank_users),
+                low_pps=(build_extremes_data(rank_users, pps, _min)),
+                low_apm=(build_extremes_data(rank_users, apm, _min)),
+                low_vs=(build_extremes_data(rank_users, vs, _min)),
+                avg_pps=mean({i.league.pps for i in rank_users}),
+                avg_apm=mean({i.league.apm for i in rank_users}),
+                avg_vs=mean({i.league.vs for i in rank_users}),
+                high_pps=(build_extremes_data(rank_users, pps, _max)),
+                high_apm=(build_extremes_data(rank_users, apm, _max)),
+                high_vs=(build_extremes_data(rank_users, vs, _max)),
+                update_time=league.cache.cached_at,
+                file_hash=data_hash,
+            )
+        )
+    async with get_session() as session:
+        session.add_all(rank_info)
+        await session.commit()
+
+
+@driver.on_startup
+async def _() -> None:
+    async with get_session() as session:
+        latest_time = await session.scalar(select(IORank.update_time).order_by(IORank.id.desc()).limit(1))
+    if latest_time is None or datetime.now(tz=UTC) - latest_time.replace(tzinfo=UTC) > timedelta(hours=6):
+        await get_tetra_league_data()

nonebot_plugin_tetris_stats/games/tetrio/record/__init__.py

@@ -0,0 +1,6 @@
+from . import blitz, sprint
+
+__all__ = [
+    'blitz',
+    'sprint',
+]

nonebot_plugin_tetris_stats/games/tetrio/record/blitz.py

@@ -0,0 +1,124 @@
+from asyncio import gather
+from datetime import timedelta
+from hashlib import md5
+from urllib.parse import urlencode
+
+from nonebot.adapters import Event
+from nonebot.matcher import Matcher
+from nonebot_plugin_alconna import At
+from nonebot_plugin_alconna.uniseg import UniMessage
+from nonebot_plugin_orm import get_session
+from nonebot_plugin_session import EventSession  # type: ignore[import-untyped]
+from nonebot_plugin_user import get_user  # type: ignore[import-untyped]
+
+from ....db import query_bind_info
+from ....utils.exception import RecordNotFoundError
+from ....utils.host import HostPage, get_self_netloc
+from ....utils.metrics import get_metrics
+from ....utils.render import render
+from ....utils.render.schemas.base import Avatar
+from ....utils.render.schemas.tetrio.tetrio_record_base import Finesse, Max, Mini, Tspins, User
+from ....utils.render.schemas.tetrio.tetrio_record_blitz import Record, Statistic
+from ....utils.screenshot import screenshot
+from ....utils.typing import Me
+from ...constant import CANT_VERIFY_MESSAGE
+from .. import alc
+from ..api.player import Player
+from ..constant import GAME_TYPE
+
+
+@alc.assign('TETRIO.record.blitz')
+async def _(
+    event: Event,
+    matcher: Matcher,
+    target: At | Me,
+    event_session: EventSession,
+):
+    async with get_session() as session:
+        bind = await query_bind_info(
+            session=session,
+            user=await get_user(
+                event_session.platform, target.target if isinstance(target, At) else event.get_user_id()
+            ),
+            game_platform=GAME_TYPE,
+        )
+    if bind is None:
+        await matcher.finish('未查询到绑定信息')
+    message = UniMessage(CANT_VERIFY_MESSAGE)
+    player = Player(user_id=bind.game_account, trust=True)
+    await (message + UniMessage.image(raw=await make_blitz_image(player))).finish()
+
+
+@alc.assign('TETRIO.record.blitz')
+async def _(account: Player):
+    await UniMessage.image(raw=await make_blitz_image(account)).finish()
+
+
+async def make_blitz_image(player: Player) -> bytes:
+    user, user_info, blitz = await gather(player.user, player.get_info(), player.blitz)
+    if blitz.record is None:
+        msg = f'未找到用户 {user.name.upper()} 的 Blitz 记录'
+        raise RecordNotFoundError(msg)
+    endcontext = blitz.record.endcontext
+    clears = endcontext.clears
+    duration = timedelta(milliseconds=endcontext.final_time).total_seconds()
+    metrics = get_metrics(pps=endcontext.piecesplaced / duration)
+    netloc = get_self_netloc()
+    async with HostPage(
+        page=await render(
+            'v2/tetrio/record/blitz',
+            Record(
+                user=User(
+                    id=user.ID,
+                    name=user.name.upper(),
+                    avatar=f'http://{netloc}/host/resource/tetrio/avatars/{user.ID}?{urlencode({"revision": user_info.data.user.avatar_revision})}'
+                    if user_info.data.user.avatar_revision is not None and user_info.data.user.avatar_revision != 0
+                    else Avatar(
+                        type='identicon',
+                        hash=md5(user.ID.encode()).hexdigest(),  # noqa: S324
+                    ),
+                ),
+                replay_id=blitz.record.replayid,
+                rank=blitz.rank,
+                statistic=Statistic(
+                    keys=endcontext.inputs,
+                    kpp=round(endcontext.inputs / endcontext.piecesplaced, 2),
+                    kps=round(endcontext.inputs / duration, 2),
+                    max=Max(
+                        combo=max((0, endcontext.topcombo - 1)),
+                        btb=max((0, endcontext.topbtb - 1)),
+                    ),
+                    pieces=endcontext.piecesplaced,
+                    pps=metrics.pps,
+                    lines=endcontext.lines,
+                    lpm=metrics.lpm,
+                    holds=endcontext.holds,
+                    score=endcontext.score,
+                    spp=round(endcontext.score / endcontext.piecesplaced, 2),
+                    single=clears.singles,
+                    double=clears.doubles,
+                    triple=clears.triples,
+                    quad=clears.quads,
+                    tspins=Tspins(
+                        total=clears.realtspins,
+                        single=clears.tspinsingles,
+                        double=clears.tspindoubles,
+                        triple=clears.triples,
+                        mini=Mini(
+                            total=clears.minitspins,
+                            single=clears.minitspinsingles,
+                            double=clears.minitspindoubles,
+                        ),
+                    ),
+                    all_clear=clears.allclear,
+                    finesse=Finesse(
+                        faults=endcontext.finesse.faults,
+                        accuracy=round(endcontext.finesse.perfectpieces / endcontext.piecesplaced * 100, 2),
+                    ),
+                    level=endcontext.level,
+                ),
+                play_at=blitz.record.ts,
+            ),
+        )
+    ) as page_hash:
+        return await screenshot(f'http://{netloc}/host/{page_hash}.html')

nonebot_plugin_tetris_stats/games/tetrio/record/sprint.py

@@ -0,0 +1,124 @@
+from asyncio import gather
+from datetime import timedelta
+from hashlib import md5
+from urllib.parse import urlencode
+
+from nonebot.adapters import Event
+from nonebot.matcher import Matcher
+from nonebot_plugin_alconna import At
+from nonebot_plugin_alconna.uniseg import UniMessage
+from nonebot_plugin_orm import get_session
+from nonebot_plugin_session import EventSession  # type: ignore[import-untyped]
+from nonebot_plugin_user import get_user  # type: ignore[import-untyped]
+
+from ....db import query_bind_info
+from ....utils.exception import RecordNotFoundError
+from ....utils.host import HostPage, get_self_netloc
+from ....utils.metrics import get_metrics
+from ....utils.render import render
+from ....utils.render.schemas.base import Avatar
+from ....utils.render.schemas.tetrio.tetrio_record_base import Finesse, Max, Mini, Tspins, User
+from ....utils.render.schemas.tetrio.tetrio_record_sprint import Record, Statistic
+from ....utils.screenshot import screenshot
+from ....utils.typing import Me
+from ...constant import CANT_VERIFY_MESSAGE
+from .. import alc
+from ..api.player import Player
+from ..constant import GAME_TYPE
+
+
+@alc.assign('TETRIO.record.sprint')
+async def _(
+    event: Event,
+    matcher: Matcher,
+    target: At | Me,
+    event_session: EventSession,
+):
+    async with get_session() as session:
+        bind = await query_bind_info(
+            session=session,
+            user=await get_user(
+                event_session.platform, target.target if isinstance(target, At) else event.get_user_id()
+            ),
+            game_platform=GAME_TYPE,
+        )
+    if bind is None:
+        await matcher.finish('未查询到绑定信息')
+    message = UniMessage(CANT_VERIFY_MESSAGE)
+    player = Player(user_id=bind.game_account, trust=True)
+    await (message + UniMessage.image(raw=await make_sprint_image(player))).finish()
+
+
+@alc.assign('TETRIO.record.sprint')
+async def _(account: Player):
+    await UniMessage.image(raw=await make_sprint_image(account)).finish()
+
+
+async def make_sprint_image(player: Player) -> bytes:
+    user, user_info, sprint = await gather(player.user, player.get_info(), player.sprint)
+    if sprint.record is None:
+        msg = f'未找到用户 {user.name.upper()} 的 40L 记录'
+        raise RecordNotFoundError(msg)
+    endcontext = sprint.record.endcontext
+    clears = endcontext.clears
+    duration = timedelta(milliseconds=endcontext.final_time).total_seconds()
+    sprint_value = f'{duration:.3f}s' if duration < 60 else f'{duration // 60:.0f}m {duration % 60:.3f}s'  # noqa: PLR2004
+    metrics = get_metrics(pps=endcontext.piecesplaced / duration)
+    netloc = get_self_netloc()
+    async with HostPage(
+        page=await render(
+            'v2/tetrio/record/40l',
+            Record(
+                user=User(
+                    id=user.ID,
+                    name=user.name.upper(),
+                    avatar=f'http://{netloc}/host/resource/tetrio/avatars/{user.ID}?{urlencode({"revision": user_info.data.user.avatar_revision})}'
+                    if user_info.data.user.avatar_revision is not None and user_info.data.user.avatar_revision != 0
+                    else Avatar(
+                        type='identicon',
+                        hash=md5(user.ID.encode()).hexdigest(),  # noqa: S324
+                    ),
+                ),
+                time=sprint_value,
+                replay_id=sprint.record.replayid,
+                rank=sprint.rank,
+                statistic=Statistic(
+                    keys=endcontext.inputs,
+                    kpp=round(endcontext.inputs / endcontext.piecesplaced, 2),
+                    kps=round(endcontext.inputs / duration, 2),
+                    max=Max(
+                        combo=max((0, endcontext.topcombo - 1)),
+                        btb=max((0, endcontext.topbtb - 1)),
+                    ),
+                    pieces=endcontext.piecesplaced,
+                    pps=metrics.pps,
+                    lines=endcontext.lines,
+                    lpm=metrics.lpm,
+                    holds=endcontext.holds,
+                    score=endcontext.score,
+                    single=clears.singles,
+                    double=clears.doubles,
+                    triple=clears.triples,
+                    quad=clears.quads,
+                    tspins=Tspins(
+                        total=clears.realtspins,
+                        single=clears.tspinsingles,
+                        double=clears.tspindoubles,
+                        triple=clears.triples,
+                        mini=Mini(
+                            total=clears.minitspins,
+                            single=clears.minitspinsingles,
+                            double=clears.minitspindoubles,
+                        ),
+                    ),
+                    all_clear=clears.allclear,
+                    finesse=Finesse(
+                        faults=endcontext.finesse.faults,
+                        accuracy=round(endcontext.finesse.perfectpieces / endcontext.piecesplaced * 100, 2),
+                    ),
+                ),
+                play_at=sprint.record.ts,
+            ),
+        )
+    ) as page_hash:
+        return await screenshot(f'http://{netloc}/host/{page_hash}.html')

nonebot_plugin_tetris_stats/games/tetrio/typing.py

@@ -0,0 +1,3 @@
+from typing import Literal
+
+Template = Literal['v1', 'v2']

nonebot_plugin_tetris_stats/games/top/__init__.py

@@ -0,0 +1,59 @@
+from arclet.alconna import Arg, ArgFlag, Args, Subcommand
+from nonebot_plugin_alconna import At
+
+from ...utils.exception import MessageFormatError
+from ...utils.typing import Me
+from .. import add_block_handlers, alc, command
+from .api import Player
+from .constant import USER_NAME
+
+
+def get_player(name: str) -> Player | MessageFormatError:
+    if USER_NAME.match(name):
+        return Player(user_name=name, trust=True)
+    return MessageFormatError('用户名/ID不合法')
+
+
+command.add(
+    Subcommand(
+        'TOP',
+        Subcommand(
+            'bind',
+            Args(
+                Arg(
+                    'account',
+                    get_player,
+                    notice='TOP 用户名 / ID',
+                    flags=[ArgFlag.HIDDEN],
+                )
+            ),
+            help_text='绑定 TOP 账号',
+        ),
+        Subcommand(
+            'query',
+            Args(
+                Arg(
+                    'target',
+                    At | Me,
+                    notice='@想要查询的人 / 自己',
+                    flags=[ArgFlag.HIDDEN, ArgFlag.OPTIONAL],
+                ),
+                Arg(
+                    'account',
+                    get_player,
+                    notice='TOP 用户名',
+                    flags=[ArgFlag.HIDDEN, ArgFlag.OPTIONAL],
+                ),
+            ),
+            help_text='查询 TOP 游戏信息',
+        ),
+        help_text='TOP 游戏相关指令',
+    )
+)
+
+alc.shortcut('(?i:top)(?i:绑定|绑|bind)', {'command': 'tstats TOP bind', 'humanized': 'top绑定'})
+alc.shortcut('(?i:top)(?i:查询|查|query|stats)', {'command': 'tstats TOP query', 'humanized': 'top查'})
+
+add_block_handlers(alc.assign('TOP.query'))
+
+from . import bind, query  # noqa: E402, F401

nonebot_plugin_tetris_stats/games/top/api/__init__.py

@@ -0,0 +1,3 @@
+from .player import Player
+
+__all__ = ['Player']

nonebot_plugin_tetris_stats/games/top/api/models.py

@@ -0,0 +1,17 @@
+from datetime import datetime
+from typing import Literal
+
+from nonebot_plugin_orm import Model
+from sqlalchemy import DateTime, String
+from sqlalchemy.orm import Mapped, MappedAsDataclass, mapped_column
+
+from ....db.models import PydanticType
+from .schemas.user_profile import UserProfile
+
+
+class TOPHistoricalData(MappedAsDataclass, Model):
+    id: Mapped[int] = mapped_column(init=False, primary_key=True)
+    user_unique_identifier: Mapped[str] = mapped_column(String(24), index=True)
+    api_type: Mapped[Literal['User Profile']] = mapped_column(String(16), index=True)
+    data: Mapped[UserProfile] = mapped_column(PydanticType(get_model=[], models={UserProfile}))
+    update_time: Mapped[datetime] = mapped_column(DateTime, index=True)

nonebot_plugin_tetris_stats/games/top/api/player.py

@@ -0,0 +1,71 @@
+from contextlib import suppress
+from datetime import datetime, timezone
+from io import StringIO
+from urllib.parse import urlencode
+
+from lxml import etree
+from pandas import read_html
+
+from ....db import anti_duplicate_add
+from ....utils.request import Request, splice_url
+from ..constant import BASE_URL, USER_NAME
+from .models import TOPHistoricalData
+from .schemas.user import User
+from .schemas.user_profile import Data, UserProfile
+
+UTC = timezone.utc
+
+
+class Player:
+    def __init__(self, *, user_name: str, trust: bool = False) -> None:
+        self.user_name = user_name
+        if not trust and not USER_NAME.match(self.user_name):
+            msg = 'Invalid user name'
+            raise ValueError(msg)
+        self.__user: User | None = None
+        self._user_profile: UserProfile | None = None
+
+    @property
+    async def user(self) -> User:
+        if self.__user is None:
+            profile = await self.get_profile()
+            self.__user = User(user_name=profile.user_name)
+        return self.__user
+
+    async def get_profile(self) -> UserProfile:
+        """获取用户信息"""
+        if self._user_profile is None:
+            url = splice_url([BASE_URL, 'profile.php', f'?{urlencode({"user":self.user_name})}'])
+            raw_user_profile = await Request.request(url, is_json=False)
+            self._user_profile = self._parse_profile(raw_user_profile)
+            await anti_duplicate_add(
+                TOPHistoricalData,
+                TOPHistoricalData(
+                    user_unique_identifier=(await self.user).unique_identifier,
+                    api_type='User Profile',
+                    data=self._user_profile,
+                    update_time=datetime.now(tz=UTC),
+                ),
+            )
+        return self._user_profile
+
+    def _parse_profile(self, original_user_profile: bytes) -> UserProfile:
+        html = etree.HTML(original_user_profile)
+        user_name = html.xpath('//div[@class="mycontent"]/h1/text()')[0].replace("'s profile", '')
+        today = None
+        with suppress(ValueError):
+            today = Data(
+                lpm=float(str(html.xpath('//div[@class="mycontent"]/text()[3]')[0]).replace('lpm:', '').strip()),
+                apm=float(str(html.xpath('//div[@class="mycontent"]/text()[4]')[0]).replace('apm:', '').strip()),
+            )
+        table = StringIO(
+            etree.tostring(
+                html.xpath('//div[@class="mycontent"]/table[@class="mytable"]')[0],
+                encoding='utf-8',
+            ).decode()
+        )
+        dataframe = read_html(table, encoding='utf-8', header=0)[0]
+        total: list[Data] = []
+        for _, value in dataframe.iterrows():
+            total.append(Data(lpm=value['lpm'], apm=value['apm']))
+        return UserProfile(user_name=user_name, today=today, total=total)

nonebot_plugin_tetris_stats/games/top/api/schemas/user.py

@@ -0,0 +1,17 @@
+from typing import Literal
+
+from typing_extensions import override
+
+from ....schemas import BaseUser
+from ...constant import GAME_TYPE
+
+
+class User(BaseUser):
+    platform: Literal['TOP'] = GAME_TYPE
+
+    user_name: str
+
+    @property
+    @override
+    def unique_identifier(self) -> str:
+        return self.user_name

nonebot_plugin_tetris_stats/games/top/api/schemas/user_profile.py

@@ -0,0 +1,12 @@
+from pydantic import BaseModel
+
+
+class Data(BaseModel):
+    lpm: float
+    apm: float
+
+
+class UserProfile(BaseModel):
+    user_name: str
+    today: Data | None
+    total: list[Data] | None

nonebot_plugin_tetris_stats/games/top/bind.py

@@ -0,0 +1,62 @@
+from nonebot_plugin_alconna.uniseg import UniMessage
+from nonebot_plugin_orm import get_session
+from nonebot_plugin_session import EventSession  # type: ignore[import-untyped]
+from nonebot_plugin_session_orm import get_session_persist_id  # type: ignore[import-untyped]
+from nonebot_plugin_user import User  # type: ignore[import-untyped]
+from nonebot_plugin_userinfo import BotUserInfo, EventUserInfo, UserInfo  # type: ignore[import-untyped]
+
+from ...db import BindStatus, create_or_update_bind, trigger
+from ...utils.host import HostPage, get_self_netloc
+from ...utils.image import get_avatar
+from ...utils.render import Bind, render
+from ...utils.render.schemas.base import People
+from ...utils.screenshot import screenshot
+from . import alc
+from .api import Player
+from .constant import GAME_TYPE
+
+
+@alc.assign('TOP.bind')
+async def _(
+    nb_user: User,
+    account: Player,
+    event_session: EventSession,
+    event_user_info: UserInfo = EventUserInfo(),  # noqa: B008
+    bot_info: UserInfo = BotUserInfo(),  # noqa: B008
+):
+    async with trigger(
+        session_persist_id=await get_session_persist_id(event_session),
+        game_platform=GAME_TYPE,
+        command_type='bind',
+        command_args=[],
+    ):
+        user = await account.user
+        async with get_session() as session:
+            bind_status = await create_or_update_bind(
+                session=session,
+                user=nb_user,
+                game_platform=GAME_TYPE,
+                game_account=user.unique_identifier,
+            )
+        if bind_status in (BindStatus.SUCCESS, BindStatus.UPDATE):
+            async with HostPage(
+                await render(
+                    'v1/binding',
+                    Bind(
+                        platform=GAME_TYPE,
+                        status='unknown',
+                        user=People(
+                            avatar=await get_avatar(event_user_info, 'Data URI', None),
+                            name=user.user_name,
+                        ),
+                        bot=People(
+                            avatar=await get_avatar(bot_info, 'Data URI', '../../static/logo/logo.svg'),
+                            name=bot_info.user_name,
+                        ),
+                        command='top查我',
+                    ),
+                )
+            ) as page_hash:
+                await UniMessage.image(
+                    raw=await screenshot(f'http://{get_self_netloc()}/host/{page_hash}.html')
+                ).finish()

nonebot_plugin_tetris_stats/games/top/constant.py

@@ -0,0 +1,8 @@
+from re import compile
+from typing import Literal
+
+GAME_TYPE: Literal['TOP'] = 'TOP'
+
+BASE_URL = 'http://tetrisonline.pl/top/'
+
+USER_NAME = compile(r'^[a-zA-Z0-9_]{1,16}$')

nonebot_plugin_tetris_stats/games/top/query.py

@@ -0,0 +1,74 @@
+from nonebot.adapters import Event
+from nonebot.matcher import Matcher
+from nonebot_plugin_alconna import At
+from nonebot_plugin_alconna.uniseg import UniMessage
+from nonebot_plugin_orm import get_session
+from nonebot_plugin_session import EventSession  # type: ignore[import-untyped]
+from nonebot_plugin_session_orm import get_session_persist_id  # type: ignore[import-untyped]
+from nonebot_plugin_user import get_user  # type: ignore[import-untyped]
+
+from ...db import query_bind_info, trigger
+from ...utils.metrics import get_metrics
+from ...utils.typing import Me
+from ..constant import CANT_VERIFY_MESSAGE
+from . import alc
+from .api import Player
+from .api.schemas.user_profile import UserProfile
+from .constant import GAME_TYPE
+
+
+@alc.assign('TOP.query')
+async def _(event: Event, matcher: Matcher, target: At | Me, event_session: EventSession):
+    async with trigger(
+        session_persist_id=await get_session_persist_id(event_session),
+        game_platform=GAME_TYPE,
+        command_type='query',
+        command_args=[],
+    ):
+        async with get_session() as session:
+            bind = await query_bind_info(
+                session=session,
+                user=await get_user(
+                    event_session.platform, target.target if isinstance(target, At) else event.get_user_id()
+                ),
+                game_platform=GAME_TYPE,
+            )
+        if bind is None:
+            await matcher.finish('未查询到绑定信息')
+        message = CANT_VERIFY_MESSAGE
+        await (message + make_query_text(await Player(user_name=bind.game_account, trust=True).get_profile())).finish()
+
+
+@alc.assign('TOP.query')
+async def _(account: Player, event_session: EventSession):
+    async with trigger(
+        session_persist_id=await get_session_persist_id(event_session),
+        game_platform=GAME_TYPE,
+        command_type='query',
+        command_args=[],
+    ):
+        await (make_query_text(await account.get_profile())).finish()
+
+
+def make_query_text(profile: UserProfile) -> UniMessage:
+    message = ''
+    if profile.today is not None:
+        today = get_metrics(lpm=profile.today.lpm, apm=profile.today.apm)
+        message += f'用户 {profile.user_name} 24小时内统计数据为: '
+        message += f"\nL'PM: {today.lpm} ( {today.pps} pps )"
+        message += f'\nAPM: {today.apm} ( x{today.apl} )'
+    else:
+        message += f'用户 {profile.user_name} 暂无24小时内统计数据'
+    if profile.total is not None:
+        total_lpm = total_apm = 0.0
+        for value in profile.total:
+            total_lpm += value.lpm
+            total_apm += value.apm
+        num = len(profile.total)
+        total = get_metrics(lpm=total_lpm / num, apm=total_apm / num)
+        message += '\n历史统计数据为: '
+        message += f"\nL'PM: {total.lpm} ( {total.pps} pps )"
+        message += f'\nAPM: {total.apm} ( x{total.apl} )'
+    else:
+        message += '\n暂无历史统计数据'
+    return UniMessage(message)

nonebot_plugin_tetris_stats/games/tos/__init__.py

@@ -0,0 +1,64 @@
+from arclet.alconna import Arg, ArgFlag, Args, Subcommand
+from nonebot_plugin_alconna import At
+
+from ...utils.exception import MessageFormatError
+from ...utils.typing import Me
+from .. import add_block_handlers, alc, command
+from .api import Player
+from .constant import USER_NAME
+
+
+def get_player(teaid_or_name: str) -> Player | MessageFormatError:
+    if (
+        teaid_or_name.startswith(('onebot-', 'qqguild-', 'kook-', 'discord-'))
+        and teaid_or_name.split('-', maxsplit=1)[1].isdigit()
+    ):
+        return Player(teaid=teaid_or_name, trust=True)
+    if USER_NAME.match(teaid_or_name) and not teaid_or_name.isdigit() and 2 <= len(teaid_or_name) <= 18:  # noqa: PLR2004
+        return Player(user_name=teaid_or_name, trust=True)
+    return MessageFormatError('用户名/ID不合法')
+
+
+command.add(
+    Subcommand(
+        'TOS',
+        Subcommand(
+            'bind',
+            Args(
+                Arg(
+                    'account',
+                    get_player,
+                    notice='茶服 用户名 / ID',
+                    flags=[ArgFlag.HIDDEN],
+                )
+            ),
+            help_text='绑定 茶服 账号',
+        ),
+        Subcommand(
+            'query',
+            Args(
+                Arg(
+                    'target',
+                    At | Me,
+                    notice='@想要查询的人 / 自己',
+                    flags=[ArgFlag.HIDDEN, ArgFlag.OPTIONAL],
+                ),
+                Arg(
+                    'account',
+                    get_player,
+                    notice='茶服 用户名 / TeaID',
+                    flags=[ArgFlag.HIDDEN, ArgFlag.OPTIONAL],
+                ),
+            ),
+            help_text='查询 茶服 游戏信息',
+        ),
+        help_text='茶服 游戏相关指令',
+    )
+)
+
+alc.shortcut('(?i:tos|茶服)(?i:绑定|绑|bind)', {'command': 'tstats TOS bind', 'humanized': '茶服绑定'})
+alc.shortcut('(?i:tos|茶服)(?i:查询|查|query|stats)', {'command': 'tstats TOS query', 'humanized': '茶服查'})
+
+add_block_handlers(alc.assign('TOS.query'))
+
+from . import bind, query  # noqa: E402, F401

nonebot_plugin_tetris_stats/games/tos/api/__init__.py

@@ -0,0 +1,3 @@
+from .player import Player
+
+__all__ = ['Player']

nonebot_plugin_tetris_stats/games/tos/api/models.py

@@ -0,0 +1,20 @@
+from datetime import datetime
+from typing import Literal
+
+from nonebot_plugin_orm import Model
+from sqlalchemy import DateTime, String
+from sqlalchemy.orm import Mapped, MappedAsDataclass, mapped_column
+
+from ....db.models import PydanticType
+from .schemas.user_info import UserInfoSuccess
+from .schemas.user_profile import UserProfile
+
+
+class TOSHistoricalData(MappedAsDataclass, Model):
+    id: Mapped[int] = mapped_column(init=False, primary_key=True)
+    user_unique_identifier: Mapped[str] = mapped_column(String(24), index=True)
+    api_type: Mapped[Literal['User Info', 'User Profile']] = mapped_column(String(16), index=True)
+    data: Mapped[UserInfoSuccess | UserProfile] = mapped_column(
+        PydanticType(get_model=[], models={UserInfoSuccess, UserProfile})
+    )
+    update_time: Mapped[datetime] = mapped_column(DateTime, index=True)

nonebot_plugin_tetris_stats/games/tos/api/player.py

@@ -0,0 +1,128 @@
+from datetime import datetime, timezone
+from typing import overload
+from urllib.parse import urlencode
+
+from httpx import TimeoutException
+from nonebot.compat import type_validate_json
+
+from ....db import anti_duplicate_add
+from ....utils.exception import RequestError
+from ....utils.request import Request, splice_url
+from ..constant import BASE_URL, USER_NAME
+from .models import TOSHistoricalData
+from .schemas.user import User
+from .schemas.user_info import UserInfo, UserInfoSuccess
+from .schemas.user_profile import UserProfile
+
+UTC = timezone.utc
+
+
+class Player:
+    @overload
+    def __init__(self, *, teaid: str, trust: bool = False): ...
+    @overload
+    def __init__(self, *, user_name: str, trust: bool = False): ...
+    def __init__(self, *, teaid: str | None = None, user_name: str | None = None, trust: bool = False):
+        self.teaid = teaid
+        self.user_name = user_name
+        if not trust:
+            if self.teaid is not None:
+                if (
+                    not self.teaid.startswith(('onebot-', 'qqguild-', 'kook-', 'discord-'))
+                    or not self.teaid.split('-', maxsplit=1)[1].isdigit()
+                ):
+                    msg = 'Invalid teaid'
+                    raise ValueError(msg)
+            elif self.user_name is not None:
+                if not USER_NAME.match(self.user_name) or self.user_name.isdigit() or 2 > len(self.user_name) > 18:  # noqa: PLR2004
+                    msg = 'Invalid user name'
+                    raise ValueError(msg)
+            else:
+                msg = 'Invalid user'
+                raise ValueError(msg)
+        self.__user: User | None = None
+        self._user_info: UserInfoSuccess | None = None
+        self._user_profile: dict[str, UserProfile] = {}
+
+    @property
+    async def user(self) -> User:
+        if self.__user is None:
+            user_info = await self.get_info()
+            self.__user = User(teaid=user_info.data.teaid, name=user_info.data.name)
+            self.teaid = user_info.data.teaid
+            self.user_name = user_info.data.name
+        return self.__user
+
+    async def get_info(self) -> UserInfoSuccess:
+        """获取用户信息"""
+        if self._user_info is None:
+            if self.teaid is not None:
+                url = [
+                    splice_url(
+                        [
+                            i,
+                            'getTeaIdInfo',
+                            f'?{urlencode({"teaId":self.teaid})}',
+                        ]
+                    )
+                    for i in BASE_URL
+                ]
+            else:
+                url = [
+                    splice_url(
+                        [
+                            i,
+                            'getUsernameInfo',
+                            f'?{urlencode({"username":self.user_name})}',
+                        ]
+                    )
+                    for i in BASE_URL
+                ]
+            raw_user_info = await Request.failover_request(url, failover_code=[502], failover_exc=(TimeoutException,))
+            user_info: UserInfo = type_validate_json(UserInfo, raw_user_info)  # type: ignore[arg-type]
+            if not isinstance(user_info, UserInfoSuccess):
+                msg = f'用户信息请求错误:\n{user_info.error}'
+                raise RequestError(msg)
+            self._user_info = user_info
+            await anti_duplicate_add(
+                TOSHistoricalData,
+                TOSHistoricalData(
+                    user_unique_identifier=(await self.user).unique_identifier,
+                    api_type='User Info',
+                    data=user_info,
+                    update_time=datetime.now(UTC),
+                ),
+            )
+        return self._user_info
+
+    async def get_profile(self, other_parameter: dict[str, str | bytes] | None = None) -> UserProfile:
+        """获取用户数据"""
+        if other_parameter is None:
+            other_parameter = {}
+        params = urlencode(dict(sorted(other_parameter.items())))
+        if self._user_profile.get(params) is None:
+            raw_user_profile = await Request.failover_request(
+                [
+                    splice_url(
+                        [
+                            i,
+                            'getProfile',
+                            f'?{urlencode({"id":self.teaid or self.user_name,**other_parameter})}',
+                        ]
+                    )
+                    for i in BASE_URL
+                ],
+                failover_code=[502],
+                failover_exc=(TimeoutException,),
+            )
+            self._user_profile[params] = type_validate_json(UserProfile, raw_user_profile)
+            await anti_duplicate_add(
+                TOSHistoricalData,
+                TOSHistoricalData(
+                    user_unique_identifier=(await self.user).unique_identifier,
+                    api_type='User Profile',
+                    data=self._user_profile[params],
+                    update_time=datetime.now(UTC),
+                ),
+            )
+        return self._user_profile[params]

nonebot_plugin_tetris_stats/games/tos/api/schemas/user.py

@@ -0,0 +1,18 @@
+from typing import Literal
+
+from typing_extensions import override
+
+from ....schemas import BaseUser
+from ...constant import GAME_TYPE
+
+
+class User(BaseUser):
+    platform: Literal['TOS'] = GAME_TYPE
+
+    teaid: str
+    name: str
+
+    @property
+    @override
+    def unique_identifier(self) -> str:
+        return self.teaid

nonebot_plugin_tetris_stats/games/tos/api/schemas/user_info.py

@@ -0,0 +1,89 @@
+from datetime import datetime
+from typing import Literal
+
+from pydantic import BaseModel, Field
+
+
+class PeriodMatch(BaseModel):
+    name: str
+    teaid: str = Field(..., alias='teaId')
+    rating: str
+    rd: str
+    start_time: datetime = Field(..., alias='startTime')
+    end_time: datetime = Field(..., alias='endTime')
+    win: str
+    lose: str
+    score: str
+
+
+class UserDataTotalItem(BaseModel):
+    time_map: str = Field(..., alias='timeMap')
+    pieces_map: str = Field(..., alias='piecesMap')
+    clear_lines_map: str = Field(..., alias='clearLinesMap')
+    attacks_map: str = Field(..., alias='attacksMap')
+    dig_map: str = Field(..., alias='digMap')
+    send_map: str = Field(..., alias='sendMap')
+    rise_map: str = Field(..., alias='riseMap')
+    offset_map: str = Field(..., alias='offsetMap')
+    receive_map: str = Field(..., alias='receiveMap')
+    games_map: str = Field(..., alias='gamesMap')
+    tetris_map: str = Field(..., alias='tetrisMap')
+    combo_map: str = Field(..., alias='comboMap')
+    tspin_map: str = Field(..., alias='tspinMap')
+    b2b_map: str = Field(..., alias='b2bMap')
+    perfect_clear_map: str = Field(..., alias='perfectClearMap')
+    time_no_map: str = Field(..., alias='timeNoMap')
+    pieces_no_map: str = Field(..., alias='piecesNoMap')
+    clear_lines_no_map: str = Field(..., alias='clearLinesNoMap')
+    attacks_no_map: str = Field(..., alias='attacksNoMap')
+    dig_no_map: str = Field(..., alias='digNoMap')
+    send_no_map: str = Field(..., alias='sendNoMap')
+    rise_no_map: str = Field(..., alias='riseNoMap')
+    offset_no_map: str = Field(..., alias='offsetNoMap')
+    receive_no_map: str = Field(..., alias='receiveNoMap')
+    games_no_map: str = Field(..., alias='gamesNoMap')
+    tetris_no_map: str = Field(..., alias='tetrisNoMap')
+    combo_no_map: str = Field(..., alias='comboNoMap')
+    tspin_no_map: str = Field(..., alias='tspinNoMap')
+    b2b_no_map: str = Field(..., alias='b2bNoMap')
+    perfect_clear_no_map: str = Field(..., alias='perfectClearNoMap')
+
+
+class Data(BaseModel):
+    teaid: str = Field(..., alias='teaId')
+    name: str
+    total_exp: str = Field(..., alias='totalExp')
+    ranking: str
+    ranked_games: str = Field(..., alias='rankedGames')
+    rating_now: str = Field(..., alias='ratingNow')
+    rd_now: str = Field(..., alias='rdNow')
+    vol_now: str = Field(..., alias='volNow')
+    rating_last: str = Field(..., alias='ratingLast')
+    rd_last: str = Field(..., alias='rdLast')
+    vol_last: str = Field(..., alias='volLast')
+    period_matches: list[PeriodMatch] = Field(..., alias='periodMatches')
+    user_data_total: list[UserDataTotalItem] = Field(..., alias='userDataTotal')
+    ranking_items: str = Field(..., alias='rankingItems')
+    ranking_game_items: str = Field(..., alias='rankingGameItems')
+    training_level: str = Field(..., alias='trainingLevel')
+    training_wins: str = Field(..., alias='trainingWins')
+    pb_sprint: str = Field(..., alias='PBSprint')
+    pb_marathon: str = Field(..., alias='PBMarathon')
+    pb_challenge: str = Field(..., alias='PBChallenge')
+    register_date: datetime = Field(..., alias='registerDate')
+    last_login_date: datetime = Field(..., alias='lastLoginDate')
+
+
+class UserInfoSuccess(BaseModel):
+    code: int
+    success: Literal[True]
+    data: Data
+
+
+class FailedModel(BaseModel):
+    code: int
+    success: Literal[False]
+    error: str
+
+
+UserInfo = UserInfoSuccess | FailedModel

nonebot_plugin_tetris_stats/games/tos/api/schemas/user_profile.py

@@ -0,0 +1,34 @@
+from datetime import datetime
+from typing import Literal
+
+from pydantic import BaseModel
+
+
+class Data(BaseModel):
+    idmultiplayergameresult: int
+    iduser: str
+    teaid: str
+    time: int
+    clear_lines: int
+    attack: int
+    send: int
+    offset: int
+    receive: int
+    rise: int
+    dig: int
+    pieces: int
+    max_combo: int
+    pc_count: int
+    place: int
+    num_players: int
+    fumen_code: Literal['0', '1']  # wtf
+    rule_set: str
+    garbage: str
+    idmultiplayergame: int
+    datetime: datetime
+
+
+class UserProfile(BaseModel):
+    code: int
+    success: bool
+    data: list[Data]

nonebot_plugin_tetris_stats/games/tos/bind.py

@@ -0,0 +1,62 @@
+from nonebot_plugin_alconna.uniseg import UniMessage
+from nonebot_plugin_orm import get_session
+from nonebot_plugin_session import EventSession  # type: ignore[import-untyped]
+from nonebot_plugin_session_orm import get_session_persist_id  # type: ignore[import-untyped]
+from nonebot_plugin_user import User  # type: ignore[import-untyped]
+from nonebot_plugin_userinfo import BotUserInfo, EventUserInfo, UserInfo  # type: ignore[import-untyped]
+
+from ...db import BindStatus, create_or_update_bind, trigger
+from ...utils.host import HostPage, get_self_netloc
+from ...utils.image import get_avatar
+from ...utils.render import Bind, render
+from ...utils.render.schemas.base import People
+from ...utils.screenshot import screenshot
+from . import alc
+from .api import Player
+from .constant import GAME_TYPE
+
+
+@alc.assign('TOS.bind')
+async def _(
+    nb_user: User,
+    account: Player,
+    event_session: EventSession,
+    event_user_info: UserInfo = EventUserInfo(),  # noqa: B008
+    bot_info: UserInfo = BotUserInfo(),  # noqa: B008
+):
+    async with trigger(
+        session_persist_id=await get_session_persist_id(event_session),
+        game_platform=GAME_TYPE,
+        command_type='bind',
+        command_args=[],
+    ):
+        user = await account.user
+        async with get_session() as session:
+            bind_status = await create_or_update_bind(
+                session=session,
+                user=nb_user,
+                game_platform=GAME_TYPE,
+                game_account=user.unique_identifier,
+            )
+        user_info = await account.get_info()
+        if bind_status in (BindStatus.SUCCESS, BindStatus.UPDATE):
+            async with HostPage(
+                await render(
+                    'v1/binding',
+                    Bind(
+                        platform=GAME_TYPE,
+                        status='unknown',
+                        user=People(
+                            avatar=await get_avatar(event_user_info, 'Data URI', None), name=user_info.data.name
+                        ),
+                        bot=People(
+                            avatar=await get_avatar(bot_info, 'Data URI', '../../static/logo/logo.svg'),
+                            name=bot_info.user_remark or bot_info.user_displayname or bot_info.user_name,
+                        ),
+                        command='茶服查我',
+                    ),
+                )
+            ) as page_hash:
+                await UniMessage.image(
+                    raw=await screenshot(f'http://{get_self_netloc()}/host/{page_hash}.html')
+                ).finish()

nonebot_plugin_tetris_stats/games/tos/constant.py

@@ -0,0 +1,16 @@
+from re import compile
+from typing import Literal
+
+GAME_TYPE: Literal['TOS'] = 'TOS'
+
+BASE_URL = {
+    'https://teatube.cn:8888/',
+    'http://cafuuchino1.studio26f.org:19970',
+    'http://cafuuchino2.studio26f.org:19970',
+    'http://cafuuchino3.studio26f.org:19970',
+    'http://cafuuchino4.studio26f.org:19970',
+}
+
+USER_NAME = compile(
+    r'^(?!\.)(?!com[0-9]$)(?!con$)(?!lpt[0-9]$)(?!nul$)(?!prn$)[^\-][^\+][^\|\*\?\\\s\!:<>/$"]*[^\.\|\*\?\\\s\!:<>/$"]+$'
+)

nonebot_plugin_tetris_stats/games/tos/query.py

@@ -0,0 +1,252 @@
+from asyncio import gather
+from datetime import timedelta
+from http import HTTPStatus
+from typing import Literal, NamedTuple
+
+from nonebot.adapters import Event
+from nonebot.matcher import Matcher
+from nonebot_plugin_alconna import At
+from nonebot_plugin_alconna.uniseg import UniMessage
+from nonebot_plugin_orm import get_session
+from nonebot_plugin_session import EventSession  # type: ignore[import-untyped]
+from nonebot_plugin_session_orm import get_session_persist_id  # type: ignore[import-untyped]
+from nonebot_plugin_user import get_user  # type: ignore[import-untyped]
+from nonebot_plugin_userinfo import EventUserInfo, UserInfo  # type: ignore[import-untyped]
+
+from ...db import query_bind_info, trigger
+from ...utils.exception import RequestError
+from ...utils.host import HostPage, get_self_netloc
+from ...utils.image import get_avatar
+from ...utils.metrics import TetrisMetricsProWithLPMADPM, get_metrics
+from ...utils.render import render
+from ...utils.render.schemas.base import People, Ranking
+from ...utils.render.schemas.tos_info import Info, Multiplayer, Radar
+from ...utils.screenshot import screenshot
+from ...utils.typing import Me, Number
+from ..constant import CANT_VERIFY_MESSAGE
+from . import alc
+from .api import Player
+from .api.schemas.user_info import UserInfoSuccess
+from .constant import GAME_TYPE
+
+
+def add_special_handlers(
+    teaid_prefix: Literal['onebot-', 'kook-', 'discord-', 'qqguild-'], match_event: type[Event]
+) -> None:
+    @alc.assign('TOS.query')
+    async def _(
+        event: Event,
+        target: At | Me,
+        event_session: EventSession,
+        event_user_info: UserInfo = EventUserInfo(),  # noqa: B008
+    ):
+        if isinstance(event, match_event):
+            async with trigger(
+                session_persist_id=await get_session_persist_id(event_session),
+                game_platform=GAME_TYPE,
+                command_type='query',
+                command_args=[],
+            ):
+                player = Player(
+                    teaid=f'{teaid_prefix}{target.target}'
+                    if isinstance(target, At)
+                    else f'{teaid_prefix}{event.get_user_id()}',
+                    trust=True,
+                )
+                try:
+                    user_info, game_data = await gather(player.get_info(), get_game_data(player))
+                    if game_data is not None:
+                        await UniMessage.image(
+                            raw=await make_query_image(user_info, game_data, event_user_info)
+                        ).finish()
+                    await make_query_text(user_info, game_data).finish()
+                except RequestError as e:
+                    if e.status_code == HTTPStatus.BAD_REQUEST and '未找到此用户' in e.message:
+                        return
+                    raise
+
+
+try:
+    from nonebot.adapters.onebot.v11 import MessageEvent as OB11MessageEvent
+
+    add_special_handlers('onebot-', OB11MessageEvent)
+except ImportError:
+    pass
+
+try:
+    from nonebot.adapters.qq.event import GuildMessageEvent as QQGuildMessageEvent
+    from nonebot.adapters.qq.event import QQMessageEvent
+
+    add_special_handlers('qqguild-', QQGuildMessageEvent)
+    add_special_handlers('onebot-', QQMessageEvent)
+except ImportError:
+    pass
+
+try:
+    from nonebot.adapters.kaiheila.event import MessageEvent as KookMessageEvent
+
+    add_special_handlers('kook-', KookMessageEvent)
+except ImportError:
+    pass
+
+try:
+    from nonebot.adapters.discord import MessageEvent as DiscordMessageEvent
+
+    add_special_handlers('discord-', DiscordMessageEvent)
+except ImportError:
+    pass
+
+
+@alc.assign('TOS.query')
+async def _(
+    event: Event,
+    matcher: Matcher,
+    target: At | Me,
+    event_session: EventSession,
+    event_user_info: UserInfo = EventUserInfo(),  # noqa: B008
+):
+    async with trigger(
+        session_persist_id=await get_session_persist_id(event_session),
+        game_platform=GAME_TYPE,
+        command_type='query',
+        command_args=[],
+    ):
+        async with get_session() as session:
+            bind = await query_bind_info(
+                session=session,
+                user=await get_user(
+                    event_session.platform, target.target if isinstance(target, At) else event.get_user_id()
+                ),
+                game_platform=GAME_TYPE,
+            )
+        if bind is None:
+            await matcher.finish('未查询到绑定信息')
+        message = CANT_VERIFY_MESSAGE
+        player = Player(teaid=bind.game_account, trust=True)
+        user_info, game_data = await gather(player.get_info(), get_game_data(player))
+        if game_data is not None:
+            await (
+                message + UniMessage.image(raw=await make_query_image(user_info, game_data, event_user_info))
+            ).finish()
+        await (message + make_query_text(user_info, game_data)).finish()
+
+
+@alc.assign('TOS.query')
+async def _(account: Player, event_session: EventSession, event_user_info: UserInfo = EventUserInfo()):  # noqa: B008
+    async with trigger(
+        session_persist_id=await get_session_persist_id(event_session),
+        game_platform=GAME_TYPE,
+        command_type='query',
+        command_args=[],
+    ):
+        user_info, game_data = await gather(account.get_info(), get_game_data(account))
+        if game_data is not None:
+            await UniMessage.image(raw=await make_query_image(user_info, game_data, event_user_info)).finish()
+        await make_query_text(user_info, game_data).finish()
+
+
+class GameData(NamedTuple):
+    game_num: int
+    metrics: TetrisMetricsProWithLPMADPM
+    OR: Number
+    dspp: Number
+    ge: Number
+
+
+async def get_game_data(player: Player, query_num: int = 50) -> GameData | None:
+    """获取游戏数据"""
+    user_profile = await player.get_profile()
+    if user_profile.data == []:
+        return None
+    weighted_total_lpm = weighted_total_apm = weighted_total_adpm = total_time = 0.0
+    total_attack = total_dig = total_offset = total_pieses = total_receive = num = 0
+    for i in user_profile.data:
+        # 排除单人局和时间为0的游戏
+        # 茶: 不计算没挖掘的局, 即使apm和lpm也如此
+        if i.num_players == 1 or i.time == 0 or i.dig is None:
+            continue
+        # 加权计算
+        time = i.time / 1000
+        lpm = 24 * (i.pieces / time)
+        apm = (i.attack / time) * 60
+        adpm = ((i.attack + i.dig) / time) * 60
+        weighted_total_lpm += lpm * time
+        weighted_total_apm += apm * time
+        weighted_total_adpm += adpm * time
+        total_attack += i.attack
+        total_dig += i.dig
+        total_offset += i.offset
+        total_pieses += i.pieces
+        total_receive += i.receive
+        total_time += time
+        num += 1
+        if num >= query_num:
+            break
+    if num == 0:
+        return None
+    # TODO: 如果有效局数小于 {查询数} , 并且没有无dig信息的局, 且 user_profile.data 内有{请求数}个局, 则继续往前获取信息
+    metrics = get_metrics(
+        lpm=weighted_total_lpm / total_time, apm=weighted_total_apm / total_time, adpm=weighted_total_adpm / total_time
+    )
+    return GameData(
+        game_num=num,
+        metrics=metrics,
+        OR=total_offset / total_receive * 100,
+        dspp=total_dig / total_pieses,
+        ge=2 * ((total_attack * total_dig) / total_pieses**2),
+    )
+
+
+async def make_query_image(user_info: UserInfoSuccess, game_data: GameData, event_user_info: UserInfo) -> bytes:
+    metrics = game_data.metrics
+    duration = timedelta(milliseconds=float(user_info.data.pb_sprint)).total_seconds()
+    sprint_value = f'{duration:.3f}s' if duration < 60 else f'{duration // 60:.0f}m {duration % 60:.3f}s'  # noqa: PLR2004
+    async with HostPage(
+        await render(
+            'v1/tos/info',
+            Info(
+                user=People(avatar=await get_avatar(event_user_info, 'Data URI', None), name=user_info.data.name),
+                ranking=Ranking(rating=float(user_info.data.ranking), rd=round(float(user_info.data.rd_now), 2)),
+                multiplayer=Multiplayer(
+                    pps=metrics.pps,
+                    lpm=metrics.lpm,
+                    apm=metrics.apm,
+                    apl=metrics.apl,
+                    vs=metrics.vs,
+                    adpm=metrics.adpm,
+                    adpl=metrics.adpl,
+                ),
+                radar=Radar(
+                    app=(app := (metrics.apm / (60 * metrics.pps))),
+                    OR=game_data.OR,
+                    dspp=game_data.dspp,
+                    ci=150 * game_data.dspp - 125 * app + 50 * (metrics.vs / metrics.apm) - 25,
+                    ge=game_data.ge,
+                ),
+                sprint=sprint_value,
+                challenge=f'{int(user_info.data.pb_challenge):,}' if user_info.data.pb_challenge != '0' else 'N/A',
+                marathon=f'{int(user_info.data.pb_marathon):,}' if user_info.data.pb_marathon != '0' else 'N/A',
+            ),
+        )
+    ) as page_hash:
+        return await screenshot(f'http://{get_self_netloc()}/host/{page_hash}.html')
+
+
+def make_query_text(user_info: UserInfoSuccess, game_data: GameData | None) -> UniMessage:
+    user_data = user_info.data
+    message = f'用户 {user_data.name} ({user_data.teaid}) '
+    if user_data.ranked_games == '0':
+        message += '暂无段位统计数据'
+    else:
+        message += f', 段位分 {round(float(user_data.rating_now),2)}±{round(float(user_data.rd_now),2)} ({round(float(user_data.vol_now),2)}) '
+    if game_data is None:
+        message += ', 暂无游戏数据'
+    else:
+        message += f', 最近 {game_data.game_num} 局数据'
+        message += f"\nL'PM: {game_data.metrics.lpm} ( {game_data.metrics.pps} pps )"
+        message += f'\nAPM: {game_data.metrics.apm} ( x{game_data.metrics.apl} )'
+        message += f'\nADPM: {game_data.metrics.adpm} ( x{game_data.metrics.adpl} ) ( {game_data.metrics.vs}vs )'
+    message += f'\n40L: {float(user_data.pb_sprint)/1000:.2f}s' if user_data.pb_sprint != '2147483647' else ''
+    message += f'\nMarathon: {user_data.pb_marathon}' if user_data.pb_marathon != '0' else ''
+    message += f'\nChallenge: {user_data.pb_challenge}' if user_data.pb_challenge != '0' else ''
+    return UniMessage(message)

nonebot_plugin_tetris_stats/utils/__init__.py

@@ -1 +0,0 @@
-from . import config, database, message_analyzer

nonebot_plugin_tetris_stats/utils/browser.py

@@ -0,0 +1,87 @@
+import sys
+from os import environ
+from platform import system
+from re import sub
+
+from nonebot import get_driver
+from nonebot.log import logger
+from playwright.__main__ import main
+from playwright.async_api import Browser, async_playwright
+
+driver = get_driver()
+
+global_config = driver.config
+
+
+@driver.on_startup
+async def _():
+    await BrowserManager.init_playwright()
+
+
+@driver.on_shutdown
+async def _():
+    await BrowserManager.close_browser()
+
+
+class BrowserManager:
+    """浏览器管理类"""
+
+    _browser: Browser | None = None
+
+    @classmethod
+    async def init_playwright(cls) -> None:
+        if system() == 'Windows' and getattr(global_config, 'fastapi_reload', False):
+            msg = '加载失败, Windows 必须设置 FASTAPI_RELOAD=false 才能正常运行 playwright'
+            raise ImportError(msg)
+        logger.info('开始 安装/更新 playwright 浏览器')
+        environ['PLAYWRIGHT_DOWNLOAD_HOST'] = 'https://npmmirror.com/mirrors/playwright/'
+        if cls._call_playwright(['', 'install', 'firefox']):
+            logger.success('安装/更新 playwright 浏览器成功')
+        else:
+            logger.warning('playwright 浏览器 安装/更新 失败, 尝试使用原始仓库下载')
+            del environ['PLAYWRIGHT_DOWNLOAD_HOST']
+            if cls._call_playwright(['', 'install', 'firefox']):
+                logger.success('安装/更新 playwright 浏览器成功')
+            else:
+                logger.error('安装/更新 playwright 浏览器失败')
+        try:
+            await cls._start_browser()
+        except BaseException as e:  # 不知道会有什么异常, 交给用户解决
+            msg = 'playwright 启动失败, 请尝试在命令行运行 playwright install-deps firefox, 如果仍然启动失败, 请参考上面的报错👆'
+            raise ImportError(msg) from e
+        else:
+            logger.success('playwright 启动成功')
+
+    @classmethod
+    def _call_playwright(cls, argv: list[str]) -> bool:
+        """等价于调用 playwright 的命令行程序"""
+        argv_backup = sys.argv.copy()
+        sys.argv[0] = sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
+        sys.argv = argv
+        try:
+            main()
+        except SystemExit as e:
+            return e.code == 0
+        except BaseException:  # noqa: BLE001
+            return False
+        finally:
+            sys.argv = argv_backup
+        return True
+
+    @classmethod
+    async def _start_browser(cls) -> Browser:
+        """启动浏览器实例"""
+        playwright = await async_playwright().start()
+        cls._browser = await playwright.firefox.launch()
+        return cls._browser
+
+    @classmethod
+    async def get_browser(cls) -> Browser:
+        """获取浏览器实例"""
+        return cls._browser or await cls._start_browser()
+
+    @classmethod
+    async def close_browser(cls) -> None:
+        """关闭浏览器实例"""
+        if isinstance(cls._browser, Browser):
+            await cls._browser.close()

nonebot_plugin_tetris_stats/utils/config.py

@@ -1,7 +0,0 @@
-from pydantic import BaseModel
-
-
-class Config(BaseModel):
-    '''配置类'''
-    cache_path: str = 'cache/nonebot_plugin_tetris_stats/cache'
-    db_path: str = 'data/nonebot_plugin_tetris_stats/data.db'

nonebot_plugin_tetris_stats/utils/database.py

@@ -1,140 +0,0 @@
-import datetime
-import os
-from asyncio import gather
-from sqlite3 import Connection, connect
-
-from nonebot import get_driver
-from nonebot.log import logger
-
-from .config import Config
-
-driver = get_driver()
-
-config = Config.parse_obj(get_driver().config)
-
-
-@driver.on_startup
-async def _():
-    await DataBase.init_db()
-
-
-@driver.on_shutdown
-async def _():
-    await DataBase.close_db()
-
-
-class DataBase():
-    '''数据库交互类'''
-    _db: Connection | None = None
-
-    @classmethod
-    async def init_db(cls) -> Connection:
-        '''初始化数据库'''
-        if not os.path.exists(os.path.dirname(config.db_path)):
-            os.makedirs(os.path.dirname(config.db_path))
-        cls._db = connect(config.db_path)
-        cursor = cls._db.cursor()
-        cursor.execute('''CREATE TABLE IF NOT EXISTS IOBIND
-                        (QQ INTEGER NOT NULL,
-                        USER TEXT NOT NULL)''')
-        cursor.execute('''CREATE TABLE IF NOT EXISTS TOPBIND
-                        (QQ INTEGER NOT NULL,
-                        USER TEXT NOT NULL)''')
-        cursor.execute('''CREATE TABLE IF NOT EXISTS IORANK
-                        (RANK VARCHAR(2) NOT NULL,
-                         TRENDING CHAR(1) NOT NULL,
-                         TRLINE FLOAT NOT NULL,
-                         PLAYERCOUNT INTEGER NOT NULL,
-                         AVGAPM FLOAT NOT NULL,
-                         AVGPPS FLOAT NOT NULL,
-                         ARGVS FLOAT NOT NULL,
-                         DATE TEXT NOT NULL)''')
-        cls._db.commit()
-        logger.info('数据库初始化完成')
-        return cls._db
-
-    @classmethod
-    async def query_rank_info_today(cls, rank: str) -> list | None:
-        '''查询段位信息'''
-        db = await cls._get_db()
-        cursor = db.cursor()
-        cursor.execute('''SELECT TRENDING, TRLINE, PLAYERCOUNT, AVGAPM, AVGPPS, ARGVS, DATE
-                        FROM IORANK
-                        WHERE RANK = ? AND DATE = ?''', (rank, datetime.date.today()))
-        result = cursor.fetchone()
-
-        if result is None:
-            return None
-
-        return list(result)
-
-    @classmethod
-    async def write_rank_info_today(cls, rank: str, trline: int, playercount: int, avgapm: float, avgpps: float, avgvs: float):
-        '''写入段位信息'''
-
-        db = await cls._get_db()
-        cursor = db.cursor()
-        cursor.execute('''SELECT TRLINE
-                        FROM IORANK
-                        WHERE RANK = ? AND DATE = ?''', (rank, datetime.date.today() - datetime.timedelta(days=1)))
-
-        result = cursor.fetchone()
-
-        if result is None:
-            trending = '?'
-        else:
-            if result[0] > trline:
-                trending = '↑'
-            else:
-                trending = '↓'
-
-        cursor.execute('''INSERT INTO IORANK
-                        (RANK, TRENDING, TRLINE, PLAYERCOUNT, AVGAPM, AVGPPS, ARGVS, DATE)
-                        VALUES (?,?,?,?,?,?,?,?)''',
-                       (rank, trending, trline, playercount, avgapm, avgpps, avgvs, datetime.date.today()))
-
-        db.commit()
-
-    @classmethod
-    async def _get_db(cls) -> Connection:
-        '''获取数据库对象'''
-        return cls._db or await cls.init_db()
-
-    @classmethod
-    async def query_bind_info(cls, qq_number: str | int, game_type: str) -> str | None:
-        '''查询绑定信息'''
-        db = await cls._get_db()
-        cursor = db.cursor()
-        cursor.execute(
-            f'SELECT USER FROM {game_type}BIND WHERE QQ = {qq_number}')
-        user = cursor.fetchone()
-        if user is None:
-            return None
-        return user[0]
-
-    @classmethod
-    async def write_bind_info(cls, qq_number: str | int, user: str, game_type: str) -> str:
-        '''写入绑定信息'''
-        bind_info, db = await gather(
-            cls.query_bind_info(qq_number=qq_number, game_type=game_type),
-            cls._get_db()
-        )
-        cursor = db.cursor()
-        if bind_info is not None:
-            cursor.execute(
-                f'UPDATE {game_type}BIND SET USER = ? WHERE QQ = ?', (user, qq_number))
-            message = '更新成功'
-        elif bind_info is None:
-            cursor.execute(
-                f'INSERT INTO {game_type}BIND (QQ, USER) VALUES (?, ?)', (qq_number, user))
-            message = '绑定成功'
-        else:
-            raise ValueError('预期外行为, 请上报GitHub')
-        db.commit()
-        return message
-
-    @classmethod
-    async def close_db(cls) -> None:
-        '''关闭数据库对象'''
-        if isinstance(cls._db, Connection):
-            cls._db.close()

nonebot_plugin_tetris_stats/utils/exception.py

@@ -0,0 +1,47 @@
+class TetrisStatsError(Exception):
+    """所有 TetrisStats 发生的异常基类"""
+
+    def __init__(self, message: str = ''):
+        self.message = message
+
+    def __str__(self) -> str:
+        return self.message
+
+    def __repr__(self) -> str:
+        return self.message
+
+
+class NeedCatchError(TetrisStatsError):
+    """需要被捕获的异常基类"""
+
+
+class RequestError(NeedCatchError):
+    """请求错误"""
+
+    def __init__(self, message: str = '', *, status_code: int | None = None):
+        super().__init__(message)
+        self.status_code = status_code
+
+
+class MessageFormatError(NeedCatchError):
+    """用户发送的消息格式不正确"""
+
+
+class RecordNotFoundError(NeedCatchError):
+    """找不到用户的某种记录"""
+
+
+class FallbackError(NeedCatchError):
+    """需要回滚至更通用的方法"""
+
+
+class DoNotCatchError(TetrisStatsError):
+    """不应该被捕获的异常基类"""
+
+
+class WhatTheFuckError(DoNotCatchError):
+    """用于表示不应该出现的情况 ("""
+
+
+class NoFallbackError(DoNotCatchError):  # 暂时没用 但是先写了
+    """没有可用的回退方法"""

nonebot_plugin_tetris_stats/utils/host.py

@@ -0,0 +1,90 @@
+from hashlib import sha256
+from ipaddress import IPv4Address, IPv6Address
+from typing import TYPE_CHECKING, ClassVar, Literal
+
+from fastapi import FastAPI, Path, status
+from fastapi.responses import FileResponse, HTMLResponse, Response
+from fastapi.staticfiles import StaticFiles
+from nonebot import get_app, get_driver
+from nonebot.log import logger
+
+from ..config.config import CACHE_PATH
+from .image import img_to_png
+from .request import Request
+from .templates import templates_dir
+
+if TYPE_CHECKING:
+    from pydantic import IPvAnyAddress
+
+app: FastAPI = get_app()
+
+driver = get_driver()
+
+global_config = driver.config
+
+
+if not isinstance(app, FastAPI):
+    msg = '本插件需要 FastAPI 驱动器才能运行'
+    raise RuntimeError(msg)  # noqa: TRY004
+
+NOT_FOUND = HTMLResponse('404 Not Found', status_code=status.HTTP_404_NOT_FOUND)
+
+
+class HostPage:
+    pages: ClassVar[dict[str, str]] = {}
+
+    def __init__(self, page: str) -> None:
+        self.page_hash = sha256(page.encode()).hexdigest()
+        self.pages[self.page_hash] = page
+
+    async def __aenter__(self) -> str:
+        return self.page_hash
+
+    async def __aexit__(self, exc_type, exc, tb) -> None:  # noqa: ANN001
+        self.pages.pop(self.page_hash, None)
+
+
+@driver.on_startup
+def _():
+    app.mount(
+        '/host/assets',
+        StaticFiles(directory=templates_dir / 'assets'),
+        name='assets',
+    )
+    logger.success('assets mounted')
+
+
+@app.get('/host/{page_hash}.html', status_code=status.HTTP_200_OK)
+async def _(page_hash: str) -> HTMLResponse:
+    if page_hash in HostPage.pages:
+        return HTMLResponse(HostPage.pages[page_hash])
+    return NOT_FOUND
+
+
+@app.get('/host/resource/tetrio/{resource_type}/{user_id}', status_code=status.HTTP_200_OK)
+async def _(
+    resource_type: Literal['avatars', 'banners'], revision: int, user_id: str = Path(regex=r'^[a-f0-9]{24}$')
+) -> Response:
+    if not (path := CACHE_PATH / 'tetrio' / resource_type / f'{user_id}_{revision}.png').exists():
+        path.parent.mkdir(parents=True, exist_ok=True)
+        path.write_bytes(
+            img_to_png(
+                await Request.request(
+                    f'https://tetr.io/user-content/{resource_type}/{user_id}.jpg?rv={revision}', is_json=False
+                )
+            )
+        )
+    return FileResponse(path)
+
+
+def get_self_netloc() -> str:
+    host: IPv4Address | IPv6Address | IPvAnyAddress = global_config.host
+    if isinstance(host, IPv4Address):
+        if host == IPv4Address('0.0.0.0'):  # noqa: S104
+            host = IPv4Address('127.0.0.1')
+        netloc = f'{host}:{global_config.port}'
+    else:
+        if host == IPv6Address('::'):
+            host = IPv6Address('::1')
+        netloc = f'[{host}]:{global_config.port}'
+    return netloc

nonebot_plugin_tetris_stats/utils/image.py

@@ -0,0 +1,62 @@
+from base64 import b64encode
+from io import BytesIO
+from typing import Literal, overload
+
+from nonebot_plugin_userinfo import UserInfo  # type: ignore[import-untyped]
+from PIL import Image
+
+
+@overload
+async def get_avatar(user: UserInfo, scheme: Literal['Data URI'], default: str | None) -> str:
+    """获取用户头像的指定格式
+
+    Args:
+        user (UserInfo): 要获取的用户
+        scheme (Literal['Data URI']): 格式
+        default (str | None): 获取不到时的默认值
+
+    Raises:
+        TypeError: Can't get avatar: 当获取不到头像并且没有设置默认值时抛出
+        TypeError: Can't get avatar format: 当获取到的头像无法识别格式时抛出
+
+    Returns:
+        str: Data URI 格式的头像
+    """
+
+
+@overload
+async def get_avatar(user: UserInfo, scheme: Literal['bytes'], default: str | None) -> bytes:
+    """获取用户头像的指定格式
+
+    Args:
+        user (UserInfo): 要获取的用户
+        scheme (Literal['bytes']): 格式
+        default (str | None): 获取不到时的默认值
+
+    Returns:
+        bytes: bytes 格式的头像
+    """
+
+
+async def get_avatar(user: UserInfo, scheme: Literal['Data URI', 'bytes'], default: str | None) -> str | bytes:
+    if user.user_avatar is None:
+        if default is None:
+            msg = "Can't get avatar"
+            raise TypeError(msg)
+        return default
+    bot_avatar = await user.user_avatar.get_image()
+    if scheme == 'Data URI':
+        avatar_format = Image.open(BytesIO(bot_avatar)).format
+        if avatar_format is None:
+            msg = "Can't get avatar format"
+            raise TypeError(msg)
+        return f'data:{Image.MIME[avatar_format]};base64,{b64encode(bot_avatar).decode()}'
+    return bot_avatar
+
+
+def img_to_png(image: bytes) -> bytes:
+    """将图片转换为 PNG 格式"""
+    result = BytesIO()
+    with Image.open(BytesIO(image)) as img:
+        img.save(result, 'PNG')
+    return result.getvalue()

nonebot_plugin_tetris_stats/utils/message_analyzer.py

@@ -1,91 +0,0 @@
-from re import match, sub
-
-
-async def handle_bind_message(message: str, game_type: str) -> tuple[str | None, tuple]:
-    '''返回值为tuple[gameType, tuple[message, user]]'''
-    _cmd_aliases = {'IO': ['io绑定', 'iobind'],
-                    'TOP': ['top绑定', 'topbind']}
-    # 剔除命令前缀
-    for i in _cmd_aliases[game_type]:
-        if match(rf'(?i){i}', message):
-            message = sub(rf'(?i){i}', '', message)
-            message = message.strip()
-            break
-    else:
-        raise ValueError('预期外行为, 请上报GitHub')
-    if message == '' or message.isspace():
-        return None, ('用户名为空', None)
-    return await check_name(message, game_type)
-
-
-async def handle_stats_query_message(message: str, game_type: str) -> tuple[str | None, tuple]:
-    '''返回值为tuple[gameType, tuple[message, user]]'''
-    _cmd_aliases = {'IO': ['io查', 'iostats'],
-                    'TOS': ['tos查', 'tostats', 'tosstats', '茶服查', '茶服stats'],
-                    'TOP': ['top查', 'topstats']}
-    _me = [
-        '我', '自己', '我等', '卑人', '愚', '老身', '爷', '老娘', '本姑娘', '本大爷', '鄙人', '寡人',
-        '小生', '贫僧', '本人', '孤', '吾', '俺', '咱', '私', 'me', '洒家', '在下', '偶', '人家',
-        '本小姐', '老夫', '老子', '朕', '本尊', '僕', '拙者', '妾', '儂', '自分', '吾輩', '我輩', '某',
-        '己等', '俺等', '此方', '哥', '姐', '劳资', '本宝宝', '余', '本喵',  'watashi',  'i', 'myself',
-        'self', 'oneself'
-    ]
-    # 剔除命令前缀
-    for i in _cmd_aliases[game_type]:
-        if match(rf'(?i){i}', message):
-            message = sub(rf'(?i){i}', '', message)
-            message = message.strip()
-            break
-    if message == '' or message.isspace():
-        return None, ('用户名为空', None)
-    if message.startswith('[CQ:at,qq='):
-        try:
-            user = int(str(message).split('[CQ:at,qq=')[1].split(']')[0])
-        except ValueError:
-            return None, ('QQ号码不合法', None)
-        else:
-            return 'AT', (None, user)
-    elif message in _me:
-        # 会不会有人叫本姑娘 本大爷这种或许可以成为id的名字呢
-        # TODO: 在判断是否可能是查自己的情况的时候 也去判断是否能成立为一个UserName
-        return 'ME', (None, None)
-    else:
-        return await check_name(message, game_type)
-
-
-async def handle_rank_message(message: str) -> str:
-    _cmd_aliases = ['io段位', 'iorank']
-    # 剔除命令前缀
-    for i in _cmd_aliases:
-        if match(rf'(?i){i}', message):
-            message = sub(rf'(?i){i}', '', message)
-            message = message.strip()
-            break
-    else:
-        raise ValueError('预期外行为, 请上报GitHub')
-    return message
-
-
-async def check_name(name: str, game_type: str) -> tuple[str | None, tuple]:
-    '''返回值为tuple[gameType, tuple[message, user]]'''
-    if game_type == 'IO':
-        if match(r'^[a-f0-9]{24}$', name):
-            return 'ID', (None, name)
-        if match(r'^[a-zA-Z0-9_-]{3,16}$', name):
-            return 'Name', (None, name.lower())
-        return None, ('用户名不合法', None)
-    if game_type == 'TOP':
-        if match(r'^[a-zA-Z0-9_]{1,16}$', name):
-            return 'Name', (None, name)
-        return None, ('用户名不合法', None)
-    if game_type == 'TOS':
-        if (match(r'^(?!\.)(?!com[0-9]$)(?!con$)(?!lpt[0-9]$)(?!nul$)(?!prn$)[^\-][^\+][^\|\*\?\\\s\!:<>/$"]*[^\.\|\*\?\\\s\!:<>/$"]+$', name)
-            and name.isdigit() is False
-                and 2 <= len(name) <= 18):
-            # 虽然我也不想这么长 但是似乎确实得这么长
-            # TODO 简化正则表达式
-            return 'Name', (None, name)
-        if name.isdigit() is True:
-            return 'QQ', (None, name)
-        return None, ('用户名不合法', None)
-    return None, ('游戏类型错误', None)

nonebot_plugin_tetris_stats/utils/metrics.py

@@ -0,0 +1,282 @@
+from typing import overload
+
+from .typing import Number
+
+
+class TetrisMetricsBaseWithPPS:
+    def __init__(self, pps: Number, precision: int) -> None:
+        self._pps = pps
+        self.precision = precision
+
+    @property
+    def _lpm(self) -> Number:
+        return self._pps * 24
+
+    @property
+    def lpm(self) -> Number:
+        return round(self._lpm, self.precision)
+
+    @property
+    def pps(self) -> Number:
+        return round(self._pps, self.precision)
+
+
+class TetrisMetricsBaseWithLPM:
+    def __init__(self, lpm: Number, precision: int) -> None:
+        self._lpm = lpm
+        self.precision = precision
+
+    @property
+    def lpm(self) -> Number:
+        return round(self._lpm, self.precision)
+
+    @property
+    def _pps(self) -> Number:
+        return self._lpm / 24
+
+    @property
+    def pps(self) -> Number:
+        return round(self._pps, self.precision)
+
+
+class TetrisMetricsBaseWithVS:
+    def __init__(self, vs: Number, precision: int) -> None:
+        self._vs = vs
+        self.precision = precision
+
+    @property
+    def vs(self) -> Number:
+        return round(self._vs, self.precision)
+
+    @property
+    def _adpm(self) -> Number:
+        return self._vs * 0.6
+
+    @property
+    def adpm(self) -> Number:
+        return round(self._adpm, self.precision)
+
+
+class TetrisMetricsBaseWithADPM:
+    def __init__(self, adpm: Number, precision: int) -> None:
+        self._adpm = adpm
+        self.precision = precision
+
+    @property
+    def _vs(self) -> Number:
+        return self._adpm / 0.6
+
+    @property
+    def vs(self) -> Number:
+        return round(self._vs, self.precision)
+
+    @property
+    def adpm(self) -> Number:
+        return round(self._adpm, self.precision)
+
+
+class TetrisMetricsBasicWithPPS(TetrisMetricsBaseWithPPS):
+    def __init__(self, pps: Number, apm: Number, precision: int) -> None:
+        super().__init__(pps=pps, precision=precision)
+        self._apm = apm
+
+    @property
+    def apm(self) -> Number:
+        return round(self._apm, self.precision)
+
+    @property
+    def apl(self) -> Number:
+        return round(self._apm / self._lpm, self.precision)
+
+
+class TetrisMetricsBasicWithLPM(TetrisMetricsBaseWithLPM):
+    def __init__(self, lpm: Number, apm: Number, precision: int):
+        super().__init__(lpm=lpm, precision=precision)
+        self._apm = apm
+
+    @property
+    def apm(self) -> Number:
+        return round(self._apm, self.precision)
+
+    @property
+    def apl(self) -> Number:
+        return round(self._apm / self._lpm, self.precision)
+
+
+class TetrisMetricsProWithPPSVS(TetrisMetricsBasicWithPPS, TetrisMetricsBaseWithVS):
+    def __init__(self, pps: Number, apm: Number, vs: Number, precision: int) -> None:
+        super().__init__(pps=pps, apm=apm, precision=precision)
+        super(TetrisMetricsBaseWithPPS, self).__init__(vs=vs, precision=precision)
+
+    @property
+    def adpl(self) -> Number:
+        return round(self._adpm / self._lpm, self.precision)
+
+
+class TetrisMetricsProWithPPSADPM(TetrisMetricsBasicWithPPS, TetrisMetricsBaseWithADPM):
+    def __init__(self, pps: Number, apm: Number, adpm: Number, precision: int) -> None:
+        super().__init__(pps=pps, apm=apm, precision=precision)
+        super(TetrisMetricsBaseWithPPS, self).__init__(adpm=adpm, precision=precision)
+
+    @property
+    def adpl(self) -> Number:
+        return round(self._adpm / self._lpm, self.precision)
+
+
+class TetrisMetricsProWithLPMVS(TetrisMetricsBasicWithLPM, TetrisMetricsBaseWithVS):
+    def __init__(self, lpm: Number, apm: Number, vs: Number, precision: int) -> None:
+        super().__init__(lpm=lpm, apm=apm, precision=precision)
+        super(TetrisMetricsBaseWithLPM, self).__init__(vs=vs, precision=precision)
+
+    @property
+    def adpl(self) -> Number:
+        return round(self._adpm / self._lpm, self.precision)
+
+
+class TetrisMetricsProWithLPMADPM(TetrisMetricsBasicWithLPM, TetrisMetricsBaseWithADPM):
+    def __init__(self, lpm: Number, apm: Number, adpm: Number, precision: int) -> None:
+        super().__init__(lpm=lpm, apm=apm, precision=precision)
+        super(TetrisMetricsBaseWithLPM, self).__init__(adpm=adpm, precision=precision)
+
+    @property
+    def adpl(self) -> Number:
+        return round(self._adpm / self._lpm, self.precision)
+
+
+@overload
+def get_metrics(
+    *,
+    pps: Number,
+    precision: int = 2,
+) -> TetrisMetricsBaseWithPPS: ...
+
+
+@overload
+def get_metrics(
+    *,
+    lpm: Number,
+    precision: int = 2,
+) -> TetrisMetricsBaseWithLPM: ...
+
+
+@overload
+def get_metrics(
+    *,
+    vs: Number,
+    precision: int = 2,
+) -> TetrisMetricsBaseWithVS: ...
+
+
+@overload
+def get_metrics(
+    *,
+    adpm: Number,
+    precision: int = 2,
+) -> TetrisMetricsBaseWithADPM: ...
+
+
+@overload
+def get_metrics(
+    *,
+    pps: Number,
+    apm: Number,
+    precision: int = 2,
+) -> TetrisMetricsBasicWithPPS: ...
+
+
+@overload
+def get_metrics(
+    *,
+    lpm: Number,
+    apm: Number,
+    precision: int = 2,
+) -> TetrisMetricsBasicWithLPM: ...
+
+
+@overload
+def get_metrics(
+    *,
+    pps: Number,
+    apm: Number,
+    vs: Number,
+    precision: int = 2,
+) -> TetrisMetricsProWithPPSVS: ...
+
+
+@overload
+def get_metrics(
+    *,
+    pps: Number,
+    apm: Number,
+    adpm: Number,
+    precision: int = 2,
+) -> TetrisMetricsProWithPPSADPM: ...
+
+
+@overload
+def get_metrics(
+    *,
+    lpm: Number,
+    apm: Number,
+    vs: Number,
+    precision: int = 2,
+) -> TetrisMetricsProWithLPMVS: ...
+
+
+@overload
+def get_metrics(
+    *,
+    lpm: Number,
+    apm: Number,
+    adpm: Number,
+    precision: int = 2,
+) -> TetrisMetricsProWithLPMADPM: ...
+
+
+def get_metrics(  # noqa: PLR0911, PLR0912, PLR0913, C901
+    *,
+    pps: Number | None = None,
+    lpm: Number | None = None,
+    apm: Number | None = None,
+    vs: Number | None = None,
+    adpm: Number | None = None,
+    precision: int = 2,
+) -> (
+    TetrisMetricsBaseWithPPS
+    | TetrisMetricsBaseWithLPM
+    | TetrisMetricsBaseWithVS
+    | TetrisMetricsBaseWithADPM
+    | TetrisMetricsBasicWithPPS
+    | TetrisMetricsBasicWithLPM
+    | TetrisMetricsProWithPPSVS
+    | TetrisMetricsProWithLPMVS
+    | TetrisMetricsProWithPPSADPM
+    | TetrisMetricsProWithLPMADPM
+):
+    if apm is None:
+        if pps is not None:
+            return TetrisMetricsBaseWithPPS(pps, precision=precision)
+        if lpm is not None:
+            return TetrisMetricsBaseWithLPM(lpm, precision=precision)
+        if vs is not None:
+            return TetrisMetricsBaseWithVS(vs, precision=precision)
+        if adpm is not None:
+            return TetrisMetricsBaseWithADPM(adpm, precision=precision)
+    elif vs is None and adpm is None:
+        if pps is not None:
+            return TetrisMetricsBasicWithPPS(pps, apm, precision=precision)
+        if lpm is not None:
+            return TetrisMetricsBasicWithLPM(lpm, apm, precision=precision)
+    else:
+        if vs is not None:
+            if pps is not None:
+                return TetrisMetricsProWithPPSVS(pps, apm, vs, precision=precision)
+            if lpm is not None:
+                return TetrisMetricsProWithLPMVS(lpm, apm, vs, precision=precision)
+        if adpm is not None:
+            if pps is not None:
+                return TetrisMetricsProWithPPSADPM(pps, apm, adpm, precision=precision)
+            if lpm is not None:
+                return TetrisMetricsProWithLPMADPM(lpm, apm, adpm, precision=precision)
+
+    raise TypeError

nonebot_plugin_tetris_stats/utils/render/__init__.py

@@ -0,0 +1,67 @@
+from typing import Literal, overload
+
+from jinja2 import Environment, FileSystemLoader
+from nonebot.compat import PYDANTIC_V2
+
+from ..templates import templates_dir
+from .schemas.bind import Bind
+from .schemas.tetrio.tetrio_info import Info as TETRIOInfo
+from .schemas.tetrio.tetrio_record_blitz import Record as TETRIORecordBlitz
+from .schemas.tetrio.tetrio_record_sprint import Record as TETRIORecordSprint
+from .schemas.tetrio.tetrio_user_info_v2 import Info as TETRIOUserInfoV2
+from .schemas.top_info import Info as TOPInfo
+from .schemas.tos_info import Info as TOSInfo
+
+env = Environment(
+    loader=FileSystemLoader(templates_dir), autoescape=True, trim_blocks=True, lstrip_blocks=True, enable_async=True
+)
+
+
+@overload
+async def render(render_type: Literal['v1/binding'], data: Bind) -> str: ...
+
+
+@overload
+async def render(render_type: Literal['v1/tetrio/info'], data: TETRIOInfo) -> str: ...
+
+
+@overload
+async def render(render_type: Literal['v1/top/info'], data: TOPInfo) -> str: ...
+
+
+@overload
+async def render(render_type: Literal['v1/tos/info'], data: TOSInfo) -> str: ...
+
+
+@overload
+async def render(render_type: Literal['v2/tetrio/user/info'], data: TETRIOUserInfoV2) -> str: ...
+
+
+@overload
+async def render(render_type: Literal['v2/tetrio/record/40l'], data: TETRIORecordSprint) -> str: ...
+
+
+@overload
+async def render(render_type: Literal['v2/tetrio/record/blitz'], data: TETRIORecordBlitz) -> str: ...
+
+
+async def render(
+    render_type: Literal[
+        'v1/binding',
+        'v1/tetrio/info',
+        'v1/top/info',
+        'v1/tos/info',
+        'v2/tetrio/user/info',
+        'v2/tetrio/record/40l',
+        'v2/tetrio/record/blitz',
+    ],
+    data: Bind | TETRIOInfo | TOPInfo | TOSInfo | TETRIOUserInfoV2 | TETRIORecordSprint | TETRIORecordBlitz,
+) -> str:
+    if PYDANTIC_V2:
+        return await env.get_template('index.html').render_async(
+            path=render_type, data=data.model_dump_json(by_alias=True)
+        )
+    return await env.get_template('index.html').render_async(path=render_type, data=data.json(by_alias=True))
+
+
+__all__ = ['render']

nonebot_plugin_tetris_stats/utils/render/schemas/base.py

@@ -0,0 +1,20 @@
+from typing import Literal
+
+from pydantic import BaseModel
+
+from ...typing import Number
+
+
+class Avatar(BaseModel):
+    type: Literal['identicon']
+    hash: str
+
+
+class People(BaseModel):
+    avatar: str | Avatar
+    name: str
+
+
+class Ranking(BaseModel):
+    rating: Number
+    rd: Number

nonebot_plugin_tetris_stats/utils/render/schemas/bind.py

@@ -0,0 +1,13 @@
+from typing import Literal
+
+from pydantic import BaseModel
+
+from .base import People
+
+
+class Bind(BaseModel):
+    platform: Literal['TETR.IO', 'TOP', 'TOS']
+    status: Literal['error', 'success', 'unknown', 'unlink', 'unverified']
+    user: People
+    bot: People
+    command: str

nonebot_plugin_tetris_stats/utils/render/schemas/tetrio/base.py

@@ -0,0 +1,10 @@
+from datetime import datetime
+
+from pydantic import BaseModel
+
+from ....typing import Number
+
+
+class TetraLeagueHistoryData(BaseModel):
+    record_at: datetime
+    tr: Number

nonebot_plugin_tetris_stats/utils/render/schemas/tetrio/tetrio_info.py

@@ -0,0 +1,49 @@
+from pydantic import BaseModel
+
+from .....games.tetrio.api.typing import Rank
+from ....typing import Number
+from ..base import People, Ranking
+from .base import TetraLeagueHistoryData
+
+
+class User(People):
+    bio: str | None
+
+
+class TetraLeague(BaseModel):
+    rank: Rank
+    tr: Number
+    global_rank: Number
+    pps: Number
+    lpm: Number
+    apm: Number
+    apl: Number
+    vs: Number
+    adpm: Number
+    adpl: Number
+
+
+class TetraLeagueHistory(BaseModel):
+    data: list[TetraLeagueHistoryData]
+    split_interval: Number
+    min_tr: Number
+    max_tr: Number
+    offset: Number
+
+
+class Radar(BaseModel):
+    app: Number
+    dsps: Number
+    dspp: Number
+    ci: Number
+    ge: Number
+
+
+class Info(BaseModel):
+    user: User
+    ranking: Ranking
+    tetra_league: TetraLeague
+    tetra_league_history: TetraLeagueHistory
+    radar: Radar
+    sprint: str
+    blitz: str

nonebot_plugin_tetris_stats/utils/render/schemas/tetrio/tetrio_record_base.py

@@ -0,0 +1,58 @@
+from pydantic import BaseModel
+
+from ..base import People
+
+
+class User(People):
+    id: str
+
+
+class Max(BaseModel):
+    combo: int
+    btb: int
+
+
+class Mini(BaseModel):
+    total: int
+    single: int
+    double: int
+
+
+class Tspins(BaseModel):
+    total: int
+    single: int
+    double: int
+    triple: int
+
+    mini: Mini
+
+
+class Finesse(BaseModel):
+    faults: int
+    accuracy: float
+
+
+class RecordStatistic(BaseModel):
+    keys: int
+    kpp: float
+    kps: float
+
+    max: Max
+
+    pieces: int
+    pps: float
+    lines: int
+    lpm: float
+    holds: int | None
+    score: int
+
+    single: int
+    double: int
+    triple: int
+    quad: int
+
+    tspins: Tspins
+
+    all_clear: int
+
+    finesse: Finesse

nonebot_plugin_tetris_stats/utils/render/schemas/tetrio/tetrio_record_blitz.py

@@ -0,0 +1,22 @@
+from datetime import datetime
+
+from pydantic import BaseModel
+
+from .tetrio_record_base import RecordStatistic, User
+
+
+class Statistic(RecordStatistic):
+    spp: float
+
+    level: int
+
+
+class Record(BaseModel):
+    user: User
+
+    replay_id: str
+    rank: int | None
+
+    statistic: Statistic
+
+    play_at: datetime

nonebot_plugin_tetris_stats/utils/render/schemas/tetrio/tetrio_record_sprint.py

@@ -0,0 +1,18 @@
+from datetime import datetime
+
+from pydantic import BaseModel
+
+from .tetrio_record_base import RecordStatistic as Statistic
+from .tetrio_record_base import User
+
+
+class Record(BaseModel):
+    user: User
+
+    time: str
+    replay_id: str
+    rank: int | None
+
+    statistic: Statistic
+
+    play_at: datetime

nonebot_plugin_tetris_stats/utils/render/schemas/tetrio/tetrio_user_info_v2.py

@@ -0,0 +1,97 @@
+from datetime import datetime
+
+from pydantic import BaseModel
+
+from .....games.tetrio.api.schemas.user_records import Zen
+from .....games.tetrio.api.typing import Rank
+from ....typing import Number
+from ..base import Avatar
+from .base import TetraLeagueHistoryData
+
+
+class Badge(BaseModel):
+    id: str
+    description: str
+    group: str | None
+    receive_at: datetime | None
+
+
+class User(BaseModel):
+    id: str
+    name: str
+    country: str | None
+
+    avatar: str | Avatar
+    banner: str | None
+
+    bio: str | None
+
+    friend_count: int
+    supporter_tier: int
+
+    verified: bool
+    bad_standing: bool
+
+    badges: list[Badge]
+    xp: Number
+
+    playtime: str | None
+    join_at: datetime | None
+
+
+class Statistic(BaseModel):
+    total: int | None
+    wins: int | None
+
+
+class TetraLeagueStatistic(BaseModel):
+    total: int
+    wins: int
+
+
+class TetraLeague(BaseModel):
+    rank: Rank
+    highest_rank: Rank
+
+    tr: Number
+
+    glicko: Number
+    rd: Number
+
+    global_rank: int | None
+    country_rank: int | None
+
+    pps: Number
+
+    apm: Number
+    apl: Number
+
+    vs: Number | None
+    adpl: Number | None
+
+    statistic: TetraLeagueStatistic
+
+    decaying: bool
+
+    history: list[TetraLeagueHistoryData]
+
+
+class Sprint(BaseModel):
+    time: str
+    global_rank: int | None
+    play_at: datetime
+
+
+class Blitz(BaseModel):
+    score: int
+    global_rank: int | None
+    play_at: datetime
+
+
+class Info(BaseModel):
+    user: User
+    tetra_league: TetraLeague | None
+    statistic: Statistic | None
+    sprint: Sprint | None
+    blitz: Blitz | None
+    zen: Zen

nonebot_plugin_tetris_stats/utils/render/schemas/top_info.py

@@ -0,0 +1,17 @@
+from pydantic import BaseModel
+
+from ...typing import Number
+from .base import People
+
+
+class Data(BaseModel):
+    pps: Number
+    lpm: Number
+    apm: Number
+    apl: Number
+
+
+class Info(BaseModel):
+    user: People
+    today: Data
+    history: Data

nonebot_plugin_tetris_stats/utils/render/schemas/tos_info.py

@@ -0,0 +1,32 @@
+from pydantic import BaseModel, Field
+
+from ...typing import Number
+from .base import People, Ranking
+
+
+class Multiplayer(BaseModel):
+    pps: Number
+    lpm: Number
+    apm: Number
+    apl: Number
+    vs: Number
+    adpm: Number
+    adpl: Number
+
+
+class Radar(BaseModel):
+    app: Number
+    OR: Number = Field(serialization_alias='or')
+    dspp: Number
+    ci: Number
+    ge: Number
+
+
+class Info(BaseModel):
+    user: People
+    ranking: Ranking
+    multiplayer: Multiplayer
+    radar: Radar
+    sprint: str
+    challenge: str
+    marathon: str

nonebot_plugin_tetris_stats/utils/request.py

@@ -0,0 +1,167 @@
+from collections.abc import Sequence
+from http import HTTPStatus
+from urllib.parse import urljoin, urlparse
+
+from aiofiles import open
+from httpx import AsyncClient, HTTPError
+from nonebot import get_driver, get_plugin_config
+from nonebot.log import logger
+from playwright.async_api import Response
+from ujson import JSONDecodeError, dumps, loads
+
+from ..config.config import CACHE_PATH, Config
+from .browser import BrowserManager
+from .exception import RequestError
+
+driver = get_driver()
+config = get_plugin_config(Config)
+
+
+@driver.on_startup
+async def _():
+    await Request.init_cache()
+    await Request.read_cache()
+
+
+@driver.on_shutdown
+async def _():
+    await Request.write_cache()
+
+
+def splice_url(url_list: list[str]) -> str:
+    url = ''
+    if len(url_list):
+        url = url_list.pop(0)
+        for i in url_list:
+            url = urljoin(url, i)
+    return url
+
+
+class Request:
+    """网络请求相关类"""
+
+    _CACHE_FILE = CACHE_PATH / 'cloudflare_cache.json'
+    _headers: dict | None = None
+    _cookies: dict | None = None
+
+    @classmethod
+    async def _anti_cloudflare(cls, url: str) -> bytes:
+        """用firefox硬穿五秒盾"""
+        browser = await BrowserManager.get_browser()
+        async with await browser.new_context() as context, await context.new_page() as page:
+            response = await page.goto(url)
+            attempts = 0
+            while attempts < 60:  # noqa: PLR2004
+                attempts += 1
+                text = await page.locator('body').text_content()
+                if text is None:
+                    await page.wait_for_timeout(1000)
+                    continue
+                if await page.title() == 'Please Wait... | Cloudflare':
+                    logger.warning('疑似触发了 Cloudflare 的验证码')
+                    break
+                try:
+                    loads(text)
+                except JSONDecodeError:
+                    await page.wait_for_timeout(1000)
+                else:
+                    if not isinstance(response, Response):
+                        msg = 'api请求失败'
+                        raise RequestError(msg)
+                    cls._headers = await response.request.all_headers()
+                    try:
+                        cls._cookies = {
+                            name: value
+                            for i in await context.cookies()
+                            if (name := i.get('name')) is not None and (value := i.get('value')) is not None
+                        }
+                    except KeyError:
+                        cls._cookies = None
+                    return await response.body()
+        msg = '绕过五秒盾失败'
+        raise RequestError(msg)
+
+    @classmethod
+    async def init_cache(cls) -> None:
+        """初始化缓存文件"""
+        if not cls._CACHE_FILE.exists():
+            async with open(file=cls._CACHE_FILE, mode='w', encoding='UTF-8') as file:
+                await file.write(dumps({'headers': cls._headers, 'cookies': cls._cookies}))
+
+    @classmethod
+    async def read_cache(cls) -> None:
+        """读取缓存文件"""
+        try:
+            async with open(file=cls._CACHE_FILE, mode='r', encoding='UTF-8') as file:
+                json = loads(await file.read())
+        except FileNotFoundError:
+            await cls.init_cache()
+        except (PermissionError, JSONDecodeError):
+            cls._CACHE_FILE.unlink()
+            await cls.init_cache()
+        else:
+            cls._headers = json['headers']
+            cls._cookies = json['cookies']
+
+    @classmethod
+    async def write_cache(cls) -> None:
+        """写入缓存文件"""
+        try:
+            async with open(file=cls._CACHE_FILE, mode='r+', encoding='UTF-8') as file:
+                await file.write(dumps({'headers': cls._headers, 'cookies': cls._cookies}))
+        except FileNotFoundError:
+            await cls.init_cache()
+        except (PermissionError, JSONDecodeError):
+            cls._CACHE_FILE.unlink()
+            await cls.init_cache()
+
+    @classmethod
+    async def request(cls, url: str, *, is_json: bool = True) -> bytes:
+        """请求api"""
+        try:
+            async with AsyncClient(cookies=cls._cookies, timeout=config.tetris_req_timeout) as session:
+                response = await session.get(url, headers=cls._headers)
+                if response.status_code != HTTPStatus.OK:
+                    msg = f'请求错误 code: {response.status_code} {HTTPStatus(response.status_code).phrase}\n{response.text}'
+                    raise RequestError(msg, status_code=response.status_code)
+                if is_json:
+                    loads(response.content)
+                return response.content
+        except HTTPError as e:
+            msg = f'请求错误 \n{e!r}'
+            raise RequestError(msg) from e
+        except JSONDecodeError:
+            if urlparse(url).netloc.lower().endswith('tetr.io'):
+                return await cls._anti_cloudflare(url)
+            raise
+
+    @classmethod
+    async def failover_request(
+        cls,
+        urls: Sequence[str],
+        *,
+        failover_code: Sequence[int],
+        failover_exc: tuple[type[BaseException], ...],
+        is_json: bool = True,
+    ) -> bytes:
+        error_list: list[RequestError] = []
+        for i in urls:
+            logger.debug(f'尝试请求 {i}')
+            try:
+                return await cls.request(i, is_json=is_json)
+            except RequestError as e:
+                if e.status_code in failover_code:  # 如果状态码在 failover_code 中, 则继续尝试下一个URL
+                    error_list.append(e)
+                    continue
+                # 如果状态码不在故障转移列表中, 则查找异常栈, 如果异常栈内有 failover_exc 内的异常类型, 则继续尝试下一个URL
+                tb = e.__traceback__
+                while tb is not None:
+                    if isinstance(tb.tb_frame.f_locals.get('exc_value'), failover_exc):
+                        error_list.append(e)
+                        break
+                    tb = tb.tb_next
+                else:
+                    raise
+                continue
+        msg = f'所有地址皆不可用\n{error_list!r}'
+        raise RequestError(msg)

nonebot_plugin_tetris_stats/utils/retry.py

@@ -0,0 +1,44 @@
+from asyncio import sleep
+from collections.abc import Callable, Coroutine
+from contextlib import suppress
+from datetime import timedelta
+from functools import wraps
+from typing import Any, ParamSpec, TypeVar
+
+from nonebot.log import logger
+from nonebot_plugin_alconna.uniseg import SerializeFailed, UniMessage
+
+T = TypeVar('T')
+P = ParamSpec('P')
+
+
+def retry(
+    max_attempts: int = 3,
+    exception_type: type[BaseException] | tuple[type[BaseException], ...] = Exception,
+    delay: timedelta | None = None,
+    reply: str | UniMessage | None = None,
+) -> Callable[[Callable[P, Coroutine[Any, Any, T]]], Callable[P, Coroutine[Any, Any, T]]]:
+    def decorator(func: Callable[P, Coroutine[Any, Any, T]]) -> Callable[P, Coroutine[Any, Any, T]]:
+        @wraps(func)
+        async def wrapper(*args: P.args, **kwargs: P.kwargs) -> T:
+            for i in range(max_attempts + 1):
+                if i > 0:
+                    message = f'Retrying: {func.__name__} ({i}/{max_attempts})'
+                    logger.debug(message)
+                    with suppress(SerializeFailed):
+                        await UniMessage(reply or message).send()
+                if i == max_attempts:
+                    break
+                try:
+                    return await func(*args, **kwargs)
+                except exception_type as e:
+                    if i == max_attempts:
+                        raise
+                    logger.exception(e)
+                    if delay is not None:
+                        await sleep(delay.total_seconds())
+            return await func(*args, **kwargs)
+
+        return wrapper
+
+    return decorator

nonebot_plugin_tetris_stats/utils/screenshot.py

@@ -0,0 +1,15 @@
+from playwright.async_api import TimeoutError
+
+from .browser import BrowserManager
+from .retry import retry
+
+
+@retry(exception_type=TimeoutError, reply='截图失败, 重试中')
+async def screenshot(url: str) -> bytes:
+    browser = await BrowserManager.get_browser()
+    async with (
+        await browser.new_page(viewport={'width': 3000, 'height': 3000}) as page,
+    ):
+        await page.goto(url)
+        await page.wait_for_load_state('networkidle')
+        return await page.locator('id=content').screenshot(timeout=5000, type='png')

nonebot_plugin_tetris_stats/utils/templates.py

@@ -0,0 +1,71 @@
+from asyncio.subprocess import PIPE, create_subprocess_exec
+from shutil import rmtree
+
+from nonebot import get_driver
+from nonebot.log import logger
+from nonebot.permission import SUPERUSER
+from nonebot_plugin_alconna import on_alconna
+from nonebot_plugin_localstore import get_data_dir  # type: ignore[import-untyped]
+
+driver = get_driver()
+
+templates_dir = get_data_dir('nonebot_plugin_tetris_stats') / 'templates'
+
+alc = on_alconna('更新模板', permission=SUPERUSER)
+
+
+@driver.on_startup
+async def init_templates() -> None:
+    try:
+        await create_subprocess_exec('git', '--version', stdout=PIPE)
+    except FileNotFoundError as e:
+        msg = '未找到 git, 请确保 git 已安装并在环境变量中\n安装步骤请参阅: https://git-scm.com/book/zh/v2/%E8%B5%B7%E6%AD%A5-%E5%AE%89%E8%A3%85-Git'
+        raise RuntimeError(msg) from e
+    if not templates_dir.exists():
+        logger.info('模板仓库不存在, 正在尝试初始化...')
+        proc = await create_subprocess_exec(
+            'git',
+            'clone',
+            '-b',
+            'gh-pages',
+            'https://github.com/A-Minos/tetris-stats-templates',
+            templates_dir,
+            '--depth=1',
+            stdout=PIPE,
+            stderr=PIPE,
+        )
+        stdout, stderr = await proc.communicate()
+        if proc.returncode != 0:
+            for i in stderr.decode().splitlines():
+                logger.error(i)
+            msg = '初始化模板仓库失败'
+            raise RuntimeError(msg)
+        logger.success('模板仓库初始化成功')
+        return
+    proc = await create_subprocess_exec(
+        'git', 'rev-parse', '--is-inside-work-tree', stdout=PIPE, stderr=PIPE, cwd=templates_dir
+    )
+    stdout, stderr = await proc.communicate()
+    if proc.returncode != 0:
+        for i in stderr.decode().splitlines():
+            logger.error(i)
+        logger.warning('模板仓库状态异常, 尝试重新初始化')
+        rmtree(templates_dir)
+        await init_templates()
+        return
+    logger.info('正在更新模板仓库...')
+    proc = await create_subprocess_exec('git', 'pull', stdout=PIPE, stderr=PIPE, cwd=templates_dir)
+    stdout, stderr = await proc.communicate()
+    logger.info(stdout.decode().strip())
+    if proc.returncode != 0:
+        for i in stderr.decode().splitlines():
+            logger.error(i)
+        msg = '更新模板仓库失败'
+        raise RuntimeError(msg)
+    logger.success('模板仓库更新成功')
+
+
+@alc.handle()
+async def _():
+    await init_templates()
+    await alc.finish('模板仓库更新成功')

nonebot_plugin_tetris_stats/utils/typing.py

@@ -0,0 +1,61 @@
+from collections.abc import Awaitable, Callable
+from typing import Any, Literal
+
+Number = float | int
+GameType = Literal['IO', 'TOP', 'TOS']
+CommandType = Literal['bind', 'query']
+AsyncCallable = Callable[..., Awaitable[Any]]
+Me = Literal[
+    '我',
+    '自己',
+    '我等',
+    '卑人',
+    '愚',
+    '老身',
+    '爷',
+    '老娘',
+    '本姑娘',
+    '本大爷',
+    '鄙人',
+    '寡人',
+    '小生',
+    '贫僧',
+    '本人',
+    '孤',
+    '吾',
+    '俺',
+    '咱',
+    '私',
+    'me',
+    '洒家',
+    '在下',
+    '偶',
+    '人家',
+    '本小姐',
+    '老夫',
+    '老子',
+    '朕',
+    '本尊',
+    '僕',
+    '拙者',
+    '妾',
+    '儂',
+    '自分',
+    '吾輩',
+    '我輩',
+    '某',
+    '己等',
+    '俺等',
+    '此方',
+    '哥',
+    '姐',
+    '劳资',
+    '本宝宝',
+    '余',
+    '本喵',
+    'watashi',
+    'i',
+    'myself',
+    'self',
+    'oneself',
+]