Module: Clacky::Channel::Adapters::Feishu::WSClient::ProtoFrame

Defined in:
lib/clacky/server/channel/adapters/feishu/ws_client.rb

Overview

Minimal protobuf encoder/decoder for pbbp2.Frame. Fields: 1=SeqID(uint64), 2=LogID(uint64), 3=service(int32), 4=method(int32), 5=headers(repeated msg1=key,2=value), 8=payload(bytes)

Class Method Summary collapse

Class Method Details

.decode(buf) ⇒ Object 



321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
 
# File 'lib/clacky/server/channel/adapters/feishu/ws_client.rb', line 321

def self.decode(buf)
  buf = buf.b
  pos = 0
  result = { headers: {}, payload: "".b }

  while pos < buf.bytesize
    tag_byte, pos = read_varint(buf, pos)
    field_number = tag_byte >> 3
    wire_type = tag_byte & 0x7

    case wire_type
    when 0 # varint
      val, pos = read_varint(buf, pos)
      case field_number
      when 1 then result[:seq_id] = val
      when 2 then result[:log_id] = val
      when 3 then result[:service] = val
      when 4 then result[:method] = val
      end
    when 2 # length-delimited
      len, pos = read_varint(buf, pos)
      bytes = buf.byteslice(pos, len)
      pos += len
      case field_number
      when 5 # header entry
        k, v = decode_header(bytes)
        result[:headers][k] = v if k
      when 7 then result[:payload_type] = bytes.force_encoding("UTF-8")
      when 8 then result[:payload] = bytes
      end
    else
      break # unknown wire type, stop parsing
    end
  end

  result
end

.decode_header(buf) ⇒ Object 



359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
 
# File 'lib/clacky/server/channel/adapters/feishu/ws_client.rb', line 359

def self.decode_header(buf)
  buf = buf.b
  pos = 0
  key = nil
  val = nil
  while pos < buf.bytesize
    tag_byte, pos = read_varint(buf, pos)
    field_number = tag_byte >> 3
    wire_type = tag_byte & 0x7
    if wire_type == 2
      len, pos = read_varint(buf, pos)
      bytes = buf.byteslice(pos, len)
      pos += len
      case field_number
      when 1 then key = bytes.force_encoding("UTF-8")
      when 2 then val = bytes.force_encoding("UTF-8")
      end
    else
      break
    end
  end
  [key, val]
end

.encode(frame) ⇒ Object 



304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
 
# File 'lib/clacky/server/channel/adapters/feishu/ws_client.rb', line 304

def self.encode(frame)
  buf = "".b
  buf << encode_varint(1, frame[:seq_id] || 0)
  buf << encode_varint(2, frame[:log_id] || 0)
  buf << encode_varint(3, frame[:service] || 0)
  buf << encode_varint(4, frame[:method] || 0)
  (frame[:headers] || {}).each do |k, v|
    header_bytes = encode_string(1, k.to_s) + encode_string(2, v.to_s)
    buf << encode_length_delimited(5, header_bytes)
  end
  if frame[:payload]
    payload_bytes = frame[:payload].respond_to?(:b) ? frame[:payload].b : frame[:payload].to_s.b
    buf << encode_length_delimited(8, payload_bytes)
  end
  buf
end

.encode_length_delimited(field_number, bytes) ⇒ Object 



419
420
421
422
 
# File 'lib/clacky/server/channel/adapters/feishu/ws_client.rb', line 419

def self.encode_length_delimited(field_number, bytes)
  tag = (field_number << 3) | 2  # wire type 2
  encode_raw_varint(tag) + encode_raw_varint(bytes.bytesize) + bytes
end

.encode_raw_varint(value) ⇒ Object 



402
403
404
405
406
407
408
409
410
411
412
 
# File 'lib/clacky/server/channel/adapters/feishu/ws_client.rb', line 402

def self.encode_raw_varint(value)
  bytes = "".b
  loop do
    byte = value & 0x7F
    value >>= 7
    byte |= 0x80 if value > 0
    bytes << byte
    break if value == 0
  end
  bytes
end

.encode_string(field_number, str) ⇒ Object 



414
415
416
417
 
# File 'lib/clacky/server/channel/adapters/feishu/ws_client.rb', line 414

def self.encode_string(field_number, str)
  bytes = str.encode("UTF-8").b
  encode_length_delimited(field_number, bytes)
end

.encode_varint(field_number, value) ⇒ Object 



397
398
399
400
 
# File 'lib/clacky/server/channel/adapters/feishu/ws_client.rb', line 397

def self.encode_varint(field_number, value)
  tag = (field_number << 3) | 0  # wire type 0
  encode_raw_varint(tag) + encode_raw_varint(value)
end

.read_varint(buf, pos) ⇒ Object 



383
384
385
386
387
388
389
390
391
392
393
394
395
 
# File 'lib/clacky/server/channel/adapters/feishu/ws_client.rb', line 383

def self.read_varint(buf, pos)
  result = 0
  shift = 0
  loop do
    byte = buf.getbyte(pos)
    raise "unexpected end of buffer at pos #{pos}" if byte.nil?
    pos += 1
    result |= (byte & 0x7F) << shift
    break unless byte & 0x80 != 0
    shift += 7
  end
  [result, pos]
end