feat: publish sanitized Linux cloud-game MAA adapter
This commit is contained in:
+201
@@ -0,0 +1,201 @@
|
||||
Apache License
|
||||
Version 2.0, January 2004
|
||||
http://www.apache.org/licenses/
|
||||
|
||||
TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
|
||||
|
||||
1. Definitions.
|
||||
|
||||
"License" shall mean the terms and conditions for use, reproduction,
|
||||
and distribution as defined by Sections 1 through 9 of this document.
|
||||
|
||||
"Licensor" shall mean the copyright owner or entity authorized by
|
||||
the copyright owner that is granting the License.
|
||||
|
||||
"Legal Entity" shall mean the union of the acting entity and all
|
||||
other entities that control, are controlled by, or are under common
|
||||
control with that entity. For the purposes of this definition,
|
||||
"control" means (i) the power, direct or indirect, to cause the
|
||||
direction or management of such entity, whether by contract or
|
||||
otherwise, or (ii) ownership of fifty percent (50%) or more of the
|
||||
outstanding shares, or (iii) beneficial ownership of such entity.
|
||||
|
||||
"You" (or "Your") shall mean an individual or Legal Entity
|
||||
exercising permissions granted by this License.
|
||||
|
||||
"Source" form shall mean the preferred form for making modifications,
|
||||
including but not limited to software source code, documentation
|
||||
source, and configuration files.
|
||||
|
||||
"Object" form shall mean any form resulting from mechanical
|
||||
transformation or translation of a Source form, including but
|
||||
not limited to compiled object code, generated documentation,
|
||||
and conversions to other media types.
|
||||
|
||||
"Work" shall mean the work of authorship, whether in Source or
|
||||
Object form, made available under the License, as indicated by a
|
||||
copyright notice that is included in or attached to the work
|
||||
(an example is provided in the Appendix below).
|
||||
|
||||
"Derivative Works" shall mean any work, whether in Source or Object
|
||||
form, that is based on (or derived from) the Work and for which the
|
||||
editorial revisions, annotations, elaborations, or other modifications
|
||||
represent, as a whole, an original work of authorship. For the purposes
|
||||
of this License, Derivative Works shall not include works that remain
|
||||
separable from, or merely link (or bind by name) to the interfaces of,
|
||||
the Work and Derivative Works thereof.
|
||||
|
||||
"Contribution" shall mean any work of authorship, including
|
||||
the original version of the Work and any modifications or additions
|
||||
to that Work or Derivative Works thereof, that is intentionally
|
||||
submitted to Licensor for inclusion in the Work by the copyright owner
|
||||
or by an individual or Legal Entity authorized to submit on behalf of
|
||||
the copyright owner. For the purposes of this definition, "submitted"
|
||||
means any form of electronic, verbal, or written communication sent
|
||||
to the Licensor or its representatives, including but not limited to
|
||||
communication on electronic mailing lists, source code control systems,
|
||||
and issue tracking systems that are managed by, or on behalf of, the
|
||||
Licensor for the purpose of discussing and improving the Work, but
|
||||
excluding communication that is conspicuously marked or otherwise
|
||||
designated in writing by the copyright owner as "Not a Contribution."
|
||||
|
||||
"Contributor" shall mean Licensor and any individual or Legal Entity
|
||||
on behalf of whom a Contribution has been received by Licensor and
|
||||
subsequently incorporated within the Work.
|
||||
|
||||
2. Grant of Copyright License. Subject to the terms and conditions of
|
||||
this License, each Contributor hereby grants to You a perpetual,
|
||||
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||
copyright license to reproduce, prepare Derivative Works of,
|
||||
publicly display, publicly perform, sublicense, and distribute the
|
||||
Work and such Derivative Works in Source or Object form.
|
||||
|
||||
3. Grant of Patent License. Subject to the terms and conditions of
|
||||
this License, each Contributor hereby grants to You a perpetual,
|
||||
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||
(except as stated in this section) patent license to make, have made,
|
||||
use, offer to sell, sell, import, and otherwise transfer the Work,
|
||||
where such license applies only to those patent claims licensable
|
||||
by such Contributor that are necessarily infringed by their
|
||||
Contribution(s) alone or by combination of their Contribution(s)
|
||||
with the Work to which such Contribution(s) was submitted. If You
|
||||
institute patent litigation against any entity (including a
|
||||
cross-claim or counterclaim in a lawsuit) alleging that the Work
|
||||
or a Contribution incorporated within the Work constitutes direct
|
||||
or contributory patent infringement, then any patent licenses
|
||||
granted to You under this License for that Work shall terminate
|
||||
as of the date such litigation is filed.
|
||||
|
||||
4. Redistribution. You may reproduce and distribute copies of the
|
||||
Work or Derivative Works thereof in any medium, with or without
|
||||
modifications, and in Source or Object form, provided that You
|
||||
meet the following conditions:
|
||||
|
||||
(a) You must give any other recipients of the Work or
|
||||
Derivative Works a copy of this License; and
|
||||
|
||||
(b) You must cause any modified files to carry prominent notices
|
||||
stating that You changed the files; and
|
||||
|
||||
(c) You must retain, in the Source form of any Derivative Works
|
||||
that You distribute, all copyright, patent, trademark, and
|
||||
attribution notices from the Source form of the Work,
|
||||
excluding those notices that do not pertain to any part of
|
||||
the Derivative Works; and
|
||||
|
||||
(d) If the Work includes a "NOTICE" text file as part of its
|
||||
distribution, then any Derivative Works that You distribute must
|
||||
include a readable copy of the attribution notices contained
|
||||
within such NOTICE file, excluding those notices that do not
|
||||
pertain to any part of the Derivative Works, in at least one
|
||||
of the following places: within a NOTICE text file distributed
|
||||
as part of the Derivative Works; within the Source form or
|
||||
documentation, if provided along with the Derivative Works; or,
|
||||
within a display generated by the Derivative Works, if and
|
||||
wherever such third-party notices normally appear. The contents
|
||||
of the NOTICE file are for informational purposes only and
|
||||
do not modify the License. You may add Your own attribution
|
||||
notices within Derivative Works that You distribute, alongside
|
||||
or as an addendum to the NOTICE text from the Work, provided
|
||||
that such additional attribution notices cannot be construed
|
||||
as modifying the License.
|
||||
|
||||
You may add Your own copyright statement to Your modifications and
|
||||
may provide additional or different license terms and conditions
|
||||
for use, reproduction, or distribution of Your modifications, or
|
||||
for any such Derivative Works as a whole, provided Your use,
|
||||
reproduction, and distribution of the Work otherwise complies with
|
||||
the conditions stated in this License.
|
||||
|
||||
5. Submission of Contributions. Unless You explicitly state otherwise,
|
||||
any Contribution intentionally submitted for inclusion in the Work
|
||||
by You to the Licensor shall be under the terms and conditions of
|
||||
this License, without any additional terms or conditions.
|
||||
Notwithstanding the above, nothing herein shall supersede or modify
|
||||
the terms of any separate license agreement you may have executed
|
||||
with Licensor regarding such Contributions.
|
||||
|
||||
6. Trademarks. This License does not grant permission to use the trade
|
||||
names, trademarks, service marks, or product names of the Licensor,
|
||||
except as required for reasonable and customary use in describing the
|
||||
origin of the Work and reproducing the content of the NOTICE file.
|
||||
|
||||
7. Disclaimer of Warranty. Unless required by applicable law or
|
||||
agreed to in writing, Licensor provides the Work (and each
|
||||
Contributor provides its Contributions) on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
||||
implied, including, without limitation, any warranties or conditions
|
||||
of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
|
||||
PARTICULAR PURPOSE. You are solely responsible for determining the
|
||||
appropriateness of using or redistributing the Work and assume any
|
||||
risks associated with Your exercise of permissions under this License.
|
||||
|
||||
8. Limitation of Liability. In no event and under no legal theory,
|
||||
whether in tort (including negligence), contract, or otherwise,
|
||||
unless required by applicable law (such as deliberate and grossly
|
||||
negligent acts) or agreed to in writing, shall any Contributor be
|
||||
liable to You for damages, including any direct, indirect, special,
|
||||
incidental, or consequential damages of any character arising as a
|
||||
result of this License or out of the use or inability to use the
|
||||
Work (including but not limited to damages for loss of goodwill,
|
||||
work stoppage, computer failure or malfunction, or any and all
|
||||
other commercial damages or losses), even if such Contributor
|
||||
has been advised of the possibility of such damages.
|
||||
|
||||
9. Accepting Warranty or Additional Liability. While redistributing
|
||||
the Work or Derivative Works thereof, You may choose to offer,
|
||||
and charge a fee for, acceptance of support, warranty, indemnity,
|
||||
or other liability obligations and/or rights consistent with this
|
||||
License. However, in accepting such obligations, You may act only
|
||||
on Your own behalf and on Your sole responsibility, not on behalf
|
||||
of any other Contributor, and only if You agree to indemnify,
|
||||
defend, and hold each Contributor harmless for any liability
|
||||
incurred by, or claims asserted against, such Contributor by reason
|
||||
of your accepting any such warranty or additional liability.
|
||||
|
||||
END OF TERMS AND CONDITIONS
|
||||
|
||||
APPENDIX: How to apply the Apache License to your work.
|
||||
|
||||
To apply the Apache License to your work, attach the following
|
||||
boilerplate notice, with the fields enclosed by brackets "[]"
|
||||
replaced with your own identifying information. (Don't include
|
||||
the brackets!) The text should be enclosed in the appropriate
|
||||
comment syntax for the file format. We also recommend that a
|
||||
file or class name and description of purpose be included on the
|
||||
same "printed page" as the copyright notice for easier
|
||||
identification within third-party archives.
|
||||
|
||||
Copyright [yyyy] [name of copyright owner]
|
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at
|
||||
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
||||
Unless required by applicable law or agreed to in writing, software
|
||||
distributed under the License is distributed on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
Executable
+107
@@ -0,0 +1,107 @@
|
||||
#!/usr/bin/env python3
|
||||
"""Bootstrap a NetEase Cloud Game token without blocking the API server."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import argparse
|
||||
import json
|
||||
import os
|
||||
from pathlib import Path
|
||||
|
||||
import requests
|
||||
|
||||
BASE_DIR = Path(__file__).resolve().parent
|
||||
TOKEN_FILE = Path(os.getenv("MAA_ONLINE_TOKEN_FILE", str(BASE_DIR / "token")))
|
||||
PENDING_FILE = Path(os.getenv("MAA_ONLINE_PENDING_LOGIN_FILE", str(BASE_DIR / ".pending-login")))
|
||||
API_BASE = "https://n.cg.163.com/api/v1"
|
||||
|
||||
|
||||
def write_private(path: Path, value: str) -> None:
|
||||
path.parent.mkdir(parents=True, exist_ok=True)
|
||||
flags = os.O_WRONLY | os.O_CREAT | os.O_TRUNC
|
||||
fd = os.open(path, flags, 0o600)
|
||||
try:
|
||||
os.write(fd, value.encode("utf-8"))
|
||||
finally:
|
||||
os.close(fd)
|
||||
os.chmod(path, 0o600)
|
||||
|
||||
|
||||
def normalize_phone(phone: str, country_code: str) -> tuple[str, str]:
|
||||
phone = phone.strip().replace(" ", "")
|
||||
country_code = country_code.strip().lstrip("+")
|
||||
if not phone.isdigit() or not country_code.isdigit():
|
||||
raise ValueError("phone and country code must contain digits only")
|
||||
return country_code, phone
|
||||
|
||||
|
||||
def request_code(phone: str, country_code: str) -> None:
|
||||
country_code, phone = normalize_phone(phone, country_code)
|
||||
response = requests.post(
|
||||
f"{API_BASE}/phone-captchas/{country_code}-{phone}",
|
||||
timeout=15,
|
||||
)
|
||||
response.raise_for_status()
|
||||
write_private(PENDING_FILE, json.dumps({"country_code": country_code, "phone": phone}))
|
||||
print("SMS code requested. Run: python login.py verify <code>")
|
||||
|
||||
|
||||
def verify_code(code: str) -> None:
|
||||
if not PENDING_FILE.is_file():
|
||||
raise RuntimeError("no pending login; run the request command first")
|
||||
pending = json.loads(PENDING_FILE.read_text(encoding="utf-8"))
|
||||
code = code.strip()
|
||||
if not code:
|
||||
raise ValueError("verification code must not be empty")
|
||||
|
||||
payload = {
|
||||
"auth_method": "phone-captcha",
|
||||
"ctcode": pending["country_code"],
|
||||
"phone": pending["phone"],
|
||||
"captcha": code,
|
||||
"device_info": {
|
||||
"userAgent": "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 Chrome/140 Safari/537.36",
|
||||
"appVersion": "5.0 (X11)",
|
||||
"codecs": ["h264", "vp8", "vp9"],
|
||||
},
|
||||
}
|
||||
response = requests.post(f"{API_BASE}/tokens", json=payload, timeout=15)
|
||||
response.raise_for_status()
|
||||
token = response.json().get("token")
|
||||
if not token:
|
||||
raise RuntimeError("login response did not contain a token")
|
||||
|
||||
write_private(TOKEN_FILE, token)
|
||||
PENDING_FILE.unlink(missing_ok=True)
|
||||
print(f"Login token saved securely to {TOKEN_FILE}")
|
||||
|
||||
|
||||
def status() -> None:
|
||||
if TOKEN_FILE.is_file() and TOKEN_FILE.stat().st_size:
|
||||
print(f"token: present ({oct(TOKEN_FILE.stat().st_mode & 0o777)})")
|
||||
else:
|
||||
print("token: missing")
|
||||
print(f"pending login: {'yes' if PENDING_FILE.is_file() else 'no'}")
|
||||
|
||||
|
||||
def main() -> None:
|
||||
parser = argparse.ArgumentParser()
|
||||
sub = parser.add_subparsers(dest="command", required=True)
|
||||
request_parser = sub.add_parser("request", help="request an SMS verification code")
|
||||
request_parser.add_argument("phone")
|
||||
request_parser.add_argument("--country-code", default="86")
|
||||
verify_parser = sub.add_parser("verify", help="exchange an SMS code for a token")
|
||||
verify_parser.add_argument("code")
|
||||
sub.add_parser("status", help="show credential state without revealing secrets")
|
||||
args = parser.parse_args()
|
||||
|
||||
if args.command == "request":
|
||||
request_code(args.phone, args.country_code)
|
||||
elif args.command == "verify":
|
||||
verify_code(args.code)
|
||||
else:
|
||||
status()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
@@ -0,0 +1,7 @@
|
||||
aiohttp>=3.10,<4
|
||||
requests>=2.31,<3
|
||||
aiortc>=1.9,<2
|
||||
av>=12.3,<17
|
||||
numpy>=1.26,<3
|
||||
pillow>=10.3,<13
|
||||
websockets>=12,<16
|
||||
@@ -0,0 +1,201 @@
|
||||
Apache License
|
||||
Version 2.0, January 2004
|
||||
http://www.apache.org/licenses/
|
||||
|
||||
TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
|
||||
|
||||
1. Definitions.
|
||||
|
||||
"License" shall mean the terms and conditions for use, reproduction,
|
||||
and distribution as defined by Sections 1 through 9 of this document.
|
||||
|
||||
"Licensor" shall mean the copyright owner or entity authorized by
|
||||
the copyright owner that is granting the License.
|
||||
|
||||
"Legal Entity" shall mean the union of the acting entity and all
|
||||
other entities that control, are controlled by, or are under common
|
||||
control with that entity. For the purposes of this definition,
|
||||
"control" means (i) the power, direct or indirect, to cause the
|
||||
direction or management of such entity, whether by contract or
|
||||
otherwise, or (ii) ownership of fifty percent (50%) or more of the
|
||||
outstanding shares, or (iii) beneficial ownership of such entity.
|
||||
|
||||
"You" (or "Your") shall mean an individual or Legal Entity
|
||||
exercising permissions granted by this License.
|
||||
|
||||
"Source" form shall mean the preferred form for making modifications,
|
||||
including but not limited to software source code, documentation
|
||||
source, and configuration files.
|
||||
|
||||
"Object" form shall mean any form resulting from mechanical
|
||||
transformation or translation of a Source form, including but
|
||||
not limited to compiled object code, generated documentation,
|
||||
and conversions to other media types.
|
||||
|
||||
"Work" shall mean the work of authorship, whether in Source or
|
||||
Object form, made available under the License, as indicated by a
|
||||
copyright notice that is included in or attached to the work
|
||||
(an example is provided in the Appendix below).
|
||||
|
||||
"Derivative Works" shall mean any work, whether in Source or Object
|
||||
form, that is based on (or derived from) the Work and for which the
|
||||
editorial revisions, annotations, elaborations, or other modifications
|
||||
represent, as a whole, an original work of authorship. For the purposes
|
||||
of this License, Derivative Works shall not include works that remain
|
||||
separable from, or merely link (or bind by name) to the interfaces of,
|
||||
the Work and Derivative Works thereof.
|
||||
|
||||
"Contribution" shall mean any work of authorship, including
|
||||
the original version of the Work and any modifications or additions
|
||||
to that Work or Derivative Works thereof, that is intentionally
|
||||
submitted to Licensor for inclusion in the Work by the copyright owner
|
||||
or by an individual or Legal Entity authorized to submit on behalf of
|
||||
the copyright owner. For the purposes of this definition, "submitted"
|
||||
means any form of electronic, verbal, or written communication sent
|
||||
to the Licensor or its representatives, including but not limited to
|
||||
communication on electronic mailing lists, source code control systems,
|
||||
and issue tracking systems that are managed by, or on behalf of, the
|
||||
Licensor for the purpose of discussing and improving the Work, but
|
||||
excluding communication that is conspicuously marked or otherwise
|
||||
designated in writing by the copyright owner as "Not a Contribution."
|
||||
|
||||
"Contributor" shall mean Licensor and any individual or Legal Entity
|
||||
on behalf of whom a Contribution has been received by Licensor and
|
||||
subsequently incorporated within the Work.
|
||||
|
||||
2. Grant of Copyright License. Subject to the terms and conditions of
|
||||
this License, each Contributor hereby grants to You a perpetual,
|
||||
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||
copyright license to reproduce, prepare Derivative Works of,
|
||||
publicly display, publicly perform, sublicense, and distribute the
|
||||
Work and such Derivative Works in Source or Object form.
|
||||
|
||||
3. Grant of Patent License. Subject to the terms and conditions of
|
||||
this License, each Contributor hereby grants to You a perpetual,
|
||||
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||
(except as stated in this section) patent license to make, have made,
|
||||
use, offer to sell, sell, import, and otherwise transfer the Work,
|
||||
where such license applies only to those patent claims licensable
|
||||
by such Contributor that are necessarily infringed by their
|
||||
Contribution(s) alone or by combination of their Contribution(s)
|
||||
with the Work to which such Contribution(s) was submitted. If You
|
||||
institute patent litigation against any entity (including a
|
||||
cross-claim or counterclaim in a lawsuit) alleging that the Work
|
||||
or a Contribution incorporated within the Work constitutes direct
|
||||
or contributory patent infringement, then any patent licenses
|
||||
granted to You under this License for that Work shall terminate
|
||||
as of the date such litigation is filed.
|
||||
|
||||
4. Redistribution. You may reproduce and distribute copies of the
|
||||
Work or Derivative Works thereof in any medium, with or without
|
||||
modifications, and in Source or Object form, provided that You
|
||||
meet the following conditions:
|
||||
|
||||
(a) You must give any other recipients of the Work or
|
||||
Derivative Works a copy of this License; and
|
||||
|
||||
(b) You must cause any modified files to carry prominent notices
|
||||
stating that You changed the files; and
|
||||
|
||||
(c) You must retain, in the Source form of any Derivative Works
|
||||
that You distribute, all copyright, patent, trademark, and
|
||||
attribution notices from the Source form of the Work,
|
||||
excluding those notices that do not pertain to any part of
|
||||
the Derivative Works; and
|
||||
|
||||
(d) If the Work includes a "NOTICE" text file as part of its
|
||||
distribution, then any Derivative Works that You distribute must
|
||||
include a readable copy of the attribution notices contained
|
||||
within such NOTICE file, excluding those notices that do not
|
||||
pertain to any part of the Derivative Works, in at least one
|
||||
of the following places: within a NOTICE text file distributed
|
||||
as part of the Derivative Works; within the Source form or
|
||||
documentation, if provided along with the Derivative Works; or,
|
||||
within a display generated by the Derivative Works, if and
|
||||
wherever such third-party notices normally appear. The contents
|
||||
of the NOTICE file are for informational purposes only and
|
||||
do not modify the License. You may add Your own attribution
|
||||
notices within Derivative Works that You distribute, alongside
|
||||
or as an addendum to the NOTICE text from the Work, provided
|
||||
that such additional attribution notices cannot be construed
|
||||
as modifying the License.
|
||||
|
||||
You may add Your own copyright statement to Your modifications and
|
||||
may provide additional or different license terms and conditions
|
||||
for use, reproduction, or distribution of Your modifications, or
|
||||
for any such Derivative Works as a whole, provided Your use,
|
||||
reproduction, and distribution of the Work otherwise complies with
|
||||
the conditions stated in this License.
|
||||
|
||||
5. Submission of Contributions. Unless You explicitly state otherwise,
|
||||
any Contribution intentionally submitted for inclusion in the Work
|
||||
by You to the Licensor shall be under the terms and conditions of
|
||||
this License, without any additional terms or conditions.
|
||||
Notwithstanding the above, nothing herein shall supersede or modify
|
||||
the terms of any separate license agreement you may have executed
|
||||
with Licensor regarding such Contributions.
|
||||
|
||||
6. Trademarks. This License does not grant permission to use the trade
|
||||
names, trademarks, service marks, or product names of the Licensor,
|
||||
except as required for reasonable and customary use in describing the
|
||||
origin of the Work and reproducing the content of the NOTICE file.
|
||||
|
||||
7. Disclaimer of Warranty. Unless required by applicable law or
|
||||
agreed to in writing, Licensor provides the Work (and each
|
||||
Contributor provides its Contributions) on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
||||
implied, including, without limitation, any warranties or conditions
|
||||
of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
|
||||
PARTICULAR PURPOSE. You are solely responsible for determining the
|
||||
appropriateness of using or redistributing the Work and assume any
|
||||
risks associated with Your exercise of permissions under this License.
|
||||
|
||||
8. Limitation of Liability. In no event and under no legal theory,
|
||||
whether in tort (including negligence), contract, or otherwise,
|
||||
unless required by applicable law (such as deliberate and grossly
|
||||
negligent acts) or agreed to in writing, shall any Contributor be
|
||||
liable to You for damages, including any direct, indirect, special,
|
||||
incidental, or consequential damages of any character arising as a
|
||||
result of this License or out of the use or inability to use the
|
||||
Work (including but not limited to damages for loss of goodwill,
|
||||
work stoppage, computer failure or malfunction, or any and all
|
||||
other commercial damages or losses), even if such Contributor
|
||||
has been advised of the possibility of such damages.
|
||||
|
||||
9. Accepting Warranty or Additional Liability. While redistributing
|
||||
the Work or Derivative Works thereof, You may choose to offer,
|
||||
and charge a fee for, acceptance of support, warranty, indemnity,
|
||||
or other liability obligations and/or rights consistent with this
|
||||
License. However, in accepting such obligations, You may act only
|
||||
on Your own behalf and on Your sole responsibility, not on behalf
|
||||
of any other Contributor, and only if You agree to indemnify,
|
||||
defend, and hold each Contributor harmless for any liability
|
||||
incurred by, or claims asserted against, such Contributor by reason
|
||||
of your accepting any such warranty or additional liability.
|
||||
|
||||
END OF TERMS AND CONDITIONS
|
||||
|
||||
APPENDIX: How to apply the Apache License to your work.
|
||||
|
||||
To apply the Apache License to your work, attach the following
|
||||
boilerplate notice, with the fields enclosed by brackets "[]"
|
||||
replaced with your own identifying information. (Don't include
|
||||
the brackets!) The text should be enclosed in the appropriate
|
||||
comment syntax for the file format. We also recommend that a
|
||||
file or class name and description of purpose be included on the
|
||||
same "printed page" as the copyright notice for easier
|
||||
identification within third-party archives.
|
||||
|
||||
Copyright [yyyy] [name of copyright owner]
|
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at
|
||||
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
||||
Unless required by applicable law or agreed to in writing, software
|
||||
distributed under the License is distributed on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
@@ -0,0 +1,6 @@
|
||||
aiortc==1.9.0
|
||||
aioice==0.9.0
|
||||
av==12.3.0
|
||||
numpy>=1.26.0
|
||||
pillow>=10.3.0
|
||||
websockets>=12.0
|
||||
@@ -0,0 +1,234 @@
|
||||
import asyncio
|
||||
import websockets
|
||||
import base64
|
||||
import time
|
||||
import json
|
||||
import requests
|
||||
from aiortc.sdp import candidate_from_sdp, candidate_to_sdp
|
||||
from aiortc.contrib.signaling import object_from_string, object_to_string
|
||||
from aiortc import RTCIceCandidate, RTCPeerConnection, RTCSessionDescription
|
||||
from aiortc.contrib.signaling import BYE, add_signaling_arguments, create_signaling
|
||||
from aiortc.contrib.media import MediaBlackhole, MediaPlayer, MediaRecorder, MediaRelay
|
||||
|
||||
relay = MediaRelay()
|
||||
sub_key = None
|
||||
|
||||
def login(phonenumber, code=None, token_file="token"):
|
||||
"""Request an SMS code (when code is None) or exchange it for a token."""
|
||||
if "-" not in phonenumber:
|
||||
raise ValueError("phone number must use the '<country-code>-<number>' form")
|
||||
|
||||
if code is None:
|
||||
response = requests.post(
|
||||
"https://n.cg.163.com/api/v1/phone-captchas/" + phonenumber,
|
||||
timeout=15,
|
||||
)
|
||||
response.raise_for_status()
|
||||
print("input the code received by your phone")
|
||||
code = input().strip()
|
||||
|
||||
country_code, phone = phonenumber.split("-", 1)
|
||||
headers = {"Content-Type": "application/json;charset=utf-8"}
|
||||
data = {
|
||||
"auth_method": "phone-captcha",
|
||||
"ctcode": country_code,
|
||||
"phone": phone,
|
||||
"captcha": str(code).strip(),
|
||||
"device_info": {
|
||||
"userAgent": "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 Chrome/140 Safari/537.36",
|
||||
"appVersion": "5.0 (X11)",
|
||||
"codecs": ["h264", "vp8", "vp9"],
|
||||
},
|
||||
}
|
||||
response = requests.post(
|
||||
"https://n.cg.163.com/api/v1/tokens",
|
||||
headers=headers,
|
||||
json=data,
|
||||
timeout=15,
|
||||
)
|
||||
response.raise_for_status()
|
||||
user_obj = response.json()
|
||||
token = user_obj.get("token")
|
||||
if not token:
|
||||
raise RuntimeError("login response did not contain a token")
|
||||
|
||||
with open(token_file, "w", encoding="utf-8") as token_handle:
|
||||
token_handle.write(token)
|
||||
try:
|
||||
import os
|
||||
os.chmod(token_file, 0o600)
|
||||
except OSError:
|
||||
pass
|
||||
print("login ok!")
|
||||
return token
|
||||
|
||||
|
||||
def encode_mess(message):
|
||||
global sub_key
|
||||
mess = str.encode(message)
|
||||
m = b""
|
||||
for i in mess:
|
||||
m += ((i+sub_key)%256).to_bytes(1, byteorder='big')
|
||||
res = base64.b64encode(m)
|
||||
return res
|
||||
|
||||
def decode_mess(message):
|
||||
try:
|
||||
mess = base64.b64decode(message)
|
||||
except:
|
||||
return message
|
||||
global sub_key
|
||||
if sub_key == None:
|
||||
fbyte = mess[0]
|
||||
sbyte = mess[1]
|
||||
for j in range(0, 256):
|
||||
if(chr((fbyte-j)%256)=="[" and chr((sbyte-j)%256) == "{" and chr((mess[2]-j)%256) == "\""):
|
||||
sub_key = j
|
||||
break
|
||||
if chr((fbyte-j)%256) == "{" and chr((sbyte-j)%256) == "\"":
|
||||
sub_key = j
|
||||
break
|
||||
m = ""
|
||||
for i in mess:
|
||||
m += chr((i-sub_key)%256)
|
||||
return m
|
||||
|
||||
def get_basic_info(token):
|
||||
headers = {"Authorization": "Bearer " + token}
|
||||
response = requests.get(
|
||||
"https://n.cg.163.com/api/v2/users/@me",
|
||||
headers=headers,
|
||||
timeout=15,
|
||||
)
|
||||
response.raise_for_status()
|
||||
info = json.loads(decode_mess(response.text))
|
||||
return info["yunxin_account"]["accid"]
|
||||
|
||||
def request_ticket(token, game_code, regions=None, codecs=None, width=1280, height=720):
|
||||
if regions is None:
|
||||
regions = ["hdcz", "hbsjz"]
|
||||
if codecs is None:
|
||||
codecs = ["h264", "vp8", "vp9"]
|
||||
req_obj = {
|
||||
"regions": regions,
|
||||
"game_code": game_code,
|
||||
"codecs": codecs,
|
||||
"width": width,
|
||||
"height": height
|
||||
}
|
||||
global sub_key
|
||||
|
||||
req_str = encode_mess(json.dumps(req_obj).replace("'","\""))
|
||||
headers = {
|
||||
"Authorization": "Bearer " +token,
|
||||
"Content-Type": "application/octet-stream"
|
||||
}
|
||||
response = requests.post(
|
||||
"https://n.cg.163.com/api/v2/tickets",
|
||||
headers=headers,
|
||||
data=req_str,
|
||||
timeout=15,
|
||||
)
|
||||
if not response.ok:
|
||||
try:
|
||||
detail = decode_mess(response.text)
|
||||
except Exception:
|
||||
detail = ""
|
||||
raise RuntimeError(f"ticket request failed with HTTP {response.status_code}: {detail[:500]}")
|
||||
info_obj = json.loads(decode_mess(response.text))
|
||||
return info_obj["gateway_url"]
|
||||
|
||||
def find_region(token, game_code):
|
||||
headers = {"Authorization": "Bearer " + token}
|
||||
response = requests.get(
|
||||
"https://n.cg.163.com/api/v2/media-servers",
|
||||
params={"game_code": game_code},
|
||||
headers=headers,
|
||||
timeout=15,
|
||||
)
|
||||
response.raise_for_status()
|
||||
regions = json.loads(decode_mess(response.text))
|
||||
return [item["region"] for item in regions]
|
||||
|
||||
def exit_game(token, game_code):
|
||||
headers = {"Authorization": "Bearer " + token}
|
||||
response = requests.delete(
|
||||
f"https://n.cg.163.com/api/v2/users/@me/games-playing/{game_code}",
|
||||
headers=headers,
|
||||
timeout=15,
|
||||
)
|
||||
response.raise_for_status()
|
||||
return response.text
|
||||
|
||||
async def connect(token, game_code, w=1280, h=720, quality="high", codecs=None, platform=0, fps="30"):
|
||||
if codecs is None:
|
||||
codecs = ["h264", "vp8", "vp9"]
|
||||
user_id = await asyncio.to_thread(get_basic_info, token)
|
||||
regions = await asyncio.to_thread(find_region, token, game_code)
|
||||
uri = await asyncio.to_thread(
|
||||
request_ticket,
|
||||
token,
|
||||
game_code,
|
||||
regions,
|
||||
codecs,
|
||||
w,
|
||||
h,
|
||||
)
|
||||
websocket = await websockets.connect(uri)
|
||||
auth_obj = {
|
||||
"id": str(int(round(time.time() * 1000))),
|
||||
"op": "auth",
|
||||
"data": {
|
||||
"user_id": user_id,
|
||||
"token": token,
|
||||
"game_code": game_code,
|
||||
"w": w,
|
||||
"h": h,
|
||||
"quality": quality,
|
||||
"codecs": codecs,
|
||||
"platform": platform,
|
||||
"fps": fps
|
||||
}
|
||||
}
|
||||
auth_str = json.dumps(auth_obj)
|
||||
await websocket.send(auth_str)
|
||||
auth_res = await websocket.recv()
|
||||
auth_res = decode_mess(auth_res)
|
||||
try:
|
||||
de_res = json.loads(auth_res)
|
||||
except (TypeError, json.JSONDecodeError) as exc:
|
||||
await websocket.close()
|
||||
raise RuntimeError("failed to decode cloud-game auth response") from exc
|
||||
if de_res.get("op") != "offer":
|
||||
error_message = de_res.get("data", {}).get("errmsg", "unexpected signaling response")
|
||||
await websocket.close()
|
||||
raise RuntimeError(error_message)
|
||||
new_sdp = {}
|
||||
new_sdp["type"] = de_res["op"]
|
||||
new_sdp["sdp"] = de_res["data"]["sdp"]
|
||||
new_sdp = json.dumps(new_sdp)
|
||||
return new_sdp, websocket
|
||||
|
||||
|
||||
def channel_log(channel, t, message):
|
||||
print("channel(%s) %s %s" % (channel.label, t, message))
|
||||
|
||||
async def send_action(sock, action):
|
||||
|
||||
action = encode_mess(json.dumps(action).replace("'","\""))
|
||||
await sock.send(action)
|
||||
pong_waiter = await sock.ping()
|
||||
await pong_waiter
|
||||
|
||||
def pack_message(cmd, data):
|
||||
if cmd == "mm": # move mouse, data: x, y
|
||||
action = {"id":str(int(round(time.time() * 1000))),"op":"input","data":{"cmd":"1 %d %d 0" % (data["x"], data["y"])}}
|
||||
elif cmd == "cm": # click mouse, data: x, y
|
||||
action = {"id":str(int(round(time.time() * 1000))),"op":"input","data":{"cmd":"3 %d %d 0" % (data["x"], data["y"])}}
|
||||
elif cmd == "ip": # keyboard input, data: single keyborad word
|
||||
action = {"id":str(int(round(time.time() * 1000))),"op":"input","data":{"cmd":"5 %s" % data["word"]}}
|
||||
else:
|
||||
action = {}
|
||||
return action
|
||||
|
||||
#print(decode_mess("2YDHwoCYgI+VlI6SlpeTk4+Uj5eAioDNzoCYgMfMztPSgIqAwr/Sv4CY2YDBy8KAmICRfpaOln6TkZB+joDb2w=="))
|
||||
@@ -0,0 +1,573 @@
|
||||
import asyncio
|
||||
import json
|
||||
import time
|
||||
import sys
|
||||
import signal
|
||||
import platform
|
||||
import logging
|
||||
import base64
|
||||
import os
|
||||
from pathlib import Path
|
||||
from typing import Optional, Coroutine
|
||||
try:
|
||||
from websockets.legacy.client import WebSocketClientProtocol
|
||||
except ImportError: # websockets >=12 removed legacy package
|
||||
from websockets.client import WebSocketClientProtocol # type: ignore[attr-defined]
|
||||
from io import BytesIO
|
||||
|
||||
import requests
|
||||
from aiohttp import web
|
||||
|
||||
from aiortc import RTCPeerConnection, RTCSessionDescription
|
||||
from aiortc.contrib.media import MediaRelay
|
||||
from sdk.wsconnect import (
|
||||
connect, object_from_string, encode_mess,
|
||||
pack_message, send_action, login, exit_game
|
||||
)
|
||||
|
||||
# --- 配置 ---
|
||||
BASE_DIR = Path(__file__).resolve().parent
|
||||
GAME_CODE = os.getenv("MAA_ONLINE_GAME_CODE", "mrfz")
|
||||
TOKEN_FILE = Path(os.getenv("MAA_ONLINE_TOKEN_FILE", str(BASE_DIR / "token")))
|
||||
HOST = os.getenv("MAA_ONLINE_HOST", "127.0.0.1")
|
||||
PORT = int(os.getenv("MAA_ONLINE_PORT", "22888"))
|
||||
WIDTH = int(os.getenv("MAA_ONLINE_WIDTH", "1280"))
|
||||
HEIGHT = int(os.getenv("MAA_ONLINE_HEIGHT", "720"))
|
||||
FPS = os.getenv("MAA_ONLINE_FPS", "30")
|
||||
START_WAIT_SECONDS = float(os.getenv("MAA_ONLINE_START_WAIT_SECONDS", "30"))
|
||||
|
||||
# --- 颜色定义 ---
|
||||
class Colors:
|
||||
RESET = '\033[0m'
|
||||
RED = '\033[31m'
|
||||
GREEN = '\033[32m'
|
||||
YELLOW = '\033[33m'
|
||||
BLUE = '\033[34m'
|
||||
CYAN = '\033[36m'
|
||||
BOLD = '\033[1m'
|
||||
|
||||
if platform.system() == "Windows":
|
||||
asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
|
||||
|
||||
|
||||
# --- API 解码工具 ---
|
||||
def decode_api_response(encoded_data: str) -> dict:
|
||||
"""Decode base64 encoded API response from cg.163.com"""
|
||||
try:
|
||||
raw_bytes = base64.b64decode(encoded_data)
|
||||
except Exception:
|
||||
return {}
|
||||
|
||||
# Find the decryption key by detecting JSON start pattern
|
||||
decode_key = None
|
||||
if len(raw_bytes) >= 2:
|
||||
first_byte = raw_bytes[0]
|
||||
second_byte = raw_bytes[1]
|
||||
for key_candidate in range(256):
|
||||
char1 = chr((first_byte - key_candidate) % 256)
|
||||
char2 = chr((second_byte - key_candidate) % 256)
|
||||
if char1 == "{" and char2 == "\"":
|
||||
decode_key = key_candidate
|
||||
break
|
||||
|
||||
if decode_key is None:
|
||||
return {}
|
||||
|
||||
decoded_str = "".join(chr((b - decode_key) % 256) for b in raw_bytes)
|
||||
try:
|
||||
return json.loads(decoded_str)
|
||||
except json.JSONDecodeError:
|
||||
return {}
|
||||
|
||||
|
||||
def fetch_user_info(token: str) -> dict:
|
||||
"""Fetch user info including remaining game time from cg.163.com API"""
|
||||
try:
|
||||
headers = {"Authorization": f"Bearer {token}"}
|
||||
response = requests.get(
|
||||
"https://n.cg.163.com/api/v2/users/@me",
|
||||
headers=headers,
|
||||
timeout=10
|
||||
)
|
||||
if response.status_code != 200:
|
||||
return {}
|
||||
return decode_api_response(response.text)
|
||||
except Exception as e:
|
||||
logging.warning(f"Failed to fetch user info: {e}")
|
||||
return {}
|
||||
|
||||
|
||||
# --- 全局状态 ---
|
||||
class AppState:
|
||||
def __init__(self):
|
||||
self.pc: Optional[RTCPeerConnection] = None
|
||||
self.sock: Optional[WebSocketClientProtocol] = None
|
||||
self.snapshotter: Optional['VideoSnapshotter'] = None
|
||||
self.is_ready = False
|
||||
self.width = WIDTH
|
||||
self.height = HEIGHT
|
||||
self.token: Optional[str] = None
|
||||
self.cloud_game_task: Optional[asyncio.Task] = None
|
||||
self.ready_event = asyncio.Event()
|
||||
self.last_error: Optional[str] = None
|
||||
|
||||
def reset(self):
|
||||
self.pc = None
|
||||
self.sock = None
|
||||
self.snapshotter = None
|
||||
self.is_ready = False
|
||||
self.token = None
|
||||
self.cloud_game_task = None
|
||||
|
||||
app_state = AppState()
|
||||
|
||||
# --- 快照工具 (从 ark-demo.py 移植并修改) ---
|
||||
class VideoSnapshotter:
|
||||
def __init__(self, video_track):
|
||||
self._track = video_track
|
||||
self._task: Optional[asyncio.Task] = None
|
||||
self._last_frame = None
|
||||
self._got_first = asyncio.Event()
|
||||
self._running = False
|
||||
self._last_recv_ts: float = 0.0
|
||||
|
||||
def start(self):
|
||||
if self._task:
|
||||
return
|
||||
self._running = True
|
||||
self._task = asyncio.create_task(self._pump())
|
||||
|
||||
async def _pump(self):
|
||||
try:
|
||||
while self._running:
|
||||
frame = await self._track.recv()
|
||||
self._last_frame = frame
|
||||
self._last_recv_ts = time.monotonic()
|
||||
if frame.width > 0 and frame.height > 0:
|
||||
app_state.width = frame.width
|
||||
app_state.height = frame.height
|
||||
if not self._got_first.is_set():
|
||||
self._got_first.set()
|
||||
except asyncio.CancelledError:
|
||||
pass # 任务被取消是正常的
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
async def wait_ready(self, timeout=20.0):
|
||||
try:
|
||||
await asyncio.wait_for(self._got_first.wait(), timeout=timeout)
|
||||
return True
|
||||
except asyncio.TimeoutError:
|
||||
return False
|
||||
|
||||
async def snapshot_bytes(self, fmt="png") -> Optional[bytes]:
|
||||
if not self._got_first.is_set():
|
||||
return None
|
||||
if self._last_recv_ts and time.monotonic() - self._last_recv_ts > 10:
|
||||
self._last_frame = None
|
||||
if app_state.is_ready:
|
||||
app_state.is_ready = False
|
||||
logging.warning("Video stream stale for over 10s; marking service as not ready")
|
||||
return None
|
||||
frame = self._last_frame
|
||||
if frame is None:
|
||||
return None
|
||||
try:
|
||||
from PIL import Image
|
||||
img = frame.to_image()
|
||||
with BytesIO() as bio:
|
||||
img.save(bio, format=fmt.upper())
|
||||
return bio.getvalue()
|
||||
except Exception as exc:
|
||||
logging.exception("Failed to encode cloud-game frame: %s", exc)
|
||||
return None
|
||||
|
||||
async def stop(self):
|
||||
self._running = False
|
||||
if self._task:
|
||||
self._task.cancel()
|
||||
try:
|
||||
await self._task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
self._task = None
|
||||
|
||||
# --- API 处理函数 ---
|
||||
async def handle_info(request: web.Request):
|
||||
if not app_state.is_ready:
|
||||
status = "connecting" if app_state.cloud_game_task and not app_state.cloud_game_task.done() else "disconnected"
|
||||
if app_state.last_error:
|
||||
status = "error"
|
||||
return web.json_response({
|
||||
"status": status,
|
||||
"message": app_state.last_error or "Cloud gaming service not ready or not connected.",
|
||||
})
|
||||
|
||||
return web.json_response({
|
||||
"status": "ok",
|
||||
"width": app_state.width,
|
||||
"height": app_state.height,
|
||||
})
|
||||
|
||||
async def handle_account(request: web.Request):
|
||||
if not app_state.token:
|
||||
return web.json_response({"status": "login_required"}, status=401)
|
||||
loop = asyncio.get_running_loop()
|
||||
user_info = await loop.run_in_executor(None, fetch_user_info, app_state.token)
|
||||
if not user_info:
|
||||
return web.json_response({"status": "error", "message": "Failed to query account."}, status=502)
|
||||
return web.json_response({
|
||||
"status": "ok",
|
||||
"remaining_time": user_info.get("free_time_left"),
|
||||
})
|
||||
|
||||
async def handle_screencap(request: web.Request):
|
||||
if not app_state.is_ready or not app_state.snapshotter:
|
||||
return web.json_response({"status": "error", "message": "Service not ready"}, status=503)
|
||||
|
||||
image_bytes = await app_state.snapshotter.snapshot_bytes(fmt="png")
|
||||
if not image_bytes:
|
||||
return web.json_response({"status": "error", "message": "Failed to capture screen"}, status=500)
|
||||
|
||||
return web.Response(body=image_bytes, content_type="image/png")
|
||||
|
||||
async def handle_click(request: web.Request):
|
||||
if not app_state.is_ready or not app_state.sock:
|
||||
return web.json_response({"status": "error", "message": "Service not ready"}, status=503)
|
||||
|
||||
try:
|
||||
data = await request.json()
|
||||
x, y = int(data['x']), int(data['y'])
|
||||
except (json.JSONDecodeError, KeyError, TypeError, ValueError):
|
||||
return web.json_response({"status": "error", "message": "Invalid request body"}, status=400)
|
||||
if not (0 <= x < app_state.width and 0 <= y < app_state.height):
|
||||
return web.json_response({"status": "error", "message": "Coordinates out of range"}, status=400)
|
||||
|
||||
logging.warning(f"Action: Click at ({x}, {y})")
|
||||
|
||||
# 1. 先移动鼠标到目标位置
|
||||
move_action = pack_message("mm", {"x": x, "y": y})
|
||||
await send_action(app_state.sock, move_action)
|
||||
await asyncio.sleep(0.05) # 短暂等待,模拟真实操作
|
||||
|
||||
# 2. 再执行点击操作
|
||||
click_action = pack_message("cm", {"x": x, "y": y})
|
||||
await send_action(app_state.sock, click_action)
|
||||
|
||||
return web.json_response({"status": "ok"})
|
||||
|
||||
async def handle_swipe(request: web.Request):
|
||||
if not app_state.is_ready or not app_state.sock:
|
||||
return web.json_response({"status": "error", "message": "Service not ready"}, status=503)
|
||||
|
||||
try:
|
||||
data = await request.json()
|
||||
start_x, start_y = int(data['x1']), int(data['y1'])
|
||||
end_x, end_y = int(data['x2']), int(data['y2'])
|
||||
swipe_duration = int(data['duration'])
|
||||
except (json.JSONDecodeError, KeyError, TypeError, ValueError):
|
||||
return web.json_response({"status": "error", "message": "Invalid request body"}, status=400)
|
||||
if not (50 <= swipe_duration <= 10000):
|
||||
return web.json_response({"status": "error", "message": "Duration must be between 50 and 10000 ms"}, status=400)
|
||||
|
||||
logging.warning(f"Action: Swipe from ({start_x}, {start_y}) to ({end_x}, {end_y}) in {swipe_duration}ms")
|
||||
|
||||
# The web client protocol uses raw video-frame pixel coordinates, matching
|
||||
# the existing mouse move/click commands. Keep swipe in the same space.
|
||||
def clamp_coord(pixel_val: int, dimension: int) -> int:
|
||||
return max(0, min(pixel_val, dimension - 1))
|
||||
|
||||
screen_width = app_state.width
|
||||
screen_height = app_state.height
|
||||
|
||||
num_points = max(5, swipe_duration // 30)
|
||||
interval = swipe_duration / 1000.0 / num_points
|
||||
touch_id = 0
|
||||
|
||||
def create_touch_cmd(evt_type: int, px: int, py: int, tid: int) -> dict:
|
||||
"""Create a touch command in raw video-frame pixel coordinates."""
|
||||
raw_x = clamp_coord(px, screen_width)
|
||||
raw_y = clamp_coord(py, screen_height)
|
||||
timestamp = str(int(time.time() * 1000))
|
||||
cmd_str = f"{evt_type} {raw_x} {raw_y} {tid}"
|
||||
return {"id": timestamp, "op": "input", "data": {"cmd": cmd_str}}
|
||||
|
||||
# Touch down at starting point
|
||||
down_cmd = create_touch_cmd(1, start_x, start_y, touch_id)
|
||||
await send_action(app_state.sock, down_cmd)
|
||||
await asyncio.sleep(0.02)
|
||||
|
||||
# Move through path points
|
||||
delta_x = end_x - start_x
|
||||
delta_y = end_y - start_y
|
||||
for step in range(1, num_points + 1):
|
||||
progress = step / num_points
|
||||
current_x = int(start_x + delta_x * progress)
|
||||
current_y = int(start_y + delta_y * progress)
|
||||
|
||||
move_cmd = create_touch_cmd(2, current_x, current_y, touch_id)
|
||||
await send_action(app_state.sock, move_cmd)
|
||||
await asyncio.sleep(interval)
|
||||
|
||||
# Touch up at ending point
|
||||
up_cmd = create_touch_cmd(3, end_x, end_y, touch_id)
|
||||
await send_action(app_state.sock, up_cmd)
|
||||
|
||||
return web.json_response({"status": "ok"})
|
||||
|
||||
async def handle_input(request: web.Request):
|
||||
if not app_state.is_ready or not app_state.sock:
|
||||
return web.json_response({"status": "error", "message": "Service not ready"}, status=503)
|
||||
|
||||
try:
|
||||
data = await request.json()
|
||||
text = str(data['text'])
|
||||
except (json.JSONDecodeError, KeyError, TypeError):
|
||||
return web.json_response({"status": "error", "message": "Invalid request body, 'text' field is required"}, status=400)
|
||||
if not text or len(text) > 1024:
|
||||
return web.json_response({"status": "error", "message": "Text length must be between 1 and 1024"}, status=400)
|
||||
|
||||
for char in text:
|
||||
action = pack_message("ip", {"word": char})
|
||||
await send_action(app_state.sock, action)
|
||||
await asyncio.sleep(0.05)
|
||||
|
||||
return web.json_response({"status": "ok"})
|
||||
|
||||
async def handle_start(request: web.Request):
|
||||
if app_state.is_ready:
|
||||
return web.json_response({"status": "ok", "message": "Cloud game connection already active."})
|
||||
|
||||
if not TOKEN_FILE.is_file():
|
||||
return web.json_response(
|
||||
{"status": "login_required", "message": "Run login.py before starting the cloud game."},
|
||||
status=401,
|
||||
)
|
||||
|
||||
if app_state.cloud_game_task and app_state.cloud_game_task.done():
|
||||
app_state.cloud_game_task = None
|
||||
|
||||
if not app_state.cloud_game_task:
|
||||
app_state.last_error = None
|
||||
app_state.ready_event.clear()
|
||||
app_state.cloud_game_task = asyncio.create_task(run_cloud_game())
|
||||
|
||||
try:
|
||||
await asyncio.wait_for(app_state.ready_event.wait(), timeout=START_WAIT_SECONDS)
|
||||
except asyncio.TimeoutError:
|
||||
return web.json_response(
|
||||
{"status": "connecting", "message": "Cloud game connection is still starting."},
|
||||
status=202,
|
||||
)
|
||||
|
||||
if app_state.is_ready:
|
||||
return web.json_response({"status": "ok", "message": "Cloud game connection is ready."})
|
||||
|
||||
return web.json_response(
|
||||
{"status": "error", "message": app_state.last_error or "Cloud game connection failed."},
|
||||
status=502,
|
||||
)
|
||||
|
||||
async def handle_exit(request: web.Request):
|
||||
if not app_state.cloud_game_task or app_state.cloud_game_task.done():
|
||||
return web.json_response({"status": "ok", "message": "No active cloud game connection to exit."})
|
||||
|
||||
print(f"{Colors.CYAN}[*] Received exit request. Disconnecting from cloud game...{Colors.RESET}")
|
||||
|
||||
app_state.cloud_game_task.cancel()
|
||||
try:
|
||||
await app_state.cloud_game_task
|
||||
except asyncio.CancelledError:
|
||||
pass # Expected
|
||||
|
||||
return web.json_response({"status": "ok", "message": "Cloud game connection terminated."})
|
||||
|
||||
async def handle_root(request: web.Request):
|
||||
raise web.HTTPMovedPermanently('/info')
|
||||
|
||||
# --- 云游戏连接逻辑 ---
|
||||
async def run_cloud_game():
|
||||
try:
|
||||
try:
|
||||
token = TOKEN_FILE.read_text(encoding="utf-8").strip()
|
||||
except FileNotFoundError:
|
||||
token = ""
|
||||
if not token:
|
||||
raise RuntimeError("login token is missing; run login.py first")
|
||||
|
||||
app_state.token = token
|
||||
print(f"{Colors.CYAN}[*] Connecting to cloud gaming service...{Colors.RESET}")
|
||||
res, sock = await connect(token, GAME_CODE, w=app_state.width, h=app_state.height, fps=FPS)
|
||||
app_state.sock = sock
|
||||
remote = object_from_string(res)
|
||||
if not isinstance(remote, RTCSessionDescription):
|
||||
raise RuntimeError(f"Unexpected offer type from signaling: {type(remote)!r}")
|
||||
|
||||
pc = RTCPeerConnection()
|
||||
app_state.pc = pc
|
||||
relay = MediaRelay()
|
||||
track_event = asyncio.Event()
|
||||
|
||||
@pc.on("track")
|
||||
def on_track(track):
|
||||
if track.kind == "video":
|
||||
print(f"{Colors.BLUE}[*] Video track received.{Colors.RESET}")
|
||||
app_state.snapshotter = VideoSnapshotter(relay.subscribe(track))
|
||||
app_state.snapshotter.start()
|
||||
track_event.set()
|
||||
|
||||
await pc.setRemoteDescription(remote)
|
||||
answer = await pc.createAnswer()
|
||||
if answer is None:
|
||||
raise RuntimeError("Failed to create SDP answer")
|
||||
patched = RTCSessionDescription(
|
||||
sdp=answer.sdp.replace("a=setup:active", "a=setup:passive"),
|
||||
type=answer.type,
|
||||
)
|
||||
await pc.setLocalDescription(patched)
|
||||
msg = {"id": str(int(time.time() * 1000)), "op": "answer", "data": {"sdp": patched.sdp}}
|
||||
await sock.send(encode_mess(json.dumps(msg)))
|
||||
|
||||
print(f"{Colors.YELLOW}[*] Waiting for video stream...{Colors.RESET}")
|
||||
await asyncio.wait_for(track_event.wait(), timeout=20.0)
|
||||
|
||||
if app_state.snapshotter and await app_state.snapshotter.wait_ready():
|
||||
app_state.is_ready = True
|
||||
app_state.ready_event.set()
|
||||
print(f"{Colors.GREEN}{Colors.BOLD}[✓] Cloud game ready. API is active.{Colors.RESET}")
|
||||
else:
|
||||
raise RuntimeError("failed to receive the first cloud-game video frame")
|
||||
|
||||
# Use a cancellable wait instead of Event().wait()
|
||||
# This allows the task to be properly cancelled by signals
|
||||
while True:
|
||||
await asyncio.sleep(1)
|
||||
|
||||
except asyncio.CancelledError:
|
||||
print(f"\n{Colors.CYAN}[*] Cloud game task cancelled.{Colors.RESET}")
|
||||
except Exception as e:
|
||||
app_state.last_error = str(e)
|
||||
app_state.ready_event.set()
|
||||
print(f"\n{Colors.RED}[!] An error occurred in cloud game task: {e}{Colors.RESET}", file=sys.stderr)
|
||||
logging.exception("Cloud game task error")
|
||||
finally:
|
||||
print(f"{Colors.CYAN}[*] Cleaning up cloud game resources...{Colors.RESET}")
|
||||
|
||||
token_to_use = app_state.token
|
||||
pc_to_close = app_state.pc
|
||||
sock_to_close = app_state.sock
|
||||
snapshotter_to_stop = app_state.snapshotter
|
||||
|
||||
# 重置状态
|
||||
app_state.reset()
|
||||
|
||||
if token_to_use:
|
||||
print(f"{Colors.CYAN}[*] Sending exit command to cloud game server...{Colors.RESET}")
|
||||
try:
|
||||
loop = asyncio.get_running_loop()
|
||||
await loop.run_in_executor(None, exit_game, token_to_use, GAME_CODE)
|
||||
except Exception as exc:
|
||||
logging.warning("Failed to release remote cloud-game session: %s", exc)
|
||||
|
||||
async def close_with_timeout(awaitable: Coroutine, timeout=5.0):
|
||||
try:
|
||||
await asyncio.wait_for(awaitable, timeout=timeout)
|
||||
except (asyncio.TimeoutError, asyncio.CancelledError):
|
||||
pass
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
if snapshotter_to_stop:
|
||||
await close_with_timeout(snapshotter_to_stop.stop())
|
||||
if pc_to_close and pc_to_close.connectionState != "closed":
|
||||
await close_with_timeout(pc_to_close.close())
|
||||
if sock_to_close and sock_to_close.close_code is None:
|
||||
await close_with_timeout(sock_to_close.close())
|
||||
|
||||
print(f"{Colors.GREEN}[✓] Cloud game resources cleaned up.{Colors.RESET}")
|
||||
|
||||
|
||||
async def cleanup_background_tasks(app: web.Application):
|
||||
if app_state.cloud_game_task and not app_state.cloud_game_task.done():
|
||||
print(f"\n{Colors.CYAN}Server shutting down. Cleaning up active cloud game connection...{Colors.RESET}")
|
||||
app_state.cloud_game_task.cancel()
|
||||
try:
|
||||
await app_state.cloud_game_task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
# --- 主函数 ---
|
||||
async def run_server():
|
||||
"""Run the aiohttp server with proper signal handling."""
|
||||
logging.basicConfig(level=logging.INFO,
|
||||
format='%(asctime)s - %(levelname)s - %(message)s')
|
||||
access_log = logging.getLogger("aiohttp.access")
|
||||
|
||||
app = web.Application()
|
||||
app.add_routes([
|
||||
web.get('/', handle_root),
|
||||
web.get('/info', handle_info),
|
||||
web.get('/account', handle_account),
|
||||
web.get('/screencap', handle_screencap),
|
||||
web.post('/click', handle_click),
|
||||
web.post('/swipe', handle_swipe),
|
||||
web.post('/input', handle_input),
|
||||
web.post('/start', handle_start),
|
||||
web.post('/exit', handle_exit),
|
||||
])
|
||||
|
||||
app.on_cleanup.append(cleanup_background_tasks)
|
||||
|
||||
runner = web.AppRunner(app, access_log=access_log)
|
||||
await runner.setup()
|
||||
|
||||
site = web.TCPSite(runner, HOST, PORT)
|
||||
await site.start()
|
||||
|
||||
print(f"{Colors.GREEN}{Colors.BOLD}[✓] API server is running at http://{HOST}:{PORT}{Colors.RESET}")
|
||||
print(f"{Colors.YELLOW}Send POST to /start to connect to the cloud game.{Colors.RESET}")
|
||||
print(f"{Colors.YELLOW}(Press CTRL+C to quit){Colors.RESET}")
|
||||
|
||||
# Create shutdown event
|
||||
shutdown_event = asyncio.Event()
|
||||
|
||||
# Setup signal handlers
|
||||
loop = asyncio.get_running_loop()
|
||||
|
||||
def signal_handler():
|
||||
print(f"\n{Colors.CYAN}[*] Shutdown signal received...{Colors.RESET}")
|
||||
shutdown_event.set()
|
||||
|
||||
# Register signal handlers
|
||||
# Note: loop.add_signal_handler() is not supported on Windows event loops,
|
||||
# so the KeyboardInterrupt fallback in main() is needed for Windows support
|
||||
if platform.system() != "Windows":
|
||||
for sig in (signal.SIGINT, signal.SIGTERM):
|
||||
loop.add_signal_handler(sig, signal_handler)
|
||||
|
||||
try:
|
||||
# Wait for shutdown signal
|
||||
await shutdown_event.wait()
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
finally:
|
||||
print(f"{Colors.CYAN}[*] Shutting down server...{Colors.RESET}")
|
||||
# Cleanup cloud game task if running
|
||||
if app_state.cloud_game_task and not app_state.cloud_game_task.done():
|
||||
app_state.cloud_game_task.cancel()
|
||||
try:
|
||||
await app_state.cloud_game_task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
await runner.cleanup()
|
||||
print(f"{Colors.GREEN}[✓] Server stopped.{Colors.RESET}")
|
||||
|
||||
def main():
|
||||
try:
|
||||
asyncio.run(run_server())
|
||||
except KeyboardInterrupt:
|
||||
# This handles Ctrl+C on Windows
|
||||
print(f"\n{Colors.GREEN}[✓] Server stopped by user.{Colors.RESET}")
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user