[projects/pld-ftp-admin.git] / bin / pfa-checktree

#!/usr/bin/env python3
# vi: encoding=utf-8 ts=8 sts=4 sw=4 et

from __future__ import print_function

import sys, os
sys.path.insert(0, os.environ['HOME']+'/pld-ftp-admin/modules')
import ftptree
ftptree.quietmode=True
from common import checkdir
import ftpio
import config
cval=config.value

import curses

if len(sys.argv) != 3:
    print("ERR: wrong number of arguments")
    print("check-tree.py srctree dsttree")
    sys.exit(1)

checkdir(sys.argv[1])

ftpio.connect('check-tree')

if not ftpio.lock(sys.argv[1], True):
    print("ERR: %s tree already locked" % sys.argv[1])
    sys.exit(1)

if not ftpio.lock(sys.argv[2], True):
    ftpio.unlock(sys.argv[1])
    print("ERR: %s tree already locked" % sys.argv[2])
    sys.exit(1)

srctree=ftptree.FtpTree(sys.argv[1], loadall=True)
dsttree=ftptree.FtpTree(sys.argv[2])

srctree.checktree(dsttree)

# We've loaded all interesting info, so we can continue without locks
ftpio.unlock(sys.argv[1])
ftpio.unlock(sys.argv[2])

# The list(set(list)) cast is a fancy way to remove dups from a list
names=sorted(list(set([pkg.name for pkg in srctree.values()])))

pkgtree={}

for n in names:
    pkgtree[n]=[]

for pkg in srctree.values():
    pkgtree[pkg.name].append(pkg)

for key in pkgtree.keys():
    pkgtree[key]=sorted(pkgtree[key], reverse=True)

for pkg in srctree.values():
    pkg.marked4movingpool=pkgtree[pkg.name]

pkglist=[]

for key in sorted(pkgtree.keys()):
    pkglist=pkglist+pkgtree[key]

DoQuit="OMG!"

class Screen:
    def __init__(self, srctreename, dsttreename, elements):
        self.srctree=srctreename
        self.dsttree=dsttreename
        self.elements=elements
        self.pkglistselected=0
        self.pkgliststartpad=0
        self.win2startpad=0
        self.window=1
        try:
            # initial
            self.screen=curses.initscr()
            self.screen.keypad(1)
            curses.start_color()
            curses.noecho()
            curses.cbreak()
            curses.meta(1)
            curses.curs_set(0)

             # Without this refresh(), nothing will show up before first getch()
            self.screen.refresh()

            self.__init_colors()
            self.__init_windows()
            self.__bind_keys()

            try:
                # Main loop
                while True:
                    if self.window==1: #pkglist
                        self.__redraw_pkglist()
                        self.__redraw_statusbox()
                    elif self.window==2: 
                        pass
                    curses.doupdate()
                    self.__handle_input()
            except DoQuit:
                pass
        finally:
            curses.echo()
            curses.nocbreak()
            curses.curs_set(1)
            curses.endwin()

    def __bind_keys(self):
        self.globalkeys=(('qQ', 'quit'), ('KEY_RESIZE', 'resize'),
                ('KEY_CLOSE', 'quit'))
        self.perwindowkeys=[(), ()]
        self.perwindowkeys[0]=(('k', 'move_up'), ('J', 'move_pgdown'),
                ('K', 'move_pgup'), (' ', 'mark'), ('j', 'move_down'),
                ('KEY_DOWN', 'move_down'), ('KEY_UP', 'move_up'),
                ('KEY_HOME', 'move_home'), ('KEY_END', 'move_end'),
                ('KEY_NPAGE', 'move_pgdown'), ('KEY_PPAGE', 'move_pgup'))
                

    def __init_colors(self):
        curses.init_pair(1, curses.COLOR_BLACK, curses.COLOR_CYAN)
        self.COLOR_TITLEBAR=curses.color_pair(1)
        curses.init_pair(2, curses.COLOR_WHITE, curses.COLOR_BLUE)
        self.COLOR_STANDARD=curses.color_pair(2)
        curses.init_pair(3, curses.COLOR_RED, curses.COLOR_BLUE)
        self.COLOR_ERROR=curses.color_pair(3)
        curses.init_pair(4, curses.COLOR_YELLOW, curses.COLOR_BLUE)
        self.COLOR_WARNING=curses.color_pair(4)
        curses.init_pair(5, curses.COLOR_WHITE, curses.COLOR_RED)
        self.COLOR_ERRORMSG=curses.color_pair(5)
        curses.init_pair(6, curses.COLOR_WHITE, curses.COLOR_YELLOW)
        self.COLOR_WARNINGMSG=curses.color_pair(6)
        curses.init_pair(7, curses.COLOR_BLACK, curses.COLOR_WHITE)
        self.COLOR_WHITEBG=curses.color_pair(7)

    def __init_windows(self):
        self.__set_screen_coordinates()

        # Titlebar
        self.titlebar = curses.newpad(1, 200)
        self.titlebar.bkgdset(ord(' '), self.COLOR_TITLEBAR)
        self.__redraw_titlebar()

        # Pkg list pad
        self.pkglist = curses.newpad(len(self.elements), 200)
        self.pkglist.bkgdset(ord(' '), self.COLOR_STANDARD)
        self.pkglist.clrtobot()

        # Status bar
        self.statusbar = curses.newpad(1, 200)

        # Pkg status box
        self.statusbox = curses.newpad(5, 200)
        self.statusbox.bkgdset(ord(' '), self.COLOR_STANDARD)
        self.statusbox.clrtobot()
        self.__redraw_statusbox()

    def __set_screen_coordinates(self):
        self.scr_h, self.scr_w = self.screen.getmaxyx()
        self.pad_h = self.scr_h-5
        self.sbar_y = self.scr_h-4
        self.sbox_y = self.scr_h-3

    def __redraw_titlebar(self):
        self.titlebar.clrtobot()
