shithub: vdiff

ref: 78df7cace5c34f04746d3b915d236215a7a2ba66
dir: /vdiff.c/

View raw version
#include <u.h>
#include <libc.h>
#include <plumb.h>
#include <draw.h>
#include <event.h>
#include <keyboard.h>
#include <bio.h>

typedef struct Line Line;

struct Line {
	int t;
	char *s;
	char *f;
	int l;
};

enum
{
	Lfile = 0,
	Lsep,
	Ladd,
	Ldel,
	Lnone,
	Ncols,
};	

enum
{
	Scrollwidth = 12,
	Scrollgap = 2,
	Margin = 8,
	Hpadding = 4,
	Vpadding = 2,
};

Rectangle sr;
Rectangle scrollr;
Rectangle scrposr;
Rectangle listr;
Rectangle textr;
Image *cols[Ncols];
Image *bg;
Image *fg;
Image *scrollbg;
int scrollsize;
int lineh;
int nlines;
int offset;
Line **lines;
int lsize;
int lcount;
const char ellipsis[] = "...";

void
drawline(Rectangle r, Line *l)
{
	Image *bg;
	Point p;
	Rune  rn;
	char *s;

	bg = cols[l->t];
	draw(screen, r, bg, nil, ZP);
	p = Pt(r.min.x + Hpadding, r.min.y + (Dy(r)-font->height)/2);
	for(s = l->s; *s; s++){
		if(*s == '\t')
			p = string(screen, p, fg, ZP, font, "    ");
		else if((p.x+Hpadding+stringwidth(font, " ")+stringwidth(font, ellipsis)>=textr.max.x)){
			string(screen, p, fg, ZP, font, ellipsis);
			break;
		}else{
			s += chartorune(&rn, s) - 1;
			p = runestringn(screen, p, fg, ZP, font, &rn, 1);
		}
	}
}

void
redraw(void)
{
	Rectangle lr;
	int i, h, y;

	draw(screen, sr, bg, nil, ZP);
	draw(screen, scrollr, scrollbg, nil, ZP);
	if(lcount>0){
		h = ((double)nlines/lcount)*Dy(scrollr);
		y = ((double)offset/lcount)*Dy(scrollr);
		scrposr = Rect(scrollr.min.x, scrollr.min.y+y, scrollr.max.x-1, scrollr.min.y+y+h);
	}else
		scrposr = Rect(scrollr.min.x, scrollr.min.y, scrollr.max.x-1, scrollr.max.y);
	draw(screen, scrposr, bg, nil, ZP);
	for(i=0; i<nlines && offset+i<lcount; i++){
		lr = Rect(textr.min.x, textr.min.y+i*lineh, textr.max.x, textr.min.y+(i+1)*lineh);
		drawline(lr, lines[offset+i]);
	}
}

void
scroll(int off)
{
	if(off<0 && offset<=0)
		return;
	if(off>0 && offset+nlines>lcount)
		return;
	offset += off;
	if(offset<0)
		offset = 0;
	if(offset+nlines>lcount)
		offset = lcount-nlines+1;
	redraw();
}

int
indexat(Point p)
{
	int n;

	if (!ptinrect(p, textr))
		return -1;
	n = (p.y - textr.min.y) / lineh;
	if ((n+offset) >= lcount)
		return -1;
	return n;
}

void
eresized(int new)
{
	if(new && getwindow(display, Refnone)<0)
		sysfatal("cannot reattach: %r");
	sr = screen->r;
	scrollr = sr;
	scrollr.max.x = scrollr.min.x+Scrollwidth+Scrollgap;
	listr = sr;
	listr.min.x = scrollr.max.x;
	textr = insetrect(listr, Margin);
	lineh = Vpadding+font->height+Vpadding;
	nlines = Dy(textr)/lineh;
	offset = 0;
	scrollsize = mousescrollsize(nlines);
	redraw();
}

void
initcols(void)
{
	Rectangle cr;

	cr = Rect(0, 0, 1, 1);
	bg			= allocimage(display, cr, screen->chan, 1, 0x282828ff);
	fg			= allocimage(display, cr, screen->chan, 1, 0xebdbb2ff);
	cols[Lfile] = allocimage(display, cr, screen->chan, 1, 0xa89984ff);
	cols[Lsep]  = allocimage(display, cr, screen->chan, 1, 0x458588ff);
	cols[Ladd]  = allocimage(display, cr, screen->chan, 1, 0x98971aff);
	cols[Ldel]  = allocimage(display, cr, screen->chan, 1, 0xcc241dff);
	cols[Lnone] = bg;
	scrollbg    = allocimage(display, cr, screen->chan, 1, 0x504945ff);
}

