Files
OSGKeyboard/Scripts/generate_extract_events_shortcut.py
T
Rocky 320dc777c9 feat(keyboard): add Notes skill and system-style typing
Add save-to-Notes export and direct map handoff while aligning multi-touch typing, period shortcuts, return actions, and navigation visuals with system behavior.
2026-08-14 18:22:41 +08:00

274 lines
9.1 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
#!/usr/bin/env python3
"""Build the companion Shortcut for the Extract Events skill.
Writes an unsigned binary plist that `shortcuts sign` can notarize.
Workflow (must materialize Shortcut Input as Text first — Split Text's
WFInput/ExtensionInput binding finishes the run without iterating):
Text (Shortcut Input)
→ Split Text (new lines)
→ Repeat Each
→ Split Text (|)
→ Get items 14 (start, end, title, location)
→ If end == ALLDAY
→ Add New Event (all-day)
Otherwise
→ Add New Event (timed)
"""
from __future__ import annotations
import plistlib
from pathlib import Path
ROOT = Path(__file__).resolve().parents[1]
OUT_DIR = ROOT / "OSGKeyboard" / "Resources" / "Shortcuts"
UNSIGNED = OUT_DIR / "OSGExtractEvents.unsigned.shortcut"
SIGNED = OUT_DIR / "OSGExtractEvents.shortcut"
SHORTCUT_NAME = "OSGExtractEvents"
OBJECT_REPLACEMENT = "\ufffc"
TEXT_UUID = "A1B2C3D0-1111-4E00-8A5A-FD7E1B0C396F"
SPLIT_LINES_UUID = "A1B2C3D1-2222-4F11-9A6B-0E8F2C1D4A70"
REPEAT_GROUP = "A1B2C3D2-3333-4E02-B3C4-91F0E6A7B812"
REPEAT_START = "A1B2C3D3-4444-4F13-A4D5-02A1F7B8C923"
SPLIT_FIELDS_UUID = "A1B2C3D4-5555-4014-B5E6-13B2A8C9D034"
START_UUID = "A1B2C3D5-6666-4125-C6F7-24C3B9DAE145"
END_UUID = "A1B2C3D6-7777-4236-D7A8-35D4CAEBF256"
TITLE_UUID = "A1B2C3D7-8888-4347-E8B9-46E5DBF0C367"
LOCATION_UUID = "A1B2C3D8-9999-4458-F9CA-57F6EC01D478"
IF_GROUP = "A1B2C3D9-AAAA-4569-0ADB-6807FD12E589"
IF_START = "A1B2C3DA-BBBB-467A-1BEC-79180E23F69A"
EVENT_ALLDAY_UUID = "A1B2C3DB-CCCC-478B-2CFD-8A291F3407AB"
IF_ELSE = "A1B2C3DC-DDDD-489C-3D0E-9B3A204518BC"
EVENT_TIMED_UUID = "A1B2C3DD-EEEE-49AD-4E1F-AC4B315629CD"
IF_END = "A1B2C3DE-FFFF-4ABE-5F20-BD5C42673ADE"
REPEAT_END = "A1B2C3DF-0000-4BCF-6031-CE6D53784BEF"
def attachment(value: dict) -> dict:
return {"Value": value, "WFSerializationType": "WFTextTokenAttachment"}
def token_string(attachment_value: dict) -> dict:
return {
"Value": {
"attachmentsByRange": {"{0, 1}": attachment_value},
"string": OBJECT_REPLACEMENT,
},
"WFSerializationType": "WFTextTokenString",
}
def shortcut_input_text() -> dict:
return token_string({"Type": "ExtensionInput"})
def action_output_text(output_name: str, output_uuid: str) -> dict:
return token_string(
{
"Aggrandizements": [],
"OutputName": output_name,
"OutputUUID": output_uuid,
"Type": "ActionOutput",
}
)
def action_output_attachment(output_name: str, output_uuid: str) -> dict:
return attachment(
{
"Aggrandizements": [],
"OutputName": output_name,
"OutputUUID": output_uuid,
"Type": "ActionOutput",
}
)
def named_variable_text(variable_name: str) -> dict:
return token_string(
{
"Aggrandizements": [],
"Type": "Variable",
"VariableName": variable_name,
}
)
def if_input(output_name: str, output_uuid: str) -> dict:
return {
"Type": "Variable",
"Variable": {
"Value": {
"Aggrandizements": [],
"OutputName": output_name,
"OutputUUID": output_uuid,
"Type": "ActionOutput",
},
"WFSerializationType": "WFTextTokenAttachment",
},
}
def get_item(uuid: str, name: str, index: int) -> dict:
return {
"WFWorkflowActionIdentifier": "is.workflow.actions.getitemfromlist",
"WFWorkflowActionParameters": {
"UUID": uuid,
"CustomOutputName": name,
"WFInput": action_output_attachment("字段", SPLIT_FIELDS_UUID),
"WFItemSpecifier": "Item At Index",
"WFItemIndex": index,
},
}
def add_event(
uuid: str,
*,
all_day: bool,
include_end: bool,
) -> dict:
parameters = {
"UUID": uuid,
"WFCalendarItemTitle": action_output_text("标题", TITLE_UUID),
"WFCalendarItemLocation": action_output_text("地点", LOCATION_UUID),
"WFCalendarItemDates": True,
"WFCalendarItemStartDate": action_output_text("开始", START_UUID),
"WFCalendarItemAllDay": all_day,
"WFCalendarItemShowComposer": False,
"ShowComposeSheet": False,
"WFShowWhenRun": False,
}
if include_end:
parameters["WFCalendarItemEndDate"] = action_output_text("结束", END_UUID)
return {
"WFWorkflowActionIdentifier": "is.workflow.actions.addnewevent",
"WFWorkflowActionParameters": parameters,
}
def workflow() -> dict:
return {
"WFWorkflowClientVersion": "3600.0.4",
"WFWorkflowClientRelease": "26.0",
"WFWorkflowMinimumClientVersion": 900,
"WFWorkflowMinimumClientVersionString": "900",
"WFWorkflowName": SHORTCUT_NAME,
"WFWorkflowHasShortcutInputVariables": True,
"WFWorkflowHasOutputFallback": False,
"WFWorkflowImportQuestions": [],
"WFWorkflowTypes": [],
"WFWorkflowInputContentItemClasses": [
"WFStringContentItem",
"WFRichTextContentItem",
],
"WFWorkflowOutputContentItemClasses": [],
"WFWorkflowNoInputBehavior": {
"Name": "WFWorkflowNoInputBehaviorAskForInput",
"Parameters": {
"ItemClass": "WFStringContentItem",
},
},
"WFWorkflowIcon": {
"WFWorkflowIconGlyphNumber": 59675,
"WFWorkflowIconStartColor": 4286828031,
},
"WFWorkflowActions": [
{
"WFWorkflowActionIdentifier": "is.workflow.actions.gettext",
"WFWorkflowActionParameters": {
"UUID": TEXT_UUID,
"CustomOutputName": "快捷指令文本",
"WFTextActionText": shortcut_input_text(),
},
},
{
"WFWorkflowActionIdentifier": "is.workflow.actions.text.split",
"WFWorkflowActionParameters": {
"UUID": SPLIT_LINES_UUID,
"CustomOutputName": "行",
"text": action_output_text("快捷指令文本", TEXT_UUID),
"WFTextSeparator": "New Lines",
},
},
{
"WFWorkflowActionIdentifier": "is.workflow.actions.repeat.each",
"WFWorkflowActionParameters": {
"UUID": REPEAT_START,
"GroupingIdentifier": REPEAT_GROUP,
"WFControlFlowMode": 0,
"WFInput": action_output_attachment("行", SPLIT_LINES_UUID),
},
},
{
"WFWorkflowActionIdentifier": "is.workflow.actions.text.split",
"WFWorkflowActionParameters": {
"UUID": SPLIT_FIELDS_UUID,
"CustomOutputName": "字段",
"text": named_variable_text("Repeat Item"),
"WFTextSeparator": "Custom",
"WFTextCustomSeparator": "|",
},
},
get_item(START_UUID, "开始", 1),
get_item(END_UUID, "结束", 2),
get_item(TITLE_UUID, "标题", 3),
get_item(LOCATION_UUID, "地点", 4),
{
"WFWorkflowActionIdentifier": "is.workflow.actions.conditional",
"WFWorkflowActionParameters": {
"UUID": IF_START,
"GroupingIdentifier": IF_GROUP,
"WFControlFlowMode": 0,
"WFCondition": 4,
"WFConditionalActionString": "ALLDAY",
"WFInput": if_input("结束", END_UUID),
},
},
add_event(EVENT_ALLDAY_UUID, all_day=True, include_end=False),
{
"WFWorkflowActionIdentifier": "is.workflow.actions.conditional",
"WFWorkflowActionParameters": {
"UUID": IF_ELSE,
"GroupingIdentifier": IF_GROUP,
"WFControlFlowMode": 1,
},
},
add_event(EVENT_TIMED_UUID, all_day=False, include_end=True),
{
"WFWorkflowActionIdentifier": "is.workflow.actions.conditional",
"WFWorkflowActionParameters": {
"UUID": IF_END,
"GroupingIdentifier": IF_GROUP,
"WFControlFlowMode": 2,
},
},
{
"WFWorkflowActionIdentifier": "is.workflow.actions.repeat.each",
"WFWorkflowActionParameters": {
"UUID": REPEAT_END,
"GroupingIdentifier": REPEAT_GROUP,
"WFControlFlowMode": 2,
},
},
],
}
def main() -> None:
OUT_DIR.mkdir(parents=True, exist_ok=True)
UNSIGNED.write_bytes(plistlib.dumps(workflow(), fmt=plistlib.FMT_BINARY))
print(f"wrote {UNSIGNED}")
print(
"sign with: shortcuts sign --mode anyone "
f"--input {UNSIGNED} --output {SIGNED}"
)
if __name__ == "__main__":
main()