#        self.titlebar.addstr(0, self.scr_w-14, "pld-ftp-admin")
        self.titlebar.addstr(0, 1, 'Window #%s' % self.window)
        tmp="%s => %s" % (self.srctree, self.dsttree)
        self.titlebar.addstr(0, int(self.scr_w/2-len(tmp)/2)-1, tmp)
        self.titlebar.addstr(0, self.scr_w-20, "Press 'h' for help.")
        self.titlebar.noutrefresh(0, 0, 0, 0, 1, self.scr_w-1)

    def __redraw_statusbox(self):
        self.statusbox.erase()
        pkg=self.elements[self.pkglistselected]
        i=0
        for error in pkg.errors:
            self.statusbox.bkgdset(ord(' '), self.COLOR_ERRORMSG)
            self.statusbox.addstr(i, 0, 'ERR: '+error, curses.A_BOLD)
            self.statusbox.bkgdset(ord(' '), self.COLOR_STANDARD)
            i+=1
        for warning in pkg.warnings:
            self.statusbox.bkgdset(ord(' '), self.COLOR_WARNINGMSG)
            self.statusbox.addstr(i, 0, 'WARN: '+warning, curses.A_BOLD)
            self.statusbox.bkgdset(ord(' '), self.COLOR_STANDARD)
            i+=1
        self.statusbox.noutrefresh(0,0,self.sbox_y,0,self.scr_h-1,self.scr_w-1)

    def __redraw_pkglist(self):
        # Safety in case there's a bug somewhere
        if self.pkgliststartpad>len(self.elements)-self.pad_h:
            self.pkgliststartpad=len(self.elements)-self.pad_h
        if self.pkgliststartpad<0:
            self.pkgliststartpad=0

        self.pkglist.erase()
        for i in range(len(self.elements)):
            if self.elements[i].marked4moving:
                if self.elements[i].warnings:
                    self.pkglist.bkgdset(ord(' '), self.COLOR_WARNINGMSG)
                    self.pkglist.addstr(i, 0, '+', curses.A_BOLD)
                else:
                    self.pkglist.bkgdset(ord(' '), self.COLOR_WHITEBG)
                    self.pkglist.addstr(i, 0, '+')

            if self.elements[i].errors:
                self.pkglist.bkgdset(ord(' '), self.COLOR_ERROR)
            elif self.elements[i].warnings:
                self.pkglist.bkgdset(ord(' '), self.COLOR_WARNING)
            else:
                self.pkglist.bkgdset(ord(' '), self.COLOR_STANDARD)
            if i == self.pkglistselected:
                self.pkglist.addstr(i, 2, `self.elements[i]`, curses.A_REVERSE)
            else:
                self.pkglist.addstr(i, 2, `self.elements[i]`, curses.A_BOLD)
            self.pkglist.bkgdset(ord(' '), self.COLOR_STANDARD)
        self.pkglist.noutrefresh(self.pkgliststartpad,0,1,0,self.pad_h,self.scr_w-1)

    def __do_resize(self):
        self.__set_screen_coordinates()
        self.__redraw_titlebar()
        self.__redraw_statusbox()

    def __do_move_down(self):
        if 0 <= self.pkglistselected < len(self.elements)-1:
            self.pkglistselected+=1
            if (self.pkgliststartpad!=len(self.elements)-self.pad_h and
                self.pad_h-(self.pkglistselected-self.pkgliststartpad)<5):
                self.pkgliststartpad+=1

    def __do_move_up(self):
        if 0 < self.pkglistselected <= len(self.elements)-1:
            self.pkglistselected-=1
            if self.pkgliststartpad!=0 and self.pkglistselected-self.pkgliststartpad<4:
                self.pkgliststartpad-=1

    def __do_move_home(self):
        self.pkglistselected=0
        self.pkgliststartpad=0

    def __do_move_end(self):
        self.pkglistselected=len(self.elements)-1
        self.pkgliststartpad=len(self.elements)-self.pad_h

    def __do_move_pgdown(self):
        self.pkglistselected+=self.pad_h
        if self.pkglistselected >= len(self.elements):
            self.pkglistselected = len(self.elements)-1
        self.pkgliststartpad+=self.pad_h
        if self.pkgliststartpad > len(self.elements)-self.pad_h:
            self.pkgliststartpad=len(self.elements)-self.pad_h

    def __do_move_pgup(self):
        self.pkglistselected-=self.pad_h
        if self.pkglistselected < 0:
            self.pkglistselected = 0
        self.pkgliststartpad-=self.pad_h
        if self.pkgliststartpad < 0:
            self.pkgliststartpad = 0

    def __do_quit(self):
        raise DoQuit()

    def __do_mark(self):
        pkg=self.elements[self.pkglistselected]
        if pkg.errors:
            return
        elif pkg.marked4moving:
            pkg.unmark4moving()
        else:
            pkg.mark4moving()

    def __handle_input(self):
        key=self.screen.getch()
        if key==27:
            # Switch window
            metakey=self.screen.getch()
            if 0<metakey<256:
                c=chr(metakey)
                if c=='1':
                    self.window=1
                elif c=='2':
                    self.window=2
        elif 0<key<256:
            c=chr(key)
            for bind in self.globalkeys+self.perwindowkeys[self.window-1]:
                if not bind[0].startswith('KEY_') and c in bind[0]:
                    exec 'self._Screen__do_%s()' % bind[1]

        else:
#            self.elements[0]=`key`
            for bind in self.globalkeys+self.perwindowkeys[self.window-1]:
                if bind[0].startswith('KEY_'):
                    if key == getattr(curses, bind[0]):
                        exec 'self._Screen__do_%s()' % bind[1]

s = Screen(sys.argv[1], sys.argv[2], pkglist)
Commit	Line	Data
a1e62e44	1	#!/usr/bin/env python3
fb75f677 ER	2	# vi: encoding=utf-8 ts=8 sts=4 sw=4 et
fb75f677 ER	3
2ec96333 JR	4	from __future__ import print_function
2ec96333 JR	5
fb75f677 ER	6	import sys, os
	7	sys.path.insert(0, os.environ['HOME']+'/pld-ftp-admin/modules')
	8	import ftptree
	9	ftptree.quietmode=True
	10	from common import checkdir
	11	import ftpio
	12	import config
	13	cval=config.value
	14
	15	import curses
	16
	17	if len(sys.argv) != 3:
2ec96333 JR	18	print("ERR: wrong number of arguments")
2ec96333 JR	19	print("check-tree.py srctree dsttree")
fb75f677 ER	20	sys.exit(1)
	21
	22	checkdir(sys.argv[1])
	23
	24	ftpio.connect('check-tree')
	25
	26	if not ftpio.lock(sys.argv[1], True):
2ec96333	27	print("ERR: %s tree already locked" % sys.argv[1])
fb75f677 ER	28	sys.exit(1)
	29
	30	if not ftpio.lock(sys.argv[2], True):
	31	ftpio.unlock(sys.argv[1])
2ec96333	32	print("ERR: %s tree already locked" % sys.argv[2])
fb75f677 ER	33	sys.exit(1)
	34
	35	srctree=ftptree.FtpTree(sys.argv[1], loadall=True)
	36	dsttree=ftptree.FtpTree(sys.argv[2])
	37
	38	srctree.checktree(dsttree)
	39
	40	# We've loaded all interesting info, so we can continue without locks
	41	ftpio.unlock(sys.argv[1])
	42	ftpio.unlock(sys.argv[2])
	43
	44	# The list(set(list)) cast is a fancy way to remove dups from a list
	45	names=sorted(list(set([pkg.name for pkg in srctree.values()])))
	46
	47	pkgtree={}
	48
	49	for n in names:
	50	pkgtree[n]=[]
	51
	52	for pkg in srctree.values():
	53	pkgtree[pkg.name].append(pkg)
	54
	55	for key in pkgtree.keys():
	56	pkgtree[key]=sorted(pkgtree[key], reverse=True)
	57
	58	for pkg in srctree.values():
	59	pkg.marked4movingpool=pkgtree[pkg.name]
	60
	61	pkglist=[]
	62
	63	for key in sorted(pkgtree.keys()):
	64	pkglist=pkglist+pkgtree[key]
	65
	66	DoQuit="OMG!"
	67
	68	class Screen:
	69	def __init__(self, srctreename, dsttreename, elements):
	70	self.srctree=srctreename
	71	self.dsttree=dsttreename
	72	self.elements=elements
	73	self.pkglistselected=0
	74	self.pkgliststartpad=0
	75	self.win2startpad=0
	76	self.window=1
	77	try:
	78	# initial
	79	self.screen=curses.initscr()
	80	self.screen.keypad(1)
	81	curses.start_color()
	82	curses.noecho()
	83	curses.cbreak()
	84	curses.meta(1)
	85	curses.curs_set(0)
	86
	87	# Without this refresh(), nothing will show up before first getch()
	88	self.screen.refresh()
	89
	90	self.__init_colors()
	91	self.__init_windows()
	92	self.__bind_keys()
	93
	94	try:
	95	# Main loop
	96	while True:
97	if self.window==1: #pkglist
98	self.__redraw_pkglist()
99	self.__redraw_statusbox()
100	elif self.window==2:
101	pass
102	curses.doupdate()
103	self.__handle_input()
104	except DoQuit:
105	pass
106	finally:
107	curses.echo()
108	curses.nocbreak()
109	curses.curs_set(1)
110	curses.endwin()
111
112	def __bind_keys(self):
113	self.globalkeys=(('qQ', 'quit'), ('KEY_RESIZE', 'resize'),
114	('KEY_CLOSE', 'quit'))
115	self.perwindowkeys=[(), ()]
116	self.perwindowkeys[0]=(('k', 'move_up'), ('J', 'move_pgdown'),
117	('K', 'move_pgup'), (' ', 'mark'), ('j', 'move_down'),
118	('KEY_DOWN', 'move_down'), ('KEY_UP', 'move_up'),
119	('KEY_HOME', 'move_home'), ('KEY_END', 'move_end'),
120	('KEY_NPAGE', 'move_pgdown'), ('KEY_PPAGE', 'move_pgup'))
121
122
123	def __init_colors(self):
124	curses.init_pair(1, curses.COLOR_BLACK, curses.COLOR_CYAN)
125	self.COLOR_TITLEBAR=curses.color_pair(1)
126	curses.init_pair(2, curses.COLOR_WHITE, curses.COLOR_BLUE)
127	self.COLOR_STANDARD=curses.color_pair(2)
128	curses.init_pair(3, curses.COLOR_RED, curses.COLOR_BLUE)
129	self.COLOR_ERROR=curses.color_pair(3)
130	curses.init_pair(4, curses.COLOR_YELLOW, curses.COLOR_BLUE)
131	self.COLOR_WARNING=curses.color_pair(4)
132	curses.init_pair(5, curses.COLOR_WHITE, curses.COLOR_RED)
133	self.COLOR_ERRORMSG=curses.color_pair(5)
134	curses.init_pair(6, curses.COLOR_WHITE, curses.COLOR_YELLOW)
135	self.COLOR_WARNINGMSG=curses.color_pair(6)
136	curses.init_pair(7, curses.COLOR_BLACK, curses.COLOR_WHITE)
137	self.COLOR_WHITEBG=curses.color_pair(7)
138
139	def __init_windows(self):
140	self.__set_screen_coordinates()
141
142	# Titlebar
143	self.titlebar = curses.newpad(1, 200)
144	self.titlebar.bkgdset(ord(' '), self.COLOR_TITLEBAR)
145	self.__redraw_titlebar()
146
147	# Pkg list pad
148	self.pkglist = curses.newpad(len(self.elements), 200)
149	self.pkglist.bkgdset(ord(' '), self.COLOR_STANDARD)
150	self.pkglist.clrtobot()
151
152	# Status bar
153	self.statusbar = curses.newpad(1, 200)
154
155	# Pkg status box
156	self.statusbox = curses.newpad(5, 200)
157	self.statusbox.bkgdset(ord(' '), self.COLOR_STANDARD)
158	self.statusbox.clrtobot()
159	self.__redraw_statusbox()
160
161	def __set_screen_coordinates(self):
162	self.scr_h, self.scr_w = self.screen.getmaxyx()
163	self.pad_h = self.scr_h-5
164	self.sbar_y = self.scr_h-4
165	self.sbox_y = self.scr_h-3
166
167	def __redraw_titlebar(self):
168	self.titlebar.clrtobot()
169	# self.titlebar.addstr(0, self.scr_w-14, "pld-ftp-admin")
170	self.titlebar.addstr(0, 1, 'Window #%s' % self.window)
171	tmp="%s => %s" % (self.srctree, self.dsttree)
172	self.titlebar.addstr(0, int(self.scr_w/2-len(tmp)/2)-1, tmp)
173	self.titlebar.addstr(0, self.scr_w-20, "Press 'h' for help.")
174	self.titlebar.noutrefresh(0, 0, 0, 0, 1, self.scr_w-1)
175
176	def __redraw_statusbox(self):
177	self.statusbox.erase()
178	pkg=self.elements[self.pkglistselected]
179	i=0
180	for error in pkg.errors:
181	self.statusbox.bkgdset(ord(' '), self.COLOR_ERRORMSG)
182	self.statusbox.addstr(i, 0, 'ERR: '+error, curses.A_BOLD)
183	self.statusbox.bkgdset(ord(' '), self.COLOR_STANDARD)
184	i+=1
185	for warning in pkg.warnings:
186	self.statusbox.bkgdset(ord(' '), self.COLOR_WARNINGMSG)
187	self.statusbox.addstr(i, 0, 'WARN: '+warning, curses.A_BOLD)
188	self.statusbox.bkgdset(ord(' '), self.COLOR_STANDARD)
189	i+=1
190	self.statusbox.noutrefresh(0,0,self.sbox_y,0,self.scr_h-1,self.scr_w-1)
191
192	def __redraw_pkglist(self):
193	# Safety in case there's a bug somewhere
194	if self.pkgliststartpad>len(self.elements)-self.pad_h:
195	self.pkgliststartpad=len(self.elements)-self.pad_h
196	if self.pkgliststartpad<0:
197	self.pkgliststartpad=0
198
199	self.pkglist.erase()
200	for i in range(len(self.elements)):
201	if self.elements[i].marked4moving:
202	if self.elements[i].warnings:
203	self.pkglist.bkgdset(ord(' '), self.COLOR_WARNINGMSG)
204	self.pkglist.addstr(i, 0, '+', curses.A_BOLD)
205	else:
206	self.pkglist.bkgdset(ord(' '), self.COLOR_WHITEBG)
207	self.pkglist.addstr(i, 0, '+')
208
209	if self.elements[i].errors:
210	self.pkglist.bkgdset(ord(' '), self.COLOR_ERROR)
211	elif self.elements[i].warnings:
212	self.pkglist.bkgdset(ord(' '), self.COLOR_WARNING)
213	else:
214	self.pkglist.bkgdset(ord(' '), self.COLOR_STANDARD)
215	if i == self.pkglistselected:
216	self.pkglist.addstr(i, 2, `self.elements[i]`, curses.A_REVERSE)
217	else:
218	self.pkglist.addstr(i, 2, `self.elements[i]`, curses.A_BOLD)
219	self.pkglist.bkgdset(ord(' '), self.COLOR_STANDARD)
220	self.pkglist.noutrefresh(self.pkgliststartpad,0,1,0,self.pad_h,self.scr_w-1)
221
222	def __do_resize(self):
223	self.__set_screen_coordinates()
224	self.__redraw_titlebar()
225	self.__redraw_statusbox()
226
227	def __do_move_down(self):
228	if 0 <= self.pkglistselected < len(self.elements)-1:
229	self.pkglistselected+=1
230	if (self.pkgliststartpad!=len(self.elements)-self.pad_h and
231	self.pad_h-(self.pkglistselected-self.pkgliststartpad)<5):
232	self.pkgliststartpad+=1
233
234	def __do_move_up(self):
235	if 0 < self.pkglistselected <= len(self.elements)-1:
236	self.pkglistselected-=1
237	if self.pkgliststartpad!=0 and self.pkglistselected-self.pkgliststartpad<4:
238	self.pkgliststartpad-=1
239
240	def __do_move_home(self):
241	self.pkglistselected=0
242	self.pkgliststartpad=0
243
244	def __do_move_end(self):
245	self.pkglistselected=len(self.elements)-1
246	self.pkgliststartpad=len(self.elements)-self.pad_h
247
248	def __do_move_pgdown(self):
249	self.pkglistselected+=self.pad_h
250	if self.pkglistselected >= len(self.elements):
251	self.pkglistselected = len(self.elements)-1
252	self.pkgliststartpad+=self.pad_h
253	if self.pkgliststartpad > len(self.elements)-self.pad_h:
254	self.pkgliststartpad=len(self.elements)-self.pad_h
255
256	def __do_move_pgup(self):
257	self.pkglistselected-=self.pad_h
258	if self.pkglistselected < 0:
259	self.pkglistselected = 0
260	self.pkgliststartpad-=self.pad_h
261	if self.pkgliststartpad < 0:
262	self.pkgliststartpad = 0
263
264	def __do_quit(self):
9c170c61	265	raise DoQuit()
fb75f677 ER	266
	267	def __do_mark(self):
	268	pkg=self.elements[self.pkglistselected]
	269	if pkg.errors:
	270	return
	271	elif pkg.marked4moving:
	272	pkg.unmark4moving()
	273	else:
	274	pkg.mark4moving()
	275
	276	def __handle_input(self):
	277	key=self.screen.getch()
	278	if key==27:
	279	# Switch window
	280	metakey=self.screen.getch()
	281	if 0<metakey<256:
	282	c=chr(metakey)
	283	if c=='1':
	284	self.window=1
	285	elif c=='2':
	286	self.window=2
	287	elif 0<key<256:
	288	c=chr(key)
	289	for bind in self.globalkeys+self.perwindowkeys[self.window-1]:
	290	if not bind[0].startswith('KEY_') and c in bind[0]:
	291	exec 'self._Screen__do_%s()' % bind[1]
	292
	293	else:
	294	# self.elements[0]=`key`
	295	for bind in self.globalkeys+self.perwindowkeys[self.window-1]:
	296	if bind[0].startswith('KEY_'):
	297	if key == getattr(curses, bind[0]):
	298	exec 'self._Screen__do_%s()' % bind[1]
	299
	300	s = Screen(sys.argv[1], sys.argv[2], pkglist)
	301