bcproxy-python/bcprotocol.py at master · zither/bcproxy-python · GitHub

Repository navigation

History

210 lines (183 loc) · 7.7 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

190

191

192

193

194

195

196

197

198

199

200

201

202

203

204

205

206

207

208

209

210

#!/usr/bin/python

# -*- coding: UTF-8 –*-

# This is a simple BatMUD protocol parser, written using only the default python

# library. If you want to make a suggestion or fix something you can contact-me

# at ilorn.mc_at_gmail.com

# Distributed over IDC(I Don't Care) license

import sys

import colortrans

S_TEXT = 0

S_ESC = 1

S_TAG_OPEN = 2

S_OPEN_CODE = 3

S_TAG_CLOSE = 4

S_CLOSE_CODE = 5

S_AFTER_TEN = 6

S_IAC = 7

class Options:

def __init__(self, codes, enable_color, enable_combat_plugin):

self.codes = codes

self.enable_color = enable_color

self.enable_combat_plugin = enable_combat_plugin

class Expression:

def __init__(self):

self.code = ""

self.argu = ""

self.content = ""

class Parser:

def __init__(self, options):

self.options = options

self.stats = S_TEXT

self.output = ""

self.stack = []

# 当收到的数据不完整时，code 很可能被截断，导致前一位数字丢失

# 因此添加了 tmp_code 变量临时存储前一位，保证 code 始终为两位

self.tmp_code = ""

self.expression = None

def is_valid_code(self, char):

return char.isdigit()

def parse(self, data):

self.reset()

return self.process(data)

def reset(self):

self.output = ""

def do_with_text(self, chars):

if self.expression:

self.expression.content += chars

else:

self.output += chars

def process(self, data):

for char in data:

if self.stats == S_TEXT:

if char == "\033":

self.stats = S_ESC

else:

self.do_with_text(char)

continue

if self.stats == S_ESC:

if char == "<":

self.stats = S_TAG_OPEN

elif char == ">":

self.stats = S_TAG_CLOSE

elif char == "|" and self.expression:

self.expression.argu = self.expression.content

self.expression.content = ""

self.stats = S_TEXT

else:

self.do_with_text("\033" + char)

self.stats = S_TEXT

continue

if self.stats == S_TAG_OPEN:

if self.is_valid_code(char):

self.tmp_code += char

self.stats = S_OPEN_CODE

else:

self.do_with_text("\033" + "<" + char)

self.stats = S_TEXT

continue

if self.stats == S_OPEN_CODE:

if self.is_valid_code(char):

if self.expression:

self.stack.append(self.expression)

self.expression = Expression()

self.expression.code += self.tmp_code + char

self.stats = S_TEXT

else:

self.do_with_text("\033" + "<" + self.tmp_code + char)

self.stats = S_TEXT

self.tmp_code = ""

continue

if self.stats == S_TAG_CLOSE:

if self.is_valid_code(char):

self.tmp_code += char

self.stats = S_CLOSE_CODE

else:

self.do_with_text("\033" + ">" + char)

self.stats = S_TEXT

continue

if self.stats == S_CLOSE_CODE:

if self.is_valid_code(char):

code = self.tmp_code + char

if not self.expression or self.expression.code != code:

self.stats = S_TEXT

elif self.expression.code == "10" and self.expression.argu == "spec_prompt":

self.stats = S_AFTER_TEN

else:

tmp_content = self.parse_exp(self.expression)

self.expression = self.stack.pop() if self.stack else None

self.do_with_text(tmp_content)

self.stats = S_TEXT

else:

self.do_with_text("\033" + ">" + self.tmp_code + char)

self.stats = S_TEXT

self.tmp_code = ""

continue

if self.stats == S_AFTER_TEN:

if char == "\377":

self.stats = S_IAC

elif char == "\033":

self.expression = None

self.stats = S_ESC

else:

self.expression = None

self.do_with_text(char)

self.stats = S_TEXT

continue

if self.stats == S_IAC:

if char == "\371":

self.output += self.parse_exp(self.expression)

self.expression = None

self.stats = S_TEXT

else:

self.expression = self.stack.pop() if self.stack else None

self.do_with_text("\377" + char)

self.stats = S_TEXT

continue

return self.output

def parse_exp(self, exp):

if exp.code in self.options.codes or exp.code in ["05", "06", "11", "29", "40", "41", "42"]:

return ""

if exp.code in ["22", "23", "24", "25", "31"]:

return exp.content

if exp.code == "10":

if exp.argu == "spec_battle" and self.options.enable_combat_plugin:

return "[-10-]" + exp.content.replace("\n", " ").strip() + "\r\n"

elif exp.argu == "spec_prompt":

return exp.content.strip() +"\r\n"

elif exp.content == "NoMapSupport":

return ""

else:

return exp.content.strip() +"\r\n"

if exp.code == "20" or exp.code == "21":

if not self.options.enable_color:

return exp.content

elif exp.argu:

rgb = exp.argu.zfill(6) if len(exp.argu) < 6 else exp.argu

# fix some invalid RGB color from server

if len(rgb) > 6:

rgb = ''.join(c for c in rgb if c.isdigit())

short, _ = colortrans.rgb2short(rgb)

return "\033[{}8;5;{}m{}\033[0m".format(3 if exp.code == "20" else 4, short, exp.content)

if exp.code == "99":

if exp.content.startswith("BAT_MAPPER;;REALM_MAP"):

return "Exited to realm map.\r\n"

elif exp.content.startswith("BAT_MAPPER;;"):

room = exp.content.split(";;")

return "[-{}-]{}\r\n".format(exp.code, ";;".join(room[1:5] + [room[7]]))

return "[-{}-]{}\r\n".format(exp.code, exp.content)

if __name__ == '__main__':

options = Options([], True, True)

parser = Parser(options)

with open("logs_debug.txt") as logs:

content = logs.read()

#content = """<10spec_map| yest [1m<31north|north>31>10"""

#content = "<10spec_map|<31north|north>31>10"

#content = "<10spec_map|123>10"

#content = """<10spec_map|<200|[32mT>20>10"""

#content = "<540 0 0>54<10spec_prompt|Hp:301/301 Sp:948/948 Ep:245/245 Exp:147871 >>10<52Zjmee 0 duck 65 1 147871>52<50301 301 948 948 245 245>50"

#content = """<10spec_map| <200000FF|>20 Massive bronze gate (closed) leads north.>10"""

#content = """123>20<10spec_map| <200000FF|>20 Massive bronze gate (closed) leads north.>10"""

#content = """<10spec_map|<200000FF|| >20>20 DATA>20<770000AA|>77 GOOD>10"""

#content = "<10spec_prompt|Hp:301/301 Sp:963/963 Ep:241/241 Exp:1741 >>10BAAAA"

data = parser.parse(content)

print data

bcproxy-python/bcprotocol.py at master · zither/bcproxy-python · GitHub

Repository navigation

Expand file tree

Breadcrumbs

bcprotocol.py

Latest commit

History

Breadcrumbs

bcprotocol.py

File metadata and controls