Adds width attribute which specifies number of bytes printed per line

Closes juhakivekas#3

multidiff/Render.py

@@ -1,9 +1,11 @@
 from multidiff.Ansi import Ansi
 import binascii
 import html
+import textwrap
+import os
 
 class Render():
-	def __init__(self, encoder='hexdump', color='ansi'):
+	def __init__(self, encoder='hexdump', color='ansi', width='82'):
 		'''Configure the output encoding and coloring method of this rendering object'''
 		if   color == 'ansi':
 			self.highligther = ansi_colored
@@ -16,17 +18,19 @@ def __init__(self, encoder='hexdump', color='ansi'):
 			self.encoder = HexEncoder
 		elif encoder == 'utf8':
 			self.encoder = Utf8Encoder
-
+
+		self.width = width
+
 	def render(self, model, diff):
 		'''Render the diff in the given model into a UTF-8 String'''
 		result = self.encoder(self.highligther)
 		obj = model.objects[diff.target]
 		for op in diff.opcodes:
 			data = obj.data[op[3]:op[4]]
 			if type(data) == bytes:
-				result.append(data, op[0])
+				result.append(data, op[0], self.width)
 			elif type(data) == str:
-				result.append(bytes(data, "utf8"), op[0])
+				result.append(bytes(data, "utf8"), op[0], self.width)
 		return result.final()
 
 	def dumps(self, model):
@@ -44,7 +48,7 @@ def __init__(self, highligther):
 
 	def append(self, data, color):
 		self.output += self.highligther(str(data, 'utf8'), color)
-	
+
 	def final(self):
 		return self.output
 
@@ -56,7 +60,7 @@ def __init__(self, highligther):
 	def append(self, data, color):
 		data = str(binascii.hexlify(data),'utf8')
 		self.output += self.highligther(data, color)
-	
+
 	def final(self):
 		return self.output
 
@@ -71,16 +75,16 @@ def __init__(self, highligther):
 		self.skipspace = False
 		self.asciirow = ''
 
-	def append(self, data, color):
+	def append(self, data, color, width):
 		if len(data) == 0:
-			self._append(data, color)
+			self._append(data, color, width)
 		while len(data) > 0:
 			if self.rowlen == 16:
 				self._newrow()
-			consumed = self._append(data[:16 - self.rowlen], color)
+			consumed = self._append(data[:16 - self.rowlen], color, width)
 			data = data[consumed:]
 
-	def _append(self, data, color):
+	def _append(self, data, color, width):
 		if len(data) == 0:
 			#in the case of highlightig a deletion in a target or an
 			#addition in the source, print a highlighted space and mark
@@ -101,7 +105,9 @@ def _append(self, data, color):
 					asciis += '.'
 			self.asciirow += self.highligther(asciis, color)
 
-		self.hexrow   += self.highligther(hexs, color)
+		self.hexrow += self.highligther(hexs, color)
+		if len(self.hexrow) > int(width):
+			self.hexrow = textwrap.fill(self.hexrow, int(width))
 		self.rowlen += len(data)
 		return len(data)
 
@@ -121,7 +127,6 @@ def _add_hex_space(self):
 			self.skipspace = False
 		else:
 			self.hexrow += ' '
-
 
 	def final(self):
 		self.hexrow += 3*(16 - self.rowlen) * ' '

multidiff/StreamView.py

@@ -4,9 +4,9 @@
 class StreamView():
 	'''A class for building UIs. Has some pretty serious side effects.
 	Use Render instead if you're not making a long-running UI'''
-	def __init__(self, model, encoding='hexdump', mode='sequence', color='ansi'):
+	def __init__(self, model, encoding='hexdump', mode='sequence', color='html', width='82'):
 		self.color = color
-		self.render = Render(color=color, encoder=encoding)
+		self.render = Render(color=color, encoder=encoding, width=width)
 		self.mode = mode
 		self.model = model
 		model.add_listener(self)
@@ -20,7 +20,7 @@ def diff_added(self, diff):
 			if self.color == 'ansi':
 				print(Ansi.bold + self.model.objects[diff.target].info + Ansi.reset)
 			elif self.color == 'html':
-				print('<span style="font-weight: bold;">' + html.escape(self.model.objects[diff.target].info) + '</span>')
+				print(html.escape(self.model.objects[diff.target-1].info) + ' --> ' + html.escape(self.model.objects[diff.target].info))
 		print(self.render.render(self.model, diff))
 		#StreamView is designed to run for long times so we delete
 		#old objects and diffs to not leak memory

multidiff/command_line_interface.py

@@ -1,12 +1,21 @@
 #!/usr/bin/python3
 import argparse
 from multidiff import MultidiffModel, StreamView, SocketController, FileController, StdinController
+import os
 
 def main():
+
+	rows, columns = os.popen('stty size', 'r').read().split()
+	columns = int(columns)
+
 	args = make_parser().parse_args()
 	m = MultidiffModel()
-	v = StreamView(m, encoding=args.outformat, mode=args.mode, color=args.color)
-
+
+	if args.width == 'max':
+		args.width = columns
+
+	v = StreamView(m, encoding=args.outformat, mode=args.mode, color=args.color, width=args.width)
+
 	if len(args.file) > 0:
 		informat = args.informat if args.informat else 'raw'
 		files = FileController(m, informat)
@@ -32,17 +41,17 @@ def make_parser():
   │ sensor module │
   └───────────────┘
 """)
-	
+
 	parser.add_argument('file',
 		type=str,
 		nargs='*',
 		help='file or directory to include in multidiff')
-	
+
 	parser.add_argument('-p','--port',
 		dest='port',
 		type=int,
 		help='start a local socket server on the given port')
-	
+
 	parser.add_argument('-s','--stdin',
 		dest='stdin',
 		action='store_true',
@@ -52,29 +61,34 @@ def make_parser():
 		dest='mode',
 		default='sequence',
 		help='mode of operation, either "baseline" or "sequence"')
-	
+
 	parser.add_argument('-i','--informat',
 		dest='informat',
 		help='input data format:\n' +
 			'  utf8 (stdin default)\n' +
 			'  raw (file and server default)\n' +
 			'  hex\n'+
 			'  json')
-	
+
 	parser.add_argument('-o','--outformat',
 		dest='outformat',
 		default='hexdump',
-		help='output data format:\n' + 
-			'  utf8\n' + 
-			'  hex\n' + 
+		help='output data format:\n' +
+			'  utf8\n' +
+			'  hex\n' +
 			'  hexdump (default)')
 
 	parser.add_argument('--html',
 		dest='color',
 		action='store_const',
 		const='html',
-		default='ansi',
+		default='html',
 		help='use html for colors instead of ansi codes')
+
+	parser.add_argument('-w', '--width',
+		dest='width',
+		default='82',
+		help='number of bytes per line')
 	return parser
 
 if __name__ == '__main__':