add type stubs for static type checking support

pyproject.toml

@@ -67,7 +67,10 @@ dev = [
 
 [tool.setuptools]
 license-files = []
-package-dir = { sc2 = "sc2" }
+package-dir = { sc2 = "sc2", s2clientprotocol = "s2clientprotocol" }
+
+[tool.setuptools.package-data]
+sc2 = ["py.typed", "*.pyi"]
 
 [build-system]
 # https://packaging.python.org/en/latest/tutorials/packaging-projects/#choosing-a-build-backend

s2clientprotocol/common_pb2.pyi

@@ -0,0 +1,26 @@
+# https://github.com/Blizzard/s2client-proto/blob/bff45dae1fc685e6acbaae084670afb7d1c0832c/s2clientprotocol/common.proto
+from enum import Enum
+from google.protobuf.message import Message
+
+class PointI(Message):
+    x: int
+    y: int
+    def __init__(self, x: int = ..., y: int = ...) -> None: ...
+
+class Point2D(Message):
+    x: float
+    y: float
+    def __init__(self, x: float = ..., y: float = ...) -> None: ...
+
+class Point(Message):
+    x: float
+    y: float
+    z: float
+    def __init__(self, x: float = ..., y: float = ..., z: float = ...) -> None: ...
+
+class Race(Enum):
+    NoRace: int
+    Terran: int
+    Zerg: int
+    Protoss: int
+    Random: int

sc2/constants.py

@@ -495,7 +495,7 @@ def return_NOTAUNIT() -> UnitTypeId:
     UnitTypeId.EXTRACTORRICH,
 }
 # pyre-ignore[11]
-DAMAGE_BONUS_PER_UPGRADE: dict[UnitTypeId, dict[TargetType, Any]] = {
+DAMAGE_BONUS_PER_UPGRADE: dict[UnitTypeId, dict[int, Any]] = {
     #
     # Protoss
     #

sc2/data.py

@@ -1,10 +1,7 @@
 # pyre-ignore-all-errors[16, 19]
 """For the list of enums, see here
 
-https://github.com/Blizzard/s2client-api/blob/d9ba0a33d6ce9d233c2a4ee988360c188fbe9dbf/include/sc2api/sc2_gametypes.h
-https://github.com/Blizzard/s2client-api/blob/d9ba0a33d6ce9d233c2a4ee988360c188fbe9dbf/include/sc2api/sc2_action.h
-https://github.com/Blizzard/s2client-api/blob/d9ba0a33d6ce9d233c2a4ee988360c188fbe9dbf/include/sc2api/sc2_unit.h
-https://github.com/Blizzard/s2client-api/blob/d9ba0a33d6ce9d233c2a4ee988360c188fbe9dbf/include/sc2api/sc2_data.h
+https://github.com/Blizzard/s2client-proto/tree/bff45dae1fc685e6acbaae084670afb7d1c0832c/s2clientprotocol
 """
 
 from __future__ import annotations