119304Speter/*- 219304Speter * Copyright (c) 1992, 1993, 1994 319304Speter * The Regents of the University of California. All rights reserved. 419304Speter * Copyright (c) 1992, 1993, 1994, 1995, 1996 519304Speter * Keith Bostic. All rights reserved. 619304Speter * 719304Speter * See the LICENSE file for redistribution information. 819304Speter */ 919304Speter 1019304Speter#include "config.h" 1119304Speter 1219304Speter#ifndef lint 13254225Speterstatic const char sccsid[] = "$Id: ex_shift.c,v 10.17 2001/06/25 15:19:20 skimo Exp $"; 1419304Speter#endif /* not lint */ 1519304Speter 1619304Speter#include <sys/types.h> 1719304Speter#include <sys/queue.h> 18254225Speter#include <sys/time.h> 1919304Speter 2019304Speter#include <bitstring.h> 2119304Speter#include <limits.h> 2219304Speter#include <stdio.h> 2319304Speter#include <stdlib.h> 2419304Speter#include <string.h> 2519304Speter 2619304Speter#include "../common/common.h" 2719304Speter 2819304Speterenum which {LEFT, RIGHT}; 2919304Speterstatic int shift __P((SCR *, EXCMD *, enum which)); 3019304Speter 3119304Speter/* 3219304Speter * ex_shiftl -- :<[<...] 3319304Speter * 3419304Speter * 3519304Speter * PUBLIC: int ex_shiftl __P((SCR *, EXCMD *)); 3619304Speter */ 3719304Speterint 38254225Speterex_shiftl(SCR *sp, EXCMD *cmdp) 3919304Speter{ 4019304Speter return (shift(sp, cmdp, LEFT)); 4119304Speter} 4219304Speter 4319304Speter/* 4419304Speter * ex_shiftr -- :>[>...] 4519304Speter * 4619304Speter * PUBLIC: int ex_shiftr __P((SCR *, EXCMD *)); 4719304Speter */ 4819304Speterint 49254225Speterex_shiftr(SCR *sp, EXCMD *cmdp) 5019304Speter{ 5119304Speter return (shift(sp, cmdp, RIGHT)); 5219304Speter} 5319304Speter 5419304Speter/* 5519304Speter * shift -- 5619304Speter * Ex shift support. 5719304Speter */ 5819304Speterstatic int 59254225Spetershift(SCR *sp, EXCMD *cmdp, enum which rl) 6019304Speter{ 6119304Speter recno_t from, to; 6219304Speter size_t blen, len, newcol, newidx, oldcol, oldidx, sw; 6319304Speter int curset; 64254225Speter CHAR_T *p; 65254225Speter CHAR_T *bp, *tbp; 6619304Speter 6719304Speter NEEDFILE(sp, cmdp); 6819304Speter 6919304Speter if (O_VAL(sp, O_SHIFTWIDTH) == 0) { 7019304Speter msgq(sp, M_INFO, "152|shiftwidth option set to 0"); 7119304Speter return (0); 7219304Speter } 7319304Speter 7419304Speter /* Copy the lines being shifted into the unnamed buffer. */ 7519304Speter if (cut(sp, NULL, &cmdp->addr1, &cmdp->addr2, CUT_LINEMODE)) 7619304Speter return (1); 7719304Speter 7819304Speter /* 7919304Speter * The historic version of vi permitted the user to string any number 8019304Speter * of '>' or '<' characters together, resulting in an indent of the 8119304Speter * appropriate levels. There's a special hack in ex_cmd() so that 8219304Speter * cmdp->argv[0] points to the string of '>' or '<' characters. 8319304Speter * 8419304Speter * Q: What's the difference between the people adding features 8519304Speter * to vi and the Girl Scouts? 8619304Speter * A: The Girl Scouts have mint cookies and adult supervision. 8719304Speter */ 8819304Speter for (p = cmdp->argv[0]->bp, sw = 0; *p == '>' || *p == '<'; ++p) 8919304Speter sw += O_VAL(sp, O_SHIFTWIDTH); 9019304Speter 91254225Speter GET_SPACE_RETW(sp, bp, blen, 256); 9219304Speter 9319304Speter curset = 0; 9419304Speter for (from = cmdp->addr1.lno, to = cmdp->addr2.lno; from <= to; ++from) { 9519304Speter if (db_get(sp, from, DBG_FATAL, &p, &len)) 9619304Speter goto err; 9719304Speter if (!len) { 9819304Speter if (sp->lno == from) 9919304Speter curset = 1; 10019304Speter continue; 10119304Speter } 10219304Speter 10319304Speter /* 10419304Speter * Calculate the old indent amount and the number of 10519304Speter * characters it used. 10619304Speter */ 10719304Speter for (oldidx = 0, oldcol = 0; oldidx < len; ++oldidx) 10819304Speter if (p[oldidx] == ' ') 10919304Speter ++oldcol; 11019304Speter else if (p[oldidx] == '\t') 11119304Speter oldcol += O_VAL(sp, O_TABSTOP) - 11219304Speter oldcol % O_VAL(sp, O_TABSTOP); 11319304Speter else 11419304Speter break; 11519304Speter 11619304Speter /* Calculate the new indent amount. */ 11719304Speter if (rl == RIGHT) 11819304Speter newcol = oldcol + sw; 11919304Speter else { 12019304Speter newcol = oldcol < sw ? 0 : oldcol - sw; 12119304Speter if (newcol == oldcol) { 12219304Speter if (sp->lno == from) 12319304Speter curset = 1; 12419304Speter continue; 12519304Speter } 12619304Speter } 12719304Speter 12819304Speter /* Get a buffer that will hold the new line. */ 129254225Speter ADD_SPACE_RETW(sp, bp, blen, newcol + len); 13019304Speter 13119304Speter /* 13219304Speter * Build a new indent string and count the number of 13319304Speter * characters it uses. 13419304Speter */ 13519304Speter for (tbp = bp, newidx = 0; 13619304Speter newcol >= O_VAL(sp, O_TABSTOP); ++newidx) { 13719304Speter *tbp++ = '\t'; 13819304Speter newcol -= O_VAL(sp, O_TABSTOP); 13919304Speter } 14019304Speter for (; newcol > 0; --newcol, ++newidx) 14119304Speter *tbp++ = ' '; 14219304Speter 14319304Speter /* Add the original line. */ 144254225Speter MEMCPY(tbp, p + oldidx, len - oldidx); 14519304Speter 14619304Speter /* Set the replacement line. */ 14719304Speter if (db_set(sp, from, bp, (tbp + (len - oldidx)) - bp)) { 148254225Spetererr: FREE_SPACEW(sp, bp, blen); 14919304Speter return (1); 15019304Speter } 15119304Speter 15219304Speter /* 15319304Speter * !!! 15419304Speter * The shift command in historic vi had the usual bizarre 15519304Speter * collection of cursor semantics. If called from vi, the 15619304Speter * cursor was repositioned to the first non-blank character 15719304Speter * of the lowest numbered line shifted. If called from ex, 15819304Speter * the cursor was repositioned to the first non-blank of the 15919304Speter * highest numbered line shifted. Here, if the cursor isn't 16019304Speter * part of the set of lines that are moved, move it to the 16119304Speter * first non-blank of the last line shifted. (This makes 16219304Speter * ":3>>" in vi work reasonably.) If the cursor is part of 16319304Speter * the shifted lines, it doesn't get moved at all. This 16419304Speter * permits shifting of marked areas, i.e. ">'a." shifts the 16519304Speter * marked area twice, something that couldn't be done with 16619304Speter * historic vi. 16719304Speter */ 16819304Speter if (sp->lno == from) { 16919304Speter curset = 1; 17019304Speter if (newidx > oldidx) 17119304Speter sp->cno += newidx - oldidx; 17219304Speter else if (sp->cno >= oldidx - newidx) 17319304Speter sp->cno -= oldidx - newidx; 17419304Speter } 17519304Speter } 17619304Speter if (!curset) { 17719304Speter sp->lno = to; 17819304Speter sp->cno = 0; 17919304Speter (void)nonblank(sp, to, &sp->cno); 18019304Speter } 18119304Speter 182254225Speter FREE_SPACEW(sp, bp, blen); 18319304Speter 18419304Speter sp->rptlines[L_SHIFT] += cmdp->addr2.lno - cmdp->addr1.lno + 1; 18519304Speter return (0); 18619304Speter} 187