int
linetype(char *text)
{
	int type;

	type = Lnone;
	if(strncmp(text, "+++", 3)==0)
		type = Lfile;
	else if(strncmp(text, "---", 3)==0){
		if(strlen(text) > 4)
			type = Lfile;
	}else if(strncmp(text, "@@", 2)==0)
		type = Lsep;
	else if(strncmp(text, "+", 1)==0)
		type = Ladd;
	else if(strncmp(text, "-", 1)==0)
		type = Ldel;
	return type;
}

Line*
parseline(char *f, int n, char *s)
{
	Line *l;

	l = malloc(sizeof *l);
	if(l==nil)
		sysfatal("malloc: %r");
	l->t = linetype(s);
	l->s = s;
	l->l = n;
	if(l->t != Lfile && l->t != Lsep)
		l->f = f;
	else
		l->f = nil;
	return l;
}

int
lineno(char *s)
{
	char *p, *t[5];
	int n, l;

	p = strdup(s);
	n = tokenize(p, t, 5);
	if(n<=0)
		return -1;
	l = atoi(t[2]);
	free(p);
	return l;
}

void
parse(int fd)
{
	Biobuf *bp;
	Line *l;
	char *s, *f, *t;
	int n, ab;

	ab = 0;
	n = 0;
	f = nil;
	lsize = 64;
	lcount = 0;
	lines = malloc(lsize * sizeof *lines);
	if(lines==nil)
		sysfatal("malloc: %r");
	bp = Bfdopen(fd, OREAD);
	if(bp==nil)
		sysfatal("Bfdopen: %r");
	for(;;){
		s = Brdstr(bp, '\n', 1);
		if(s==nil)
			break;
		l = parseline(f, n, s);
		if(l->t == Lfile && l->s[0] == '-' && strncmp(l->s+4, "a/", 2)==0)
			ab = 1;
		if(l->t == Lfile && l->s[0] == '+'){
			f = l->s+4;
			if(ab && strncmp(f, "b/", 2)==0)
				f += 1;
			t = strchr(f, '\t');
			if(t!=nil)
				*t = 0;
		}else if(l->t == Lsep)
			n = lineno(l->s);
		else if(l->t == Ladd || l->t == Lnone)
			++n;
		lines[lcount++] = l;
		if(lcount>=lsize){
			lsize *= 2;
			lines = realloc(lines, lsize*sizeof *lines);
			if(lines==nil)
				sysfatal("realloc: %r");
		}
	}
}

void
plumb(char *f, int l)
{
	USED(l);
	int fd;
	char wd[256], addr[300]={0};

	fd = plumbopen("send", OWRITE);
	if(fd<0)
		return;
	getwd(wd, sizeof wd);
	snprint(addr, sizeof addr, "%s:%d", f, l);
	plumbsendtext(fd, "vdiff", "edit", wd, addr);
	close(fd);
}

void
main(void)
{
	Event ev;
	int e, n;

	parse(0);
	if(lcount==0){
		fprint(2, "no diff\n");
		exits(nil);
	}
	if(initdraw(nil, nil, "vdiff")<0)
		sysfatal("initdraw: %r");
	initcols();
	einit(Emouse|Ekeyboard);
	eresized(0);
	for(;;){
		e = event(&ev);
		switch(e){
		case Emouse:
			if(ev.mouse.buttons&4 && ptinrect(ev.mouse.xy, scrollr)){
				n = (ev.mouse.xy.y - scrollr.min.y) / lineh;
				if(n<lcount-offset){
					scroll(n);
				} else {
					scroll(lcount-offset);
				}
				break;
			}
			if(ev.mouse.buttons&1 && ptinrect(ev.mouse.xy, scrollr)){
				n = (ev.mouse.xy.y - scrollr.min.y) / lineh;
				if(-n<lcount-offset){
					scroll(-n);
				} else {
					scroll(-lcount+offset);
				}
				break;
			}

			if(ev.mouse.buttons&4){
				n = indexat(ev.mouse.xy);
				if(n>=0 && lines[n+offset]->f != nil)
					plumb(lines[n+offset]->f, lines[n+offset]->l);
			}else if(ev.mouse.buttons&8)
				scroll(-scrollsize);
			else if(ev.mouse.buttons&16)
				scroll(scrollsize);
			break;
		case Ekeyboard:
			switch(ev.kbdc){
			case 'q':
			case Kdel:
				goto End;
				break;
			case Khome:
				scroll(-1000000);
				break;
			case Kend:
				scroll(1000000);
				break;
			case Kpgup:
				scroll(-nlines);
				break;
			case Kpgdown:
				scroll(nlines);
				break;
			}
			break;
		}
	}
End:
	exits(nil);
}