protoss/dump_service.go at master · Gophercraft/protoss · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
package protoss

import (
	"strconv"

	"github.com/Gophercraft/protoss/extensions/bgs/protocol"
	"google.golang.org/protobuf/proto"
	"google.golang.org/protobuf/types/descriptorpb"
)

func (fd *filedumper) dump_service_definition(service_descriptor *descriptorpb.ServiceDescriptorProto) (err error) {
	fd.printf("service %s {\n", service_descriptor.GetName())
	fd.line_indentation++

	if proto.HasExtension(service_descriptor.GetOptions(), protocol.E_SdkServiceOptions) {
		sdk_service_options := proto.GetExtension(service_descriptor.GetOptions(), protocol.E_SdkServiceOptions).(*protocol.SDKServiceOptions)
		if sdk_service_options.Inbound != nil {
			fd.indent()
			fd.printf("option (sdk_service_options).inbound = %t;\n", sdk_service_options.GetInbound())
		}
		if sdk_service_options.Outbound != nil {
			fd.indent()
			fd.printf("option (sdk_service_options).outbound = %t;\n", sdk_service_options.GetOutbound())
		}
		if sdk_service_options.UseClientId != nil {
			fd.indent()
			fd.printf("option (sdk_service_options).use_client_id = %t;\n", sdk_service_options.GetUseClientId())
		}
	}

	if proto.HasExtension(service_descriptor.GetOptions(), protocol.E_ServiceOptions) {
		service_options := proto.GetExtension(service_descriptor.GetOptions(), protocol.E_ServiceOptions).(*protocol.BGSServiceOptions)
		if service_options.DescriptorName != nil {
			fd.indent()
			fd.printf("option (service_options).descriptor_name = %s;\n", strconv.Quote(service_options.GetDescriptorName()))
		}

		if service_options.ShardName != nil {
			fd.indent()
			fd.printf("option (service_options).shard_name = %s;\n", strconv.Quote(service_options.GetShardName()))
		}
	}

	for _, rpc := range service_descriptor.Method {
		// decompile basic rpc type information
		method_name := rpc.GetName()
		argument_type := rpc.GetInputType()
		argument_type, err = fd.resolve_type_name_in_scope(argument_type)
		if err != nil {
			return
		}
		return_type := rpc.GetOutputType()
		return_type, err = fd.resolve_type_name_in_scope(return_type)
		if err != nil {
			return
		}

		stream_client_prefix := ""
		if rpc.GetClientStreaming() {
			stream_client_prefix = "stream "
		}
		stream_server_prefix := ""
		if rpc.GetServerStreaming() {
			stream_server_prefix = "stream "
		}

		fd.indent()
		fd.printf("rpc %s(%s%s) returns (%s%s) {",
			method_name,
			stream_client_prefix,
			argument_type,
			stream_server_prefix,
			return_type,
		)

		// decompile BGS options
		if proto.HasExtension(rpc.GetOptions(), protocol.E_MethodOptions) {
			method_options := proto.GetExtension(rpc.GetOptions(), protocol.E_MethodOptions).(*protocol.BGSMethodOptions)
			fd.printf("\n")
			fd.line_indentation++

			fd.indent()
			fd.printf("option (method_options).id = %d;\n", method_options.GetId())

			fd.line_indentation--
			fd.indent()
		}

		// write options
		fd.printf("}\n")
	}
	fd.line_indentation--

	// for _, field := range message_descriptor.Field {
	// 	type_name := field.GetTypeName()
	// 	// if type_name == "" {
	// 	// 	// no type name
	// 	// 	type_name = field.Type.
	// 	// }

	// 	// if field.OneofIndex != nil {
	// 	// 	oneof := message_def.OneofDecl[*field.OneofIndex]
	// 	// 	oneof.Options.
	// 	// }
	// 	number := field.GetNumber()

	// 	fd.printf("\t%s %s = %d;", type_name, field.GetName(), number)
	// }
	fd.printf("}\n")
	return
}