forked from mirrors/stickerpicker
Initial commit
This commit is contained in:
commit
7f86915857
10 changed files with 952 additions and 0 deletions
11
.gitignore
vendored
Normal file
11
.gitignore
vendored
Normal file
|
@ -0,0 +1,11 @@
|
|||
/.idea
|
||||
|
||||
/.venv
|
||||
/env
|
||||
*.pyc
|
||||
__pycache__
|
||||
|
||||
*.session
|
||||
*.json
|
||||
!example-config.json
|
||||
*.bak
|
374
LICENSE
Normal file
374
LICENSE
Normal file
|
@ -0,0 +1,374 @@
|
|||
Mozilla Public License Version 2.0
|
||||
==================================
|
||||
|
||||
1. Definitions
|
||||
--------------
|
||||
|
||||
1.1. "Contributor"
|
||||
means each individual or legal entity that creates, contributes to
|
||||
the creation of, or owns Covered Software.
|
||||
|
||||
1.2. "Contributor Version"
|
||||
means the combination of the Contributions of others (if any) used
|
||||
by a Contributor and that particular Contributor's Contribution.
|
||||
|
||||
1.3. "Contribution"
|
||||
means Covered Software of a particular Contributor.
|
||||
|
||||
1.4. "Covered Software"
|
||||
means Source Code Form to which the initial Contributor has attached
|
||||
the notice in Exhibit A, the Executable Form of such Source Code
|
||||
Form, and Modifications of such Source Code Form, in each case
|
||||
including portions thereof.
|
||||
|
||||
1.5. "Incompatible With Secondary Licenses"
|
||||
means
|
||||
|
||||
(a) that the initial Contributor has attached the notice described
|
||||
in Exhibit B to the Covered Software; or
|
||||
|
||||
(b) that the Covered Software was made available under the terms of
|
||||
version 1.1 or earlier of the License, but not also under the
|
||||
terms of a Secondary License.
|
||||
|
||||
1.6. "Executable Form"
|
||||
means any form of the work other than Source Code Form.
|
||||
|
||||
1.7. "Larger Work"
|
||||
means a work that combines Covered Software with other material, in
|
||||
a separate file or files, that is not Covered Software.
|
||||
|
||||
1.8. "License"
|
||||
means this document.
|
||||
|
||||
1.9. "Licensable"
|
||||
means having the right to grant, to the maximum extent possible,
|
||||
whether at the time of the initial grant or subsequently, any and
|
||||
all of the rights conveyed by this License.
|
||||
|
||||
1.10. "Modifications"
|
||||
means any of the following:
|
||||
|
||||
(a) any file in Source Code Form that results from an addition to,
|
||||
deletion from, or modification of the contents of Covered
|
||||
Software; or
|
||||
|
||||
(b) any new file in Source Code Form that contains any Covered
|
||||
Software.
|
||||
|
||||
1.11. "Patent Claims" of a Contributor
|
||||
means any patent claim(s), including without limitation, method,
|
||||
process, and apparatus claims, in any patent Licensable by such
|
||||
Contributor that would be infringed, but for the grant of the
|
||||
License, by the making, using, selling, offering for sale, having
|
||||
made, import, or transfer of either its Contributions or its
|
||||
Contributor Version.
|
||||
|
||||
1.12. "Secondary License"
|
||||
means either the GNU General Public License, Version 2.0, the GNU
|
||||
Lesser General Public License, Version 2.1, the GNU Affero General
|
||||
Public License, Version 3.0, or any later versions of those
|
||||
licenses.
|
||||
|
||||
1.13. "Source Code Form"
|
||||
means the form of the work preferred for making modifications.
|
||||
|
||||
1.14. "You" (or "Your")
|
||||
means an individual or a legal entity exercising rights under this
|
||||
License. For legal entities, "You" includes any entity that
|
||||
controls, is controlled by, or is under common control with You. For
|
||||
purposes of this definition, "control" means (a) the power, direct
|
||||
or indirect, to cause the direction or management of such entity,
|
||||
whether by contract or otherwise, or (b) ownership of more than
|
||||
fifty percent (50%) of the outstanding shares or beneficial
|
||||
ownership of such entity.
|
||||
|
||||
2. License Grants and Conditions
|
||||
--------------------------------
|
||||
|
||||
2.1. Grants
|
||||
|
||||
Each Contributor hereby grants You a world-wide, royalty-free,
|
||||
non-exclusive license:
|
||||
|
||||
(a) under intellectual property rights (other than patent or trademark)
|
||||
Licensable by such Contributor to use, reproduce, make available,
|
||||
modify, display, perform, distribute, and otherwise exploit its
|
||||
Contributions, either on an unmodified basis, with Modifications, or
|
||||
as part of a Larger Work; and
|
||||
|
||||
(b) under Patent Claims of such Contributor to make, use, sell, offer
|
||||
for sale, have made, import, and otherwise transfer either its
|
||||
Contributions or its Contributor Version.
|
||||
|
||||
2.2. Effective Date
|
||||
|
||||
The licenses granted in Section 2.1 with respect to any Contribution
|
||||
become effective for each Contribution on the date the Contributor first
|
||||
distributes such Contribution.
|
||||
|
||||
2.3. Limitations on Grant Scope
|
||||
|
||||
The licenses granted in this Section 2 are the only rights granted under
|
||||
this License. No additional rights or licenses will be implied from the
|
||||
distribution or licensing of Covered Software under this License.
|
||||
Notwithstanding Section 2.1(b) above, no patent license is granted by a
|
||||
Contributor:
|
||||
|
||||
(a) for any code that a Contributor has removed from Covered Software;
|
||||
or
|
||||
|
||||
(b) for infringements caused by: (i) Your and any other third party's
|
||||
modifications of Covered Software, or (ii) the combination of its
|
||||
Contributions with other software (except as part of its Contributor
|
||||
Version); or
|
||||
|
||||
(c) under Patent Claims infringed by Covered Software in the absence of
|
||||
its Contributions.
|
||||
|
||||
This License does not grant any rights in the trademarks, service marks,
|
||||
or logos of any Contributor (except as may be necessary to comply with
|
||||
the notice requirements in Section 3.4).
|
||||
|
||||
2.4. Subsequent Licenses
|
||||
|
||||
No Contributor makes additional grants as a result of Your choice to
|
||||
distribute the Covered Software under a subsequent version of this
|
||||
License (see Section 10.2) or under the terms of a Secondary License (if
|
||||
permitted under the terms of Section 3.3).
|
||||
|
||||
2.5. Representation
|
||||
|
||||
Each Contributor represents that the Contributor believes its
|
||||
Contributions are its original creation(s) or it has sufficient rights
|
||||
to grant the rights to its Contributions conveyed by this License.
|
||||
|
||||
2.6. Fair Use
|
||||
|
||||
This License is not intended to limit any rights You have under
|
||||
applicable copyright doctrines of fair use, fair dealing, or other
|
||||
equivalents.
|
||||
|
||||
2.7. Conditions
|
||||
|
||||
Sections 3.1, 3.2, 3.3, and 3.4 are conditions of the licenses granted
|
||||
in Section 2.1.
|
||||
|
||||
3. Responsibilities
|
||||
-------------------
|
||||
|
||||
3.1. Distribution of Source Form
|
||||
|
||||
All distribution of Covered Software in Source Code Form, including any
|
||||
Modifications that You create or to which You contribute, must be under
|
||||
the terms of this License. You must inform recipients that the Source
|
||||
Code Form of the Covered Software is governed by the terms of this
|
||||
License, and how they can obtain a copy of this License. You may not
|
||||
attempt to alter or restrict the recipients' rights in the Source Code
|
||||
Form.
|
||||
|
||||
3.2. Distribution of Executable Form
|
||||
|
||||
If You distribute Covered Software in Executable Form then:
|
||||
|
||||
(a) such Covered Software must also be made available in Source Code
|
||||
Form, as described in Section 3.1, and You must inform recipients of
|
||||
the Executable Form how they can obtain a copy of such Source Code
|
||||
Form by reasonable means in a timely manner, at a charge no more
|
||||
than the cost of distribution to the recipient; and
|
||||
|
||||
(b) You may distribute such Executable Form under the terms of this
|
||||
License, or sublicense it under different terms, provided that the
|
||||
license for the Executable Form does not attempt to limit or alter
|
||||
the recipients' rights in the Source Code Form under this License.
|
||||
|
||||
3.3. Distribution of a Larger Work
|
||||
|
||||
You may create and distribute a Larger Work under terms of Your choice,
|
||||
provided that You also comply with the requirements of this License for
|
||||
the Covered Software. If the Larger Work is a combination of Covered
|
||||
Software with a work governed by one or more Secondary Licenses, and the
|
||||
Covered Software is not Incompatible With Secondary Licenses, this
|
||||
License permits You to additionally distribute such Covered Software
|
||||
under the terms of such Secondary License(s), so that the recipient of
|
||||
the Larger Work may, at their option, further distribute the Covered
|
||||
Software under the terms of either this License or such Secondary
|
||||
License(s).
|
||||
|
||||
3.4. Notices
|
||||
|
||||
You may not remove or alter the substance of any license notices
|
||||
(including copyright notices, patent notices, disclaimers of warranty,
|
||||
or limitations of liability) contained within the Source Code Form of
|
||||
the Covered Software, except that You may alter any license notices to
|
||||
the extent required to remedy known factual inaccuracies.
|
||||
|
||||
3.5. Application of Additional Terms
|
||||
|
||||
You may choose to offer, and to charge a fee for, warranty, support,
|
||||
indemnity or liability obligations to one or more recipients of Covered
|
||||
Software. However, You may do so only on Your own behalf, and not on
|
||||
behalf of any Contributor. You must make it absolutely clear that any
|
||||
such warranty, support, indemnity, or liability obligation is offered by
|
||||
You alone, and You hereby agree to indemnify every Contributor for any
|
||||
liability incurred by such Contributor as a result of warranty, support,
|
||||
indemnity or liability terms You offer. You may include additional
|
||||
disclaimers of warranty and limitations of liability specific to any
|
||||
jurisdiction.
|
||||
|
||||
4. Inability to Comply Due to Statute or Regulation
|
||||
---------------------------------------------------
|
||||
|
||||
If it is impossible for You to comply with any of the terms of this
|
||||
License with respect to some or all of the Covered Software due to
|
||||
statute, judicial order, or regulation then You must: (a) comply with
|
||||
the terms of this License to the maximum extent possible; and (b)
|
||||
describe the limitations and the code they affect. Such description must
|
||||
be placed in a text file included with all distributions of the Covered
|
||||
Software under this License. Except to the extent prohibited by statute
|
||||
or regulation, such description must be sufficiently detailed for a
|
||||
recipient of ordinary skill to be able to understand it.
|
||||
|
||||
5. Termination
|
||||
--------------
|
||||
|
||||
5.1. The rights granted under this License will terminate automatically
|
||||
if You fail to comply with any of its terms. However, if You become
|
||||
compliant, then the rights granted under this License from a particular
|
||||
Contributor are reinstated (a) provisionally, unless and until such
|
||||
Contributor explicitly and finally terminates Your grants, and (b) on an
|
||||
ongoing basis, if such Contributor fails to notify You of the
|
||||
non-compliance by some reasonable means prior to 60 days after You have
|
||||
come back into compliance. Moreover, Your grants from a particular
|
||||
Contributor are reinstated on an ongoing basis if such Contributor
|
||||
notifies You of the non-compliance by some reasonable means, this is the
|
||||
first time You have received notice of non-compliance with this License
|
||||
from such Contributor, and You become compliant prior to 30 days after
|
||||
Your receipt of the notice.
|
||||
|
||||
5.2. If You initiate litigation against any entity by asserting a patent
|
||||
infringement claim (excluding declaratory judgment actions,
|
||||
counter-claims, and cross-claims) alleging that a Contributor Version
|
||||
directly or indirectly infringes any patent, then the rights granted to
|
||||
You by any and all Contributors for the Covered Software under Section
|
||||
2.1 of this License shall terminate.
|
||||
|
||||
5.3. In the event of termination under Sections 5.1 or 5.2 above, all
|
||||
end user license agreements (excluding distributors and resellers) which
|
||||
have been validly granted by You or Your distributors under this License
|
||||
prior to termination shall survive termination.
|
||||
|
||||
************************************************************************
|
||||
* *
|
||||
* 6. Disclaimer of Warranty *
|
||||
* ------------------------- *
|
||||
* *
|
||||
* Covered Software is provided under this License on an "as is" *
|
||||
* basis, without warranty of any kind, either expressed, implied, or *
|
||||
* statutory, including, without limitation, warranties that the *
|
||||
* Covered Software is free of defects, merchantable, fit for a *
|
||||
* particular purpose or non-infringing. The entire risk as to the *
|
||||
* quality and performance of the Covered Software is with You. *
|
||||
* Should any Covered Software prove defective in any respect, You *
|
||||
* (not any Contributor) assume the cost of any necessary servicing, *
|
||||
* repair, or correction. This disclaimer of warranty constitutes an *
|
||||
* essential part of this License. No use of any Covered Software is *
|
||||
* authorized under this License except under this disclaimer. *
|
||||
* *
|
||||
************************************************************************
|
||||
|
||||
************************************************************************
|
||||
* *
|
||||
* 7. Limitation of Liability *
|
||||
* -------------------------- *
|
||||
* *
|
||||
* Under no circumstances and under no legal theory, whether tort *
|
||||
* (including negligence), contract, or otherwise, shall any *
|
||||
* Contributor, or anyone who distributes Covered Software as *
|
||||
* permitted above, be liable to You for any direct, indirect, *
|
||||
* special, incidental, or consequential damages of any character *
|
||||
* including, without limitation, damages for lost profits, loss of *
|
||||
* goodwill, work stoppage, computer failure or malfunction, or any *
|
||||
* and all other commercial damages or losses, even if such party *
|
||||
* shall have been informed of the possibility of such damages. This *
|
||||
* limitation of liability shall not apply to liability for death or *
|
||||
* personal injury resulting from such party's negligence to the *
|
||||
* extent applicable law prohibits such limitation. Some *
|
||||
* jurisdictions do not allow the exclusion or limitation of *
|
||||
* incidental or consequential damages, so this exclusion and *
|
||||
* limitation may not apply to You. *
|
||||
* *
|
||||
************************************************************************
|
||||
|
||||
8. Litigation
|
||||
-------------
|
||||
|
||||
Any litigation relating to this License may be brought only in the
|
||||
courts of a jurisdiction where the defendant maintains its principal
|
||||
place of business and such litigation shall be governed by laws of that
|
||||
jurisdiction, without reference to its conflict-of-law provisions.
|
||||
Nothing in this Section shall prevent a party's ability to bring
|
||||
cross-claims or counter-claims.
|
||||
|
||||
9. Miscellaneous
|
||||
----------------
|
||||
|
||||
This License represents the complete agreement concerning the subject
|
||||
matter hereof. If any provision of this License is held to be
|
||||
unenforceable, such provision shall be reformed only to the extent
|
||||
necessary to make it enforceable. Any law or regulation which provides
|
||||
that the language of a contract shall be construed against the drafter
|
||||
shall not be used to construe this License against a Contributor.
|
||||
|
||||
10. Versions of the License
|
||||
---------------------------
|
||||
|
||||
10.1. New Versions
|
||||
|
||||
Mozilla Foundation is the license steward. Except as provided in Section
|
||||
10.3, no one other than the license steward has the right to modify or
|
||||
publish new versions of this License. Each version will be given a
|
||||
distinguishing version number.
|
||||
|
||||
10.2. Effect of New Versions
|
||||
|
||||
You may distribute the Covered Software under the terms of the version
|
||||
of the License under which You originally received the Covered Software,
|
||||
or under the terms of any subsequent version published by the license
|
||||
steward.
|
||||
|
||||
10.3. Modified Versions
|
||||
|
||||
If you create software not governed by this License, and you want to
|
||||
create a new license for such software, you may create and use a
|
||||
modified version of this License if you rename the license and remove
|
||||
any references to the name of the license steward (except to note that
|
||||
such modified license differs from this License).
|
||||
|
||||
10.4. Distributing Source Code Form that is Incompatible With Secondary
|
||||
Licenses
|
||||
|
||||
If You choose to distribute Source Code Form that is Incompatible With
|
||||
Secondary Licenses under the terms of this version of the License, the
|
||||
notice described in Exhibit B of this License must be attached.
|
||||
|
||||
Exhibit A - Source Code Form License Notice
|
||||
-------------------------------------------
|
||||
|
||||
This Source Code Form is subject to the terms of the Mozilla Public
|
||||
License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||
|
||||
If it is not possible or desirable to put the notice in a particular
|
||||
file, then You may include the notice in a location (such as a LICENSE
|
||||
file in a relevant directory) where a recipient would be likely to look
|
||||
for such a notice.
|
||||
|
||||
You may add additional accurate notices of copyright ownership.
|
||||
|
||||
Exhibit B - "Incompatible With Secondary Licenses" Notice
|
||||
---------------------------------------------------------
|
||||
|
||||
This Source Code Form is "Incompatible With Secondary Licenses", as
|
||||
defined by the Mozilla Public License, v. 2.0.
|
||||
|
47
README.md
Normal file
47
README.md
Normal file
|
@ -0,0 +1,47 @@
|
|||
# Maunium sticker picker
|
||||
A fast and simple Matrix sticker picker widget. Tested on Element Web & Android.
|
||||
|
||||
## Importing packs from Telegram
|
||||
1. (Optional) Set up a virtual environment.
|
||||
1. Create with `virtualenv -p python3 .`
|
||||
2. Activate with `source ./bin/activate`
|
||||
2. Install dependencies with `pip install -r requirements.txt`
|
||||
3. Copy `example-config.json` to `config.json` and set your homeserver URL and access token
|
||||
(used for uploading stickers to Matrix).
|
||||
4. Run `python3 import.py <pack urls...>`
|
||||
* On the first run, it'll prompt you to log in with a bot token or a telegram account.
|
||||
The session data is stored in `sticker-import.session` by default.
|
||||
* By default, the pack data will be written to `web/packs/`.
|
||||
* You can pass as many pack URLs as you want.
|
||||
* You can re-run the command with the same URLs to update packs.
|
||||
|
||||
If you want to list the URLs of all your saved packs, use `python3 import.py --list`.
|
||||
This requires logging in with your account instead of a bot token.
|
||||
|
||||
## Enabling the sticker widget
|
||||
1. Serve everything under `web/` using your webserver of choice. Make sure not to serve the
|
||||
top-level data, as `config.json` and the Telethon session file contain sensitive data.
|
||||
2. Using `/devtools` in Element Web, edit the `m.widgets` account data event to have the following content:
|
||||
|
||||
```json
|
||||
{
|
||||
"stickerpicker": {
|
||||
"content": {
|
||||
"type": "m.stickerpicker",
|
||||
"url": "https://your.sticker.picker.url/index.html",
|
||||
"name": "Stickerpicker",
|
||||
"data": {}
|
||||
},
|
||||
"sender": "@you:picker.url",
|
||||
"state_key": "stickerpicker",
|
||||
"type": "m.widget",
|
||||
"id": "stickerpicker"
|
||||
}
|
||||
}
|
||||
```
|
||||
|
||||
If you do not yet have a `m.widgets` event, simply create it with that content.
|
||||
You can also [use the client-server API directly][1] instead of using Element Web.
|
||||
3. Open the sticker picker and enjoy the fast sticker picking experience.
|
||||
|
||||
[1]: https://matrix.org/docs/spec/client_server/latest#put-matrix-client-r0-user-userid-account-data-type
|
4
example-config.json
Normal file
4
example-config.json
Normal file
|
@ -0,0 +1,4 @@
|
|||
{
|
||||
"homeserver": "https://example.com",
|
||||
"access_token": "foo"
|
||||
}
|
194
import.py
Normal file
194
import.py
Normal file
|
@ -0,0 +1,194 @@
|
|||
# Copyright (c) 2020 Tulir Asokan
|
||||
#
|
||||
# This Source Code Form is subject to the terms of the Mozilla Public
|
||||
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||
from typing import Dict, TypedDict
|
||||
from io import BytesIO
|
||||
import argparse
|
||||
import os.path
|
||||
import asyncio
|
||||
import json
|
||||
import re
|
||||
|
||||
from aiohttp import ClientSession
|
||||
from yarl import URL
|
||||
from PIL import Image
|
||||
|
||||
from telethon import TelegramClient
|
||||
from telethon.tl.functions.messages import GetAllStickersRequest, GetStickerSetRequest
|
||||
from telethon.tl.types.messages import AllStickers
|
||||
from telethon.tl.types import InputStickerSetShortName, Document
|
||||
from telethon.tl.types.messages import StickerSet as StickerSetFull
|
||||
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument("--list", help="List your saved sticker packs", action="store_true")
|
||||
parser.add_argument("--session", help="Telethon session file name", default="sticker-import")
|
||||
parser.add_argument("--config", help="Path to JSON file with Matrix homeserver and access_token",
|
||||
type=str, default="config.json")
|
||||
parser.add_argument("--output-dir", help="Directory to write packs to", default="web/packs/",
|
||||
type=str)
|
||||
parser.add_argument("pack", help="Sticker pack URLs to import", action="append", nargs="*")
|
||||
args = parser.parse_args()
|
||||
|
||||
with open(args.config) as config_file:
|
||||
config = json.load(config_file)
|
||||
homeserver_url = config["homeserver"]
|
||||
upload_url = URL(homeserver_url) / "_matrix" / "media" / "r0" / "upload"
|
||||
access_token = config["access_token"]
|
||||
|
||||
|
||||
async def upload(data: bytes, mimetype: str, filename: str) -> str:
|
||||
url = upload_url.with_query({"filename": filename})
|
||||
headers = {"Content-Type": mimetype, "Authorization": f"Bearer {access_token}"}
|
||||
async with ClientSession() as sess, sess.post(url, data=data, headers=headers) as resp:
|
||||
return (await resp.json())["content_uri"]
|
||||
|
||||
|
||||
class MatrixMediaInfo(TypedDict):
|
||||
w: int
|
||||
h: int
|
||||
size: int
|
||||
mimetype: str
|
||||
|
||||
|
||||
class MatrixStickerInfo(TypedDict, total=False):
|
||||
body: str
|
||||
url: str
|
||||
info: MatrixMediaInfo
|
||||
|
||||
|
||||
def convert_image(data: bytes) -> (bytes, int, int):
|
||||
image: Image.Image = Image.open(BytesIO(data)).convert("RGBA")
|
||||
image.thumbnail((256, 256), Image.ANTIALIAS)
|
||||
new_file = BytesIO()
|
||||
image.save(new_file, "png")
|
||||
w, h = image.size
|
||||
return new_file.getvalue(), w, h
|
||||
|
||||
|
||||
async def reupload_document(client: TelegramClient, document: Document) -> MatrixStickerInfo:
|
||||
print(f"Reuploading {document.id}", end="", flush=True)
|
||||
data = await client.download_media(document, file=bytes)
|
||||
print(".", end="", flush=True)
|
||||
data, width, height = convert_image(data)
|
||||
print(".", end="", flush=True)
|
||||
mxc = await upload(data, "image/png", f"{document.id}.png")
|
||||
print(".", flush=True)
|
||||
return {
|
||||
"body": "",
|
||||
"url": mxc,
|
||||
"info": {
|
||||
"w": width,
|
||||
"h": height,
|
||||
"size": len(data),
|
||||
"mimetype": "image/png",
|
||||
},
|
||||
}
|
||||
|
||||
|
||||
def add_to_index(name: str) -> None:
|
||||
index_path = os.path.join(args.output_dir, "index.json")
|
||||
try:
|
||||
with open(index_path) as index_file:
|
||||
index_data = json.load(index_file)
|
||||
except (FileNotFoundError, json.JSONDecodeError):
|
||||
index_data = {"packs": [], "homeserver_url": homeserver_url}
|
||||
if name not in index_data["packs"]:
|
||||
index_data["packs"].append(name)
|
||||
with open(index_path, "w") as index_file:
|
||||
json.dump(index_data, index_file, indent=" ")
|
||||
print(f"Added {name} to {index_path}")
|
||||
|
||||
|
||||
async def reupload_pack(client: TelegramClient, pack: StickerSetFull) -> None:
|
||||
if pack.set.animated:
|
||||
print("Animated stickerpacks are currently not supported")
|
||||
return
|
||||
|
||||
pack_path = os.path.join(args.output_dir, f"{pack.set.short_name}.json")
|
||||
try:
|
||||
os.mkdir(os.path.dirname(pack_path))
|
||||
except FileExistsError:
|
||||
pass
|
||||
|
||||
print(f"Reuploading {pack.set.title} with {pack.set.count} stickers "
|
||||
f"and writing output to {pack_path}")
|
||||
|
||||
already_uploaded = {}
|
||||
try:
|
||||
with open(pack_path) as pack_file:
|
||||
existing_pack = json.load(pack_file)
|
||||
already_uploaded = {sticker["net.maunium.telegram.sticker"]["id"]: sticker
|
||||
for sticker in existing_pack["stickers"]}
|
||||
print(f"Found {len(already_uploaded)} already reuploaded stickers")
|
||||
except FileNotFoundError:
|
||||
pass
|
||||
|
||||
reuploaded_documents: Dict[int, MatrixStickerInfo] = {}
|
||||
for document in pack.documents:
|
||||
try:
|
||||
reuploaded_documents[document.id] = already_uploaded[document.id]
|
||||
print(f"Skipped reuploading {document.id}")
|
||||
except KeyError:
|
||||
reuploaded_documents[document.id] = await reupload_document(client, document)
|
||||
|
||||
for sticker in pack.packs:
|
||||
for document_id in sticker.documents:
|
||||
doc = reuploaded_documents[document_id]
|
||||
doc["body"] = sticker.emoticon
|
||||
doc["net.maunium.telegram.sticker"] = {
|
||||
"pack": {
|
||||
"id": pack.set.id,
|
||||
"short_name": pack.set.short_name,
|
||||
},
|
||||
"id": document_id,
|
||||
"emoticon": sticker.emoticon,
|
||||
}
|
||||
|
||||
with open(pack_path, "w") as pack_file:
|
||||
json.dump({
|
||||
"title": pack.set.title,
|
||||
"short_name": pack.set.short_name,
|
||||
"id": pack.set.id,
|
||||
"hash": pack.set.hash,
|
||||
"stickers": list(reuploaded_documents.values()),
|
||||
}, pack_file, ensure_ascii=False)
|
||||
|
||||
add_to_index(os.path.basename(pack_path))
|
||||
|
||||
|
||||
pack_url_regex = re.compile(r"^(?:(?:https?://)?(?:t|telegram)\.(?:me|dog)/addstickers/)?"
|
||||
r"([A-Za-z0-9-_]+)"
|
||||
r"(?:\.json)?$")
|
||||
|
||||
|
||||
async def main():
|
||||
client = TelegramClient(args.session, 298751, "cb676d6bae20553c9996996a8f52b4d7")
|
||||
await client.start()
|
||||
|
||||
if args.list:
|
||||
stickers: AllStickers = await client(GetAllStickersRequest(hash=0))
|
||||
index = 1
|
||||
width = len(str(stickers.sets))
|
||||
print("Your saved sticker packs:")
|
||||
for saved_pack in stickers.sets:
|
||||
print(f"{index:>{width}}. {saved_pack.title} "
|
||||
f"(t.me/addstickers/{saved_pack.short_name})")
|
||||
elif args.pack[0]:
|
||||
input_packs = []
|
||||
for pack_url in args.pack[0]:
|
||||
match = pack_url_regex.match(pack_url)
|
||||
if not match:
|
||||
print(f"'{pack_url}' doesn't look like a sticker pack URL")
|
||||
return
|
||||
input_packs.append(InputStickerSetShortName(short_name=match.group(1)))
|
||||
for input_pack in input_packs:
|
||||
pack: StickerSetFull = await client(GetStickerSetRequest(input_pack))
|
||||
await reupload_pack(client, pack)
|
||||
print(f"Saved {pack.set.title} as {pack.set.short_name}.json")
|
||||
else:
|
||||
parser.print_help()
|
||||
|
||||
|
||||
asyncio.run(main())
|
5
requirements.txt
Normal file
5
requirements.txt
Normal file
|
@ -0,0 +1,5 @@
|
|||
aiohttp
|
||||
yarl
|
||||
pillow
|
||||
telethon
|
||||
cryptg
|
64
web/index.css
Normal file
64
web/index.css
Normal file
|
@ -0,0 +1,64 @@
|
|||
/*
|
||||
Copyright (c) 2020 Tulir Asokan
|
||||
|
||||
This Source Code Form is subject to the terms of the Mozilla Public
|
||||
License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||
*/
|
||||
|
||||
* {
|
||||
font-family: sans-serif;
|
||||
}
|
||||
|
||||
html {
|
||||
scrollbar-width: none;
|
||||
}
|
||||
|
||||
body {
|
||||
margin: 0;
|
||||
}
|
||||
|
||||
.main:not(.pack-list) {
|
||||
margin: 2rem;
|
||||
}
|
||||
|
||||
.main.empty {
|
||||
text-align: center;
|
||||
}
|
||||
|
||||
.stickerpack > .sticker-list {
|
||||
display: flex;
|
||||
flex-wrap: wrap;
|
||||
}
|
||||
|
||||
.stickerpack > h1 {
|
||||
margin: .75rem;
|
||||
}
|
||||
|
||||
.sticker {
|
||||
display: flex;
|
||||
padding: 4px;
|
||||
cursor: pointer;
|
||||
position: relative;
|
||||
width: 25vw;
|
||||
height: 25vw;
|
||||
box-sizing: border-box;
|
||||
}
|
||||
|
||||
.sticker:hover {
|
||||
background-color: #eee;
|
||||
}
|
||||
|
||||
.sticker > img {
|
||||
display: none;
|
||||
width: 100%;
|
||||
object-fit: contain;
|
||||
}
|
||||
|
||||
.sticker > img.visible {
|
||||
display: initial;
|
||||
}
|
||||
|
||||
h1 {
|
||||
font-size: 1rem;
|
||||
}
|
14
web/index.html
Normal file
14
web/index.html
Normal file
|
@ -0,0 +1,14 @@
|
|||
<!DOCTYPE html>
|
||||
<html lang="en">
|
||||
<head>
|
||||
<meta charset="utf-8">
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1, shrink-to-fit=no">
|
||||
<title>Maunium sticker picker</title>
|
||||
<link rel="stylesheet" href="index.css"/>
|
||||
<link rel="stylesheet" href="spinner.css"/>
|
||||
</head>
|
||||
<body>
|
||||
<noscript>This sticker picker requires JavaScript</noscript>
|
||||
<script src="index.js" type="module"></script>
|
||||
</body>
|
||||
</html>
|
175
web/index.js
Normal file
175
web/index.js
Normal file
|
@ -0,0 +1,175 @@
|
|||
// Copyright (c) 2020 Tulir Asokan
|
||||
//
|
||||
// This Source Code Form is subject to the terms of the Mozilla Public
|
||||
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||
import { html, render, Component } from "https://unpkg.com/htm/preact/index.mjs?module"
|
||||
|
||||
// The base URL for fetching packs. The app will first fetch ${PACK_BASE_URL}/index.json,
|
||||
// then ${PACK_BASE_URL}/${packFile} for each packFile in the packs object of the index.json file.
|
||||
const PACKS_BASE_URL = "packs"
|
||||
// This is updated from packs/index.json
|
||||
let HOMESERVER_URL = "https://matrix-client.matrix.org"
|
||||
|
||||
const makeThumbnailURL = mxc => `${HOMESERVER_URL}/_matrix/media/r0/thumbnail/${mxc.substr(6)}?height=128&width=128&method=scale`
|
||||
|
||||
class App extends Component {
|
||||
constructor(props) {
|
||||
super(props)
|
||||
this.state = {
|
||||
packs: [],
|
||||
loading: true,
|
||||
error: null,
|
||||
}
|
||||
this.observer = null
|
||||
}
|
||||
|
||||
observeIntersection = intersections => {
|
||||
for (const entry of intersections) {
|
||||
const img = entry.target.children.item(0)
|
||||
if (entry.isIntersecting) {
|
||||
img.setAttribute("src", img.getAttribute("data-src"))
|
||||
img.classList.add("visible")
|
||||
} else {
|
||||
img.removeAttribute("src")
|
||||
img.classList.remove("visible")
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
componentDidMount() {
|
||||
fetch(`${PACKS_BASE_URL}/index.json`).then(async indexRes => {
|
||||
if (indexRes.status >= 400) {
|
||||
this.setState({
|
||||
loading: false,
|
||||
error: indexRes.status !== 404 ? indexRes.statusText : null,
|
||||
})
|
||||
return
|
||||
}
|
||||
const indexData = await indexRes.json()
|
||||
HOMESERVER_URL = indexData.homeserver_url || HOMESERVER_URL
|
||||
// TODO only load pack metadata when scrolled into view?
|
||||
for (const packFile of indexData.packs) {
|
||||
const packRes = await fetch(`${PACKS_BASE_URL}/${packFile}`)
|
||||
const packData = await packRes.json()
|
||||
this.setState({
|
||||
packs: [...this.state.packs, packData],
|
||||
loading: false,
|
||||
})
|
||||
}
|
||||
}, error => this.setState({ loading: false, error }))
|
||||
this.observer = new IntersectionObserver(this.observeIntersection, {
|
||||
rootMargin: "100px",
|
||||
threshold: 0,
|
||||
})
|
||||
}
|
||||
|
||||
componentDidUpdate() {
|
||||
for (const elem of document.getElementsByClassName("sticker")) {
|
||||
this.observer.observe(elem)
|
||||
}
|
||||
}
|
||||
|
||||
componentWillUnmount() {
|
||||
this.observer.disconnect()
|
||||
}
|
||||
|
||||
render() {
|
||||
if (this.state.loading) {
|
||||
return html`<div class="main spinner"><${Spinner} size=${80} green /></div>`
|
||||
} else if (this.state.error) {
|
||||
return html`<div class="main error">
|
||||
<h1>Failed to load packs</h1>
|
||||
<p>${this.state.error}</p>
|
||||
</div>`
|
||||
} else if (this.state.packs.length === 0) {
|
||||
return html`<div class="main empty"><h1>No packs found :(</h1></div>`
|
||||
}
|
||||
return html`<div class="main pack-list">
|
||||
${this.state.packs.map(pack => html`<${Pack} id=${pack.id} ...${pack}/>`)}
|
||||
</div>`
|
||||
}
|
||||
}
|
||||
|
||||
const Spinner = ({ size = 40, noCenter = false, noMargin = false, green = false }) => {
|
||||
let margin = 0
|
||||
if (!isNaN(+size)) {
|
||||
size = +size
|
||||
margin = noMargin ? 0 : `${Math.round(size / 6)}px`
|
||||
size = `${size}px`
|
||||
}
|
||||
const noInnerMargin = !noCenter || !margin
|
||||
const comp = html`
|
||||
<div style="width: ${size}; height: ${size}; margin: ${noInnerMargin ? 0 : margin} 0;"
|
||||
class="sk-chase ${green && "green"}">
|
||||
<div class="sk-chase-dot" />
|
||||
<div class="sk-chase-dot" />
|
||||
<div class="sk-chase-dot" />
|
||||
<div class="sk-chase-dot" />
|
||||
<div class="sk-chase-dot" />
|
||||
<div class="sk-chase-dot" />
|
||||
</div>
|
||||
`
|
||||
if (!noCenter) {
|
||||
return html`<div style="margin: ${margin} 0;" class="sk-center-wrapper">${comp}</div>`
|
||||
}
|
||||
return comp
|
||||
}
|
||||
|
||||
const Pack = ({ title, stickers }) => html`<div class="stickerpack">
|
||||
<h1>${title}</h1>
|
||||
<div class="sticker-list">
|
||||
${stickers.map(sticker => html`
|
||||
<${Sticker} key=${sticker["net.maunium.telegram.sticker"].id} content=${sticker}/>
|
||||
`)}
|
||||
</div>
|
||||
</div>`
|
||||
|
||||
const Sticker = ({ content }) => html`<div class="sticker" onClick=${() => sendSticker(content)}>
|
||||
<img data-src=${makeThumbnailURL(content.url)} alt=${content.body} />
|
||||
</div>`
|
||||
|
||||
function sendSticker(content) {
|
||||
window.parent.postMessage({
|
||||
api: "fromWidget",
|
||||
action: "m.sticker",
|
||||
requestId: `sticker-${Date.now()}`,
|
||||
widgetId,
|
||||
data: {
|
||||
name: content.body,
|
||||
content,
|
||||
},
|
||||
}, "*")
|
||||
}
|
||||
|
||||
let widgetId = null
|
||||
|
||||
window.onmessage = event => {
|
||||
if (!window.parent || !event.data) {
|
||||
return
|
||||
}
|
||||
|
||||
const request = event.data
|
||||
if (!request.requestId || !request.widgetId || !request.action || request.api !== "toWidget") {
|
||||
return
|
||||
}
|
||||
|
||||
if (widgetId) {
|
||||
if (widgetId !== request.widgetId) {
|
||||
return
|
||||
}
|
||||
} else {
|
||||
widgetId = request.widgetId
|
||||
}
|
||||
|
||||
window.parent.postMessage({
|
||||
...request,
|
||||
response: request.action === "capabilities" ? {
|
||||
capabilities: ["m.sticker"],
|
||||
} : {
|
||||
error: { message: "Action not supported" },
|
||||
},
|
||||
}, event.origin)
|
||||
}
|
||||
|
||||
render(html`<${App} />`, document.body)
|
64
web/spinner.css
Normal file
64
web/spinner.css
Normal file
|
@ -0,0 +1,64 @@
|
|||
/* Chase spinner from https://tobiasahlin.com/spinkit/. MIT license */
|
||||
.sk-center-wrapper {
|
||||
width: 100%;
|
||||
display: flex;
|
||||
justify-content: space-around;
|
||||
}
|
||||
|
||||
.sk-chase {
|
||||
position: relative;
|
||||
animation: sk-chase 2.5s infinite linear both;
|
||||
}
|
||||
|
||||
.sk-chase-dot {
|
||||
width: 100%;
|
||||
height: 100%;
|
||||
position: absolute;
|
||||
left: 0;
|
||||
top: 0;
|
||||
animation: sk-chase-dot 2.0s infinite ease-in-out both;
|
||||
}
|
||||
|
||||
.sk-chase-dot:before {
|
||||
content: '';
|
||||
display: block;
|
||||
width: 25%;
|
||||
height: 25%;
|
||||
border-radius: 100%;
|
||||
animation: sk-chase-dot-before 2.0s infinite ease-in-out both;
|
||||
background-color: #FFF;
|
||||
}
|
||||
|
||||
.sk-chase.green > .sk-chase-dot:before {
|
||||
background-color: #00C853;
|
||||
}
|
||||
|
||||
.sk-chase-dot:nth-child(1) { animation-delay: -1.1s; }
|
||||
.sk-chase-dot:nth-child(2) { animation-delay: -1.0s; }
|
||||
.sk-chase-dot:nth-child(3) { animation-delay: -0.9s; }
|
||||
.sk-chase-dot:nth-child(4) { animation-delay: -0.8s; }
|
||||
.sk-chase-dot:nth-child(5) { animation-delay: -0.7s; }
|
||||
.sk-chase-dot:nth-child(6) { animation-delay: -0.6s; }
|
||||
.sk-chase-dot:nth-child(1):before { animation-delay: -1.1s; }
|
||||
.sk-chase-dot:nth-child(2):before { animation-delay: -1.0s; }
|
||||
.sk-chase-dot:nth-child(3):before { animation-delay: -0.9s; }
|
||||
.sk-chase-dot:nth-child(4):before { animation-delay: -0.8s; }
|
||||
.sk-chase-dot:nth-child(5):before { animation-delay: -0.7s; }
|
||||
.sk-chase-dot:nth-child(6):before { animation-delay: -0.6s; }
|
||||
|
||||
@keyframes sk-chase {
|
||||
100% { transform: rotate(360deg); }
|
||||
}
|
||||
|
||||
@keyframes sk-chase-dot {
|
||||
80%, 100% { transform: rotate(360deg); }
|
||||
}
|
||||
|
||||
@keyframes sk-chase-dot-before {
|
||||
50% {
|
||||
transform: scale(0.4);
|
||||
}
|
||||
100%, 0% {
|
||||
transform: scale(1.0);
|
||||
}
|
||||
}
|
Loading…
Reference in a new issue