"""Extract .9 ( plan9 archive)
This program was made for Macintosh.
"""
import macfs
import EasyDialogs
import re
import sys
import os
import string

origin="Reserve2:plan9:plan9" #target directory
def extractfile(F,name,size,buf):
	D = None
	if size > 0:
		plan9path=string.replace(name,"/",":") #chage for Macintosh style
		if plan9path[:1] == ":":
			plan9path = plan9path[1:]
		macpath=origin+":"+plan9path
		if macpath[-1] == ":":
			isdir=1
		else:
			isdir=0
		macdir=os.path.dirname(macpath)
		try:
			os.makedirs(macdir)
		except:
			pass
		if not isdir:
			try:
				D = open(macpath,"wb")
			except:
				pass
	while size > 0:
		if len(buf) > size:
			if D:
				D.write(buf[:size])
				D.close()
			buf = buf[size:]
			size = 0
		else:
			if D:
				D.write(buf)
			size = size - len(buf)
			buf = F.read(1024)
			if not buf:
				break
	return buf
	
NoFiles="No File names"
fs, ok = macfs.PromptGetFile("9 file")
if not ok:
	EasyDialogs.Message("File can't open")
	sys.exit(2)

filename=fs.as_pathname()

F = open(filename,"rb")
buf=""
bar=EasyDialogs.ProgressBar("plan9",10,"" ) #For show progress
while 1:
	if len(buf) < 100:
		line = F.read(1024)
		if not line:
			break
		buf=buf+line
	t=re.split("[\r\n]+",buf,1)
	#check split validation
	if len(t) != 2:
		buf = ""
		continue
	filecont = re.split("\s",t[0])
	#check filecont validation
	if len(filecont) != 6:
		buf = ""
		continue		
	try:
		name=filecont[0]
		size=filecont[5]
		bar.label(name)
		buf = extractfile(F,name,int(size),t[1])
	except:
		buf=t[1]