diff --git a/ee/connectors/msgcodec/messages.py b/ee/connectors/msgcodec/messages.py index b500424b7..e4013d878 100644 --- a/ee/connectors/msgcodec/messages.py +++ b/ee/connectors/msgcodec/messages.py @@ -15,6 +15,25 @@ class BatchMeta(Message): self.timestamp = timestamp +class BatchMetadata(Message): + __id__ = 81 + + def __init__(self, version, page_no, first_index, timestamp, location): + self.version = version + self.page_no = page_no + self.first_index = first_index + self.timestamp = timestamp + self.location = location + + +class PartitionedMessage(Message): + __id__ = 82 + + def __init__(self, part_no, part_total): + self.part_no = part_no + self.part_total = part_total + + class Timestamp(Message): __id__ = 0 @@ -44,13 +63,6 @@ class SessionStart(Message): self.user_id = user_id -class SessionDisconnect(Message): - __id__ = 2 - - def __init__(self, timestamp): - self.timestamp = timestamp - - class SessionEnd(Message): __id__ = 3 @@ -637,13 +649,6 @@ class CustomIssue(Message): self.payload = payload -class PageClose(Message): - __id__ = 65 - - def __init__(self, ): - - - class AssetCache(Message): __id__ = 66 @@ -679,6 +684,66 @@ class CreateIFrameDocument(Message): self.id = id +class AdoptedSSReplaceURLBased(Message): + __id__ = 71 + + def __init__(self, sheet_id, text, base_url): + self.sheet_id = sheet_id + self.text = text + self.base_url = base_url + + +class AdoptedSSReplace(Message): + __id__ = 72 + + def __init__(self, sheet_id, text): + self.sheet_id = sheet_id + self.text = text + + +class AdoptedSSInsertRuleURLBased(Message): + __id__ = 73 + + def __init__(self, sheet_id, rule, index, base_url): + self.sheet_id = sheet_id + self.rule = rule + self.index = index + self.base_url = base_url + + +class AdoptedSSInsertRule(Message): + __id__ = 74 + + def __init__(self, sheet_id, rule, index): + self.sheet_id = sheet_id + self.rule = rule + self.index = index + + +class AdoptedSSDeleteRule(Message): + __id__ = 75 + + def __init__(self, sheet_id, index): + self.sheet_id = sheet_id + self.index = index + + +class AdoptedSSAddOwner(Message): + __id__ = 76 + + def __init__(self, sheet_id, id): + self.sheet_id = sheet_id + self.id = id + + +class AdoptedSSRemoveOwner(Message): + __id__ = 77 + + def __init__(self, sheet_id, id): + self.sheet_id = sheet_id + self.id = id + + class IOSBatchMeta(Message): __id__ = 107 diff --git a/ee/connectors/msgcodec/msgcodec.py b/ee/connectors/msgcodec/msgcodec.py index 3bd74499c..d603cfe9c 100644 --- a/ee/connectors/msgcodec/msgcodec.py +++ b/ee/connectors/msgcodec/msgcodec.py @@ -26,6 +26,21 @@ class MessageCodec(Codec): timestamp=self.read_int(reader) ) + if message_id == 81: + return BatchMetadata( + version=self.read_uint(reader), + page_no=self.read_uint(reader), + first_index=self.read_uint(reader), + timestamp=self.read_int(reader), + location=self.read_string(reader) + ) + + if message_id == 82: + return PartitionedMessage( + part_no=self.read_uint(reader), + part_total=self.read_uint(reader) + ) + if message_id == 0: return Timestamp( timestamp=self.read_uint(reader) @@ -51,11 +66,6 @@ class MessageCodec(Codec): user_id=self.read_string(reader) ) - if message_id == 2: - return SessionDisconnect( - timestamp=self.read_uint(reader) - ) - if message_id == 3: return SessionEnd( timestamp=self.read_uint(reader) @@ -522,11 +532,6 @@ class MessageCodec(Codec): payload=self.read_string(reader) ) - if message_id == 65: - return PageClose( - - ) - if message_id == 66: return AssetCache( url=self.read_string(reader) @@ -554,6 +559,52 @@ class MessageCodec(Codec): id=self.read_uint(reader) ) + if message_id == 71: + return AdoptedSSReplaceURLBased( + sheet_id=self.read_uint(reader), + text=self.read_string(reader), + base_url=self.read_string(reader) + ) + + if message_id == 72: + return AdoptedSSReplace( + sheet_id=self.read_uint(reader), + text=self.read_string(reader) + ) + + if message_id == 73: + return AdoptedSSInsertRuleURLBased( + sheet_id=self.read_uint(reader), + rule=self.read_string(reader), + index=self.read_uint(reader), + base_url=self.read_string(reader) + ) + + if message_id == 74: + return AdoptedSSInsertRule( + sheet_id=self.read_uint(reader), + rule=self.read_string(reader), + index=self.read_uint(reader) + ) + + if message_id == 75: + return AdoptedSSDeleteRule( + sheet_id=self.read_uint(reader), + index=self.read_uint(reader) + ) + + if message_id == 76: + return AdoptedSSAddOwner( + sheet_id=self.read_uint(reader), + id=self.read_uint(reader) + ) + + if message_id == 77: + return AdoptedSSRemoveOwner( + sheet_id=self.read_uint(reader), + id=self.read_uint(reader) + ) + if message_id == 107: return IOSBatchMeta( timestamp=self.read_uint(reader),