# A minimal text editor. # # To be done: # - Functionality: find, etc. from Menu import DrawMenuBar from FrameWork import * import Win import Qd import Res import Fm import waste import WASTEconst import Scrap import os import macfs import regsub import string import htmllib WATCH = Qd.GetCursor(4).data LEFTMARGIN=0 UNDOLABELS = [ # Indexed by WEGetUndoInfo() value None, "", "typing", "Cut", "Paste", "Clear", "Drag", "Style"] # Style and size menu. Note that style order is important (tied to bit values) STYLES = [ ("Bold", "B"), ("Italic", "I"), ("Underline", "U"), ("Outline", "O"), ("Shadow", ""), ("Condensed", ""), ("Extended", "") ] SIZES = [ 9, 10, 12, 14, 18, 24] # Sizes for HTML tag types HTML_SIZE={ 'h1': 18, 'h2': 14 } BIGREGION=Qd.NewRgn() Qd.SetRectRgn(BIGREGION, -16000, -16000, 16000, 16000) class WasteWindow(ScrolledWindow): def open(self, path, name, data): self.path = path self.name = name r = windowbounds(400, 400) w = Win.NewWindow(r, name, 1, 0, -1, 1, 0x55555555) self.wid = w vr = LEFTMARGIN, 0, r[2]-r[0]-15, r[3]-r[1]-15 dr = (0, 0, vr[2], 0) Qd.SetPort(w) Qd.TextFont(4) Qd.TextSize(9) flags = WASTEconst.weDoAutoScroll | WASTEconst.weDoOutlineHilite | \ WASTEconst.weDoMonoStyled | WASTEconst.weDoUndo self.ted = waste.WENew(dr, vr, flags) style, soup = self.getstylesoup(self.path) self.ted.WEInsert(data, style, soup) self.ted.WESetSelection(0,0) self.ted.WECalText() self.ted.WEResetModCount() w.DrawGrowIcon() self.scrollbars() self.do_postopen() self.do_activate(1, None) def getstylesoup(self, pathname): if not pathname: return None, None oldrf = Res.CurResFile() try: rf = Res.OpenResFile(self.path) except Res.Error: return None, None try: hstyle = Res.Get1Resource('styl', 128) hstyle.DetachResource() except Res.Error: hstyle = None try: hsoup = Res.Get1Resource('SOUP', 128) hsoup.DetachResource() except Res.Error: hsoup = None Res.CloseResFile(rf) Res.UseResFile(oldrf) return hstyle, hsoup def do_idle(self, event): (what, message, when, where, modifiers) = event Qd.SetPort(self.wid) self.ted.WEIdle() if self.ted.WEAdjustCursor(where, BIGREGION): return Qd.SetCursor(Qd.qd.arrow) def getscrollbarvalues(self): dr = self.ted.WEGetDestRect() vr = self.ted.WEGetViewRect() vx = self.scalebarvalue(dr[0], dr[2], vr[0], vr[2]) vy = self.scalebarvalue(dr[1], dr[3], vr[1], vr[3]) return vx, vy def scrollbar_callback(self, which, what, value): if which == 'y': # # "line" size is minimum of top and bottom line size # topline_off,dummy = self.ted.WEGetOffset((1,1)) topline_num = self.ted.WEOffsetToLine(topline_off) toplineheight = self.ted.WEGetHeight(topline_num, topline_num+1) botlinepos = self.ted.WEGetViewRect()[3] botline_off, dummy = self.ted.WEGetOffset((1, botlinepos-1)) botline_num = self.ted.WEOffsetToLine(botline_off) botlineheight = self.ted.WEGetHeight(botline_num, botline_num+1) if botlineheight == 0: botlineheight = self.ted.WEGetHeight(botline_num-1, botline_num) if botlineheight < toplineheight: lineheight = botlineheight else: lineheight = toplineheight if lineheight <= 0: lineheight = 1 # # Now do the command. # if what == 'set': height = self.ted.WEGetHeight(0, 0x3fffffff) cur = self.getscrollbarvalues()[1] delta = (cur-value)*height/32767 if what == '-': delta = lineheight elif what == '--': delta = (self.ted.WEGetViewRect()[3]-lineheight) if delta <= 0: delta = lineheight elif what == '+': delta = -lineheight elif what == '++': delta = -(self.ted.WEGetViewRect()[3]-lineheight) if delta >= 0: delta = -lineheight self.ted.WEScroll(0, delta) else: if what == 'set': return # XXXX vr = self.ted.WEGetViewRect() winwidth = vr[2]-vr[0] if what == '-': delta = winwidth/10 elif what == '--': delta = winwidth/2 elif what == '+': delta = -winwidth/10 elif what == '++': delta = -winwidth/2 self.ted.WEScroll(delta, 0) # Pin the scroll l, t, r, b = self.ted.WEGetDestRect() vl, vt, vr, vb = self.ted.WEGetViewRect() if t > 0 or l > 0: dx = dy = 0 if t > 0: dy = -t if l > 0: dx = -l self.ted.WEScroll(dx, dy) elif b < vb: self.ted.WEScroll(0, vb-b) def do_activate(self, onoff, evt): Qd.SetPort(self.wid) ScrolledWindow.do_activate(self, onoff, evt) if onoff: self.ted.WEActivate() self.parent.active = self self.parent.updatemenubar() else: self.ted.WEDeactivate() def do_update(self, wid, event): region = wid.GetWindowPort().visRgn if Qd.EmptyRgn(region): return Qd.EraseRgn(region) self.ted.WEUpdate(region) self.updatescrollbars() def do_postresize(self, width, height, window): l, t, r, b = self.ted.WEGetViewRect() vr = (l, t, l+width-15, t+height-15) self.ted.WESetViewRect(vr) Win.InvalRect(vr) ScrolledWindow.do_postresize(self, width, height, window) def do_contentclick(self, local, modifiers, evt): (what, message, when, where, modifiers) = evt self.ted.WEClick(local, modifiers, when) self.updatescrollbars() self.parent.updatemenubar() def do_char(self, ch, event): self.ted.WESelView() (what, message, when, where, modifiers) = event self.ted.WEKey(ord(ch), modifiers) self.updatescrollbars() self.parent.updatemenubar() def close(self): if self.ted.WEGetModCount(): save = EasyDialogs.AskYesNoCancel('Save window "%s" before closing?'%self.name, 1) if save > 0: self.menu_save() elif save < 0: return if self.parent.active == self: self.parent.active = None self.parent.updatemenubar() del self.ted self.do_postclose() def menu_save(self): if not self.path: self.menu_save_as() return # Will call us recursively # # First save data # dhandle = self.ted.WEGetText() data = dhandle.data fp = open(self.path, 'wb') # NOTE: wb, because data has CR for end-of-line fp.write(data) if data[-1] <> '\r': fp.write('\r') fp.close() # # Now save style and soup # oldresfile = Res.CurResFile() try: rf = Res.OpenResFile(self.path) except Res.Error: Res.CreateResFile(self.path) rf = Res.OpenResFile(self.path) styles = Res.Resource('') soup = Res.Resource('') self.ted.WECopyRange(0, 0x3fffffff, None, styles, soup) styles.AddResource('styl', 128, '') soup.AddResource('SOUP', 128, '') Res.CloseResFile(rf) Res.UseResFile(oldresfile) self.ted.WEResetModCount() def menu_save_as(self): fss, ok = macfs.StandardPutFile('Save as:') if not ok: return self.path = fss.as_pathname() self.name = os.path.split(self.path)[-1] self.wid.SetWTitle(self.name) self.menu_save() def menu_insert(self, fp): self.ted.WESelView() data = fp.read() self.ted.WEInsert(data, None, None) self.updatescrollbars() self.parent.updatemenubar() def menu_insert_html(self, fp): import htmllib import formatter f = formatter.AbstractFormatter(self) # Remember where we are, and don't update Qd.SetCursor(WATCH) start, dummy = self.ted.WEGetSelection() self.ted.WEFeatureFlag(WASTEconst.weFInhibitRecal, 1) self.html_init() p = MyHTMLParser(f) p.feed(fp.read()) # Restore updating, recalc, set focus dummy, end = self.ted.WEGetSelection() self.ted.WECalText() self.ted.WESetSelection(start, end) self.ted.WESelView() self.ted.WEFeatureFlag(WASTEconst.weFInhibitRecal, 0) Win.InvalRect(self.ted.WEGetViewRect()) self.updatescrollbars() self.parent.updatemenubar() def menu_cut(self): self.ted.WESelView() self.ted.WECut() Scrap.ZeroScrap() self.ted.WECut() self.updatescrollbars() self.parent.updatemenubar() def menu_copy(self): Scrap.ZeroScrap() self.ted.WECopy() self.updatescrollbars() self.parent.updatemenubar() def menu_paste(self): self.ted.WESelView() self.ted.WEPaste() self.updatescrollbars() self.parent.updatemenubar() def menu_clear(self): self.ted.WESelView() self.ted.WEDelete() self.updatescrollbars() self.parent.updatemenubar() def menu_undo(self): self.ted.WEUndo() self.updatescrollbars() self.parent.updatemenubar() def menu_setfont(self, font): font = Fm.GetFNum(font) self.mysetstyle(WASTEconst.weDoFont, (font, 0, 0, (0,0,0))) self.parent.updatemenubar() def menu_modface(self, face): self.mysetstyle(WASTEconst.weDoFace|WASTEconst.weDoToggleFace, (0, face, 0, (0,0,0))) def menu_setface(self, face): self.mysetstyle(WASTEconst.weDoFace|WASTEconst.weDoReplaceFace, (0, face, 0, (0,0,0))) def menu_setsize(self, size): self.mysetstyle(WASTEconst.weDoSize, (0, 0, size, (0,0,0))) def menu_incsize(self, size): self.mysetstyle(WASTEconst.weDoAddSize, (0, 0, size, (0,0,0))) def mysetstyle(self, which, how): self.ted.WESelView() self.ted.WESetStyle(which, how) self.parent.updatemenubar() def have_selection(self): start, stop = self.ted.WEGetSelection() return start < stop def can_paste(self): return self.ted.WECanPaste() def can_undo(self): which, redo = self.ted.WEGetUndoInfo() which = UNDOLABELS[which] if which == None: return None if redo: return "Redo "+which else: return "Undo "+which def getruninfo(self): all = (WASTEconst.weDoFont | WASTEconst.weDoFace | WASTEconst.weDoSize) dummy, mode, (font, face, size, color) = self.ted.WEContinuousStyle(all) if not (mode & WASTEconst.weDoFont): font = None else: font = Fm.GetFontName(font) if not (mode & WASTEconst.weDoFace): fact = None if not (mode & WASTEconst.weDoSize): size = None return font, face, size # # Methods for writer class for html formatter # def html_init(self): self.html_font = [12, 0, 0, 0] self.html_style = 0 self.html_color = (0,0,0) self.new_font(self.html_font) def new_font(self, font): if font == None: font = (12, 0, 0, 0) font = map(lambda x:x, font) for i in range(len(font)): if font[i] == None: font[i] = self.html_font[i] [size, italic, bold, tt] = font self.html_font = font[:] if tt: font = Fm.GetFNum('Courier') else: font = Fm.GetFNum('Times') if HTML_SIZE.has_key(size): size = HTML_SIZE[size] else: size = 12 face = 0 if bold: face = face | 1 if italic: face = face | 2 face = face | self.html_style self.ted.WESetStyle(WASTEconst.weDoFont | WASTEconst.weDoFace | WASTEconst.weDoSize | WASTEconst.weDoColor, (font, face, size, self.html_color)) def new_margin(self, margin, level): self.ted.WEInsert('[Margin %s %s]'%(margin, level), None, None) def new_spacing(self, spacing): self.ted.WEInsert('[spacing %s]'%spacing, None, None) def new_styles(self, styles): self.html_style = 0 self.html_color = (0,0,0) if 'anchor' in styles: self.html_style = self.html_style | 4 self.html_color = (0xffff, 0, 0) self.new_font(self.html_font) def send_paragraph(self, blankline): self.ted.WEInsert('\r'*(blankline+1), None, None) def send_line_break(self): self.ted.WEInsert('\r', None, None) def send_hor_rule(self, *args, **kw): # Ignore ruler options, for now dummydata = Res.Resource('') self.ted.WEInsertObject('rulr', dummydata, (0,0)) def send_label_data(self, data): self.ted.WEInsert(data, None, None) def send_flowing_data(self, data): self.ted.WEInsert(data, None, None) def send_literal_data(self, data): data = regsub.gsub('\n', '\r', data) data = string.expandtabs(data) self.ted.WEInsert(data, None, None) class Wed(Application): def __init__(self): Application.__init__(self) self.num = 0 self.active = None self.updatemenubar() waste.STDObjectHandlers() # Handler for horizontal ruler waste.WEInstallObjectHandler('rulr', 'new ', self.newRuler) waste.WEInstallObjectHandler('rulr', 'draw', self.drawRuler) def makeusermenus(self): self.filemenu = m = Menu(self.menubar, "File") self.newitem = MenuItem(m, "New window", "N", self.open) self.openitem = MenuItem(m, "Open...", "O", self.openfile) self.closeitem = MenuItem(m, "Close", "W", self.closewin) m.addseparator() self.saveitem = MenuItem(m, "Save", "S", self.save) self.saveasitem = MenuItem(m, "Save as...", "", self.saveas) m.addseparator() self.insertitem = MenuItem(m, "Insert plaintext...", "", self.insertfile) self.htmlitem = MenuItem(m, "Insert HTML...", "", self.inserthtml) m.addseparator() self.quititem = MenuItem(m, "Quit", "Q", self.quit) self.editmenu = m = Menu(self.menubar, "Edit") self.undoitem = MenuItem(m, "Undo", "Z", self.undo) self.cutitem = MenuItem(m, "Cut", "X", self.cut) self.copyitem = MenuItem(m, "Copy", "C", self.copy) self.pasteitem = MenuItem(m, "Paste", "V", self.paste) self.clearitem = MenuItem(m, "Clear", "", self.clear) self.makefontmenu() # Groups of items enabled together: self.windowgroup = [self.closeitem, self.saveitem, self.saveasitem, self.editmenu, self.fontmenu, self.facemenu, self.sizemenu, self.insertitem] self.focusgroup = [self.cutitem, self.copyitem, self.clearitem] self.windowgroup_on = -1 self.focusgroup_on = -1 self.pastegroup_on = -1 self.undo_label = "never" self.ffs_values = () def makefontmenu(self): self.fontmenu = Menu(self.menubar, "Font") self.fontnames = getfontnames() self.fontitems = [] for n in self.fontnames: m = MenuItem(self.fontmenu, n, "", self.selfont) self.fontitems.append(m) self.facemenu = Menu(self.menubar, "Style") self.faceitems = [] for n, shortcut in STYLES: m = MenuItem(self.facemenu, n, shortcut, self.selface) self.faceitems.append(m) self.facemenu.addseparator() self.faceitem_normal = MenuItem(self.facemenu, "Normal", "N", self.selfacenormal) self.sizemenu = Menu(self.menubar, "Size") self.sizeitems = [] for n in SIZES: m = MenuItem(self.sizemenu, `n`, "", self.selsize) self.sizeitems.append(m) self.sizemenu.addseparator() self.sizeitem_bigger = MenuItem(self.sizemenu, "Bigger", "+", self.selsizebigger) self.sizeitem_smaller = MenuItem(self.sizemenu, "Smaller", "-", self.selsizesmaller) def selfont(self, id, item, *rest): if self.active: font = self.fontnames[item-1] self.active.menu_setfont(font) else: EasyDialogs.Message("No active window?") def selface(self, id, item, *rest): if self.active: face = (1<<(item-1)) self.active.menu_modface(face) else: EasyDialogs.Message("No active window?") def selfacenormal(self, *rest): if self.active: self.active.menu_setface(0) else: EasyDialogs.Message("No active window?") def selsize(self, id, item, *rest): if self.active: size = SIZES[item-1] self.active.menu_setsize(size) else: EasyDialogs.Message("No active window?") def selsizebigger(self, *rest): if self.active: self.active.menu_incsize(2) else: EasyDialogs.Message("No active window?") def selsizesmaller(self, *rest): if self.active: self.active.menu_incsize(-2) else: EasyDialogs.Message("No active window?") def updatemenubar(self): changed = 0 on = (self.active <> None) if on <> self.windowgroup_on: for m in self.windowgroup: m.enable(on) self.windowgroup_on = on changed = 1 if on: # only if we have an edit menu on = self.active.have_selection() if on <> self.focusgroup_on: for m in self.focusgroup: m.enable(on) self.focusgroup_on = on changed = 1 on = self.active.can_paste() if on <> self.pastegroup_on: self.pasteitem.enable(on) self.pastegroup_on = on changed = 1 on = self.active.can_undo() if on <> self.undo_label: if on: self.undoitem.enable(1) self.undoitem.settext(on) self.undo_label = on else: self.undoitem.settext("Nothing to undo") self.undoitem.enable(0) changed = 1 if self.updatefontmenus(): changed = 1 if changed: DrawMenuBar() def updatefontmenus(self): info = self.active.getruninfo() if info == self.ffs_values: return 0 # Remove old checkmarks if self.ffs_values == (): self.ffs_values = (None, None, None) font, face, size = self.ffs_values if font <> None: fnum = self.fontnames.index(font) self.fontitems[fnum].check(0) if face <> None: for i in range(len(self.faceitems)): if face & (1< None: for i in range(len(self.sizeitems)): if SIZES[i] == size: self.sizeitems[i].check(0) self.ffs_values = info # Set new checkmarks font, face, size = self.ffs_values if font <> None: fnum = self.fontnames.index(font) self.fontitems[fnum].check(1) if face <> None: for i in range(len(self.faceitems)): if face & (1< None: for i in range(len(self.sizeitems)): if SIZES[i] == size: self.sizeitems[i].check(1) # Set outline/normal for sizes if font: exists = getfontsizes(font, SIZES) for i in range(len(self.sizeitems)): if exists[i]: self.sizeitems[i].setstyle(0) else: self.sizeitems[i].setstyle(8) # # Apple menu # def do_about(self, id, item, window, event): EasyDialogs.Message("A simple single-font text editor based on WASTE") # # File menu # def open(self, *args): self._open(0) def openfile(self, *args): self._open(1) def _open(self, askfile): if askfile: fss, ok = macfs.StandardGetFile('TEXT') if not ok: return path = fss.as_pathname() name = os.path.split(path)[-1] try: fp = open(path, 'rb') # NOTE binary, we need cr as end-of-line data = fp.read() fp.close() except IOError, arg: EasyDialogs.Message("IOERROR: "+`arg`) return else: path = None name = "Untitled %d"%self.num data = '' w = WasteWindow(self) w.open(path, name, data) self.num = self.num + 1 def insertfile(self, *args): if self.active: fss, ok = macfs.StandardGetFile('TEXT') if not ok: return path = fss.as_pathname() try: fp = open(path, 'rb') # NOTE binary, we need cr as end-of-line except IOError, arg: EasyDialogs.Message("IOERROR: "+`arg`) return self.active.menu_insert(fp) else: EasyDialogs.Message("No active window?") def inserthtml(self, *args): if self.active: fss, ok = macfs.StandardGetFile('TEXT') if not ok: return path = fss.as_pathname() try: fp = open(path, 'r') except IOError, arg: EasyDialogs.Message("IOERROR: "+`arg`) return self.active.menu_insert_html(fp) else: EasyDialogs.Message("No active window?") def closewin(self, *args): if self.active: self.active.close() else: EasyDialogs.Message("No active window?") def save(self, *args): if self.active: self.active.menu_save() else: EasyDialogs.Message("No active window?") def saveas(self, *args): if self.active: self.active.menu_save_as() else: EasyDialogs.Message("No active window?") def quit(self, *args): for w in self._windows.values(): w.close() if self._windows: return self._quit() # # Edit menu # def undo(self, *args): if self.active: self.active.menu_undo() else: EasyDialogs.Message("No active window?") def cut(self, *args): if self.active: self.active.menu_cut() else: EasyDialogs.Message("No active window?") def copy(self, *args): if self.active: self.active.menu_copy() else: EasyDialogs.Message("No active window?") def paste(self, *args): if self.active: self.active.menu_paste() else: EasyDialogs.Message("No active window?") def clear(self, *args): if self.active: self.active.menu_clear() else: EasyDialogs.Message("No active window?") # # Other stuff # def idle(self, event): if self.active: self.active.do_idle(event) def newRuler(self, obj): """Insert a new ruler. Make it as wide as the window minus 2 pxls""" ted = obj.WEGetObjectOwner() l, t, r, b = ted.WEGetDestRect() return r-l, 4 def drawRuler(self, (l, t, r, b), obj): y = (t+b)/2 Qd.MoveTo(l+2, y) Qd.LineTo(r-2, y) return 0 class MyHTMLParser(htmllib.HTMLParser): def anchor_bgn(self, href, name, type): self.anchor = href if self.anchor: self.anchorlist.append(href) self.formatter.push_style('anchor') def anchor_end(self): if self.anchor: self.anchor = None self.formatter.pop_style() def getfontnames(): names = [] for i in range(256): n = Fm.GetFontName(i) if n: names.append(n) return names def getfontsizes(name, sizes): exist = [] num = Fm.GetFNum(name) for sz in sizes: if Fm.RealFont(num, sz): exist.append(1) else: exist.append(0) return exist def main(): App = Wed() App.mainloop() if __name__ == '__main__': main()