From 3dd5422293a7e318b71fa8af799dee3deb21c103 Mon Sep 17 00:00:00 2001
From: Kelvin Sherlock <ksherlock@gmail.com>
Date: Thu, 31 Oct 2024 10:19:47 -0400
Subject: [PATCH] orca/m indent/reformat utility.

---
 Default.sublime-commands |   4 +
 orca-indent.py           | 207 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 211 insertions(+)
 create mode 100644 orca-indent.py

diff --git a/Default.sublime-commands b/Default.sublime-commands
index eb63b94..5dd768e 100644
--- a/Default.sublime-commands
+++ b/Default.sublime-commands
@@ -15,6 +15,10 @@
 		"caption": "ORCA/M MacGen",
 		"command": "orca_mac_gen"
 	},
+	{
+		"caption": "ORCA/M Indent",
+		"command": "orca_indent"
+	},
 	{
 		"caption": "MPW Asm IIgs MacGen",
 		"command": "mpw_mac_gen"
diff --git a/orca-indent.py b/orca-indent.py
new file mode 100644
index 0000000..1fd0f67
--- /dev/null
+++ b/orca-indent.py
@@ -0,0 +1,207 @@
+import sublime
+import sublime_plugin
+
+TABS = (9, 15, 40)
+QUOTES = "\"'"
+
+# opcodes w/o operands
+NOPS = set((
+	b'ANOP',
+
+#	b'ASL',
+	b'CLC',
+	b'CLD',
+	b'CLI',
+	b'CLV',
+#	b'DEC',
+	b'DEX',
+	b'DEY',
+#	b'INC',
+	b'INX',
+	b'INY',
+#	b'LSR',
+	b'NOP',
+	b'PHA',
+	b'PHB',
+	b'PHD',
+	b'PHK',
+	b'PHP',
+	b'PHX',
+	b'PHY',
+	b'PLA',
+	b'PLB',
+	b'PLD',
+	b'PLP',
+	b'PLX',
+	b'PLY',
+#	b'ROL',
+#	b'ROR',
+	b'RTI',
+	b'RTL',
+	b'RTS',
+	b'SEC',
+	b'SED',
+	b'SEI',
+	b'STP',
+	b'TAX',
+	b'TAY',
+	b'TCD',
+	b'TCS',
+	b'TDC',
+	b'TSC',
+	b'TSX',
+	b'TXA',
+	b'TXS',
+	b'TXY',
+	b'TYA',
+	b'TYX',
+	b'WAI',
+	b'XBA',
+	b'XCE',
+))
+
+def tab_to(line, ts):
+	pos = TABS[ts-1]
+	line.append(0x20)
+	while len(line) < pos: line.append(0x20)
+
+
+def indent(s):
+
+	st = 0
+	q = 0
+	rv = bytearray()
+	opc = bytearray()
+
+	for c in s:
+		if c == "\t": c = " "
+		cc = ord(c)
+
+		# if c == " ":
+		# 	if st in (2,4,6): continue
+		# 	if st in (1,3):
+		# 		st += 1
+		# 		continue
+
+		if st == 0:
+			if c == " ":
+				st = 2
+				continue
+
+			if c in "!*;":
+				# leave ; in the first column
+				st = 7
+			else:
+				st += 1
+			rv.append(cc)
+			continue
+
+		if st == 1:
+			# label
+			if c == " ":
+				st += 1
+				continue
+			rv.append(cc)
+			continue
+
+		if st == 2:
+			# label? ws
+			if c == " ": continue
+			if c in "*!" and not rv:
+				st = 7
+			elif c == ";":
+				st = 7
+				tab_to(rv, 3)
+			else:
+				st += 1
+				tab_to(rv, 1)
+				opc.append(cc & ~0x20)
+			rv.append(cc)
+			continue
+
+		if st == 3:
+			# opcode
+			if c == " ":
+				st += 1
+				continue
+			rv.append(cc)
+			opc.append(cc & ~0x20)
+			continue
+
+		if st == 4:
+			# opcode ws
+			if c == " ": continue
+
+			if c == ";":
+				st = 7
+				tab_to(rv, 3)
+			elif bytes(opc) in NOPS:
+				# jump to comment...
+				st = 7
+				tab_to(rv, 3)
+				rv.append(0x3b) # ;
+			else:
+				st += 1
+				tab_to(rv, 2)
+				if c in QUOTES: q = cc
+			rv.append(cc)
+			continue
+
+		if st == 5:
+			# operand
+			if q:
+				rv.append(cc)
+				if q == cc: q = 0
+				continue
+			if c == " ":
+				st += 1
+				continue
+			rv.append(cc)
+			if c in QUOTES: q = cc
+			continue
+
+		if st == 6:
+			# operand ws
+			if c == " ": continue
+			st += 1
+			tab_to(rv, 3)
+			if c != ";": rv.append(0x3b) # ;
+			rv.append(cc)
+			continue
+
+		if st == 7:
+			# comment
+			rv.append(cc)
+			continue
+
+
+	return rv.decode('ascii')
+
+
+
+class OrcaIndent(sublime_plugin.TextCommand):
+
+	def is_enabled(self):
+		scope = self.view.scope_name(0)
+		lang = scope.split(' ')[0]
+		return lang == 'source.asm.65816.orcam'
+
+
+	def run(self, edit):
+		view = self.view
+		all = sublime.Region(0, view.size())
+
+		# disable space indentation.
+		view.settings().set('translate_tabs_to_spaces', False)
+
+		data = []
+		for r in view.lines(all):
+			text = view.substr(r).rstrip()
+
+			data.append(indent(text))
+
+
+		data.append('')
+		text = '\n'.join(data)
+
+		view.replace(edit, all, text)