Differential D34411 Diff 103491 contrib/mg/main.c

Changeset View

Standalone View

contrib/mg/main.c

This file was added.

				/* $OpenBSD: main.c,v 1.90 2021/05/03 12:18:43 lum Exp $ */

				/* This file is in the public domain. */

				/*
				* Mainline.
				*/

				#include <sys/queue.h>
				#include <err.h>
				#include <limits.h>
				#include <locale.h>
				#include <signal.h>
				#include <stdio.h>
				#include <stdlib.h>
				#include <string.h>
				#include <termios.h>
				#include <unistd.h>
				#if defined(__linux__) \|\| defined(__CYGWIN__)
				#include <pty.h>
				#elif defined(__OpenBSD__) \|\| defined(__NetBSD__) \|\| defined(__APPLE__)
				#include <util.h>
				#else
				#include <libutil.h>
				#endif

				#include "def.h"
				#include "kbd.h"
				#include "funmap.h"
				#include "macro.h"

				#ifdef MGLOG
				#include "log.h"
				#endif

				int thisflag; /* flags, this command */
				int lastflag; /* flags, last command */
				int curgoal; /* goal column */
				int startrow; /* row to start */
				int doaudiblebell; /* audible bell toggle */
				int dovisiblebell; /* visible bell toggle */
				int dblspace; /* sentence end #spaces */
				int allbro; /* all buffs read-only */
				int batch; /* for regress tests */
				struct buffer curbp; / current buffer */
				struct buffer bheadp; / BUFFER list head */
				struct mgwin curwp; / current window */
				struct mgwin wheadp; / MGWIN listhead */
				struct vhead varhead; /* Variable list head */
				char pat[NPAT]; /* pattern */

				static void edinit(struct buffer *);
				static void pty_init(void);
				static __dead void usage(void);

				extern char *__progname;
				extern void closetags(void);

				static __dead void
				usage()
				{
				fprintf(stderr, "usage: %s [-nR] [-b file] [-f mode] [-u file] "
				"[+number] [file ...]\n",
				__progname);
				exit(1);
				}

				int
				main(int argc, char **argv)
				{
				char cp, conffile = NULL, *init_fcn_name = NULL;
				char *batchfile = NULL;
				PF init_fcn = NULL;
				int o, i, nfiles;
				int nobackups = 0;
				struct buffer *bp = NULL;

				#ifdef HAVE_PLEDGE
				if (pledge("stdio rpath wpath cpath fattr chown getpw tty proc exec",
				NULL) == -1)
				err(1, "pledge");
				#endif

				while ((o = getopt(argc, argv, "nRb:f:u:")) != -1)
				switch (o) {
				case 'b':
				batch = 1;
				batchfile = optarg;
				break;
				case 'R':
				allbro = 1;
				break;
				case 'n':
				nobackups = 1;
				break;
				case 'f':
				if (init_fcn_name != NULL)
				errx(1, "cannot specify more than one "
				"initial function");
				init_fcn_name = optarg;
				break;
				case 'u':
				conffile = optarg;
				break;
				default:
				usage();
				}

				if (batch && (conffile != NULL)) {
				fprintf(stderr, "%s: -b and -u are mutually exclusive.\n",
				__progname);
				exit(1);
				}
				if (batch) {
				pty_init();
				conffile = batchfile;
				}
				if (conffile != NULL && access(conffile, R_OK) != 0) {
				fprintf(stderr, "%s: Problem with file: %s\n", __progname,
				conffile);
				exit(1);
				}

				argc -= optind;
				argv += optind;

				setlocale(LC_CTYPE, "");

				maps_init(); /* Keymaps and modes. */
				funmap_init(); /* Functions. */

				#ifdef MGLOG
				if (!mgloginit())
				errx(1, "Unable to create logging environment.");
				#endif

				/*
				* This is where we initialize standalone extensions that should
				* be loaded dynamically sometime in the future.
				*/
				{
				extern void grep_init(void);
				extern void cmode_init(void);
				extern void dired_init(void);

				dired_init();
				grep_init();
				cmode_init();
				}

				if (init_fcn_name &&
				(init_fcn = name_function(init_fcn_name)) == NULL)
				errx(1, "Unknown function `%s'", init_fcn_name);

				vtinit(); /* Virtual terminal. */
				dirinit(); /* Get current directory. */
				edinit(bp); /* Buffers, windows. */
				ttykeymapinit(); /* Symbols, bindings. */
				bellinit(); /* Audible and visible bell. */
				dblspace = 1; /* two spaces for sentence end. */

				/*
				* doing update() before reading files causes the error messages from
				* the file I/O show up on the screen. (and also an extra display of
				* the mode line if there are files specified on the command line.)
				*/
				update(CMODE);

				/* user startup file. */
				if ((cp = startupfile(NULL, conffile)) != NULL)
				(void)load(cp);

				if (batch)
				return (0);

				/*
				* Now ensure any default buffer modes from the startup file are
				* given to any files opened when parsing the startup file.
				* Note scratch will also be updated.
				*/
				for (bp = bheadp; bp != NULL; bp = bp->b_bufp) {
				bp->b_flag = defb_flag;
				for (i = 0; i <= defb_nmodes; i++) {
				bp->b_modes[i] = defb_modes[i];
				}
				}

				/* Force FFOTHARG=1 so that this mode is enabled, not simply toggled */
				if (init_fcn)
				init_fcn(FFOTHARG, 1);

				if (nobackups)
				makebkfile(FFARG, 0);

				for (nfiles = 0, i = 0; i < argc; i++) {
				if (argv[i][0] == '+' && strlen(argv[i]) >= 2) {
				long long lval;
				const char *errstr;

				lval = strtonum(&argv[i][1], INT_MIN, INT_MAX, &errstr);
				if (argv[i][1] == '\0' \|\| errstr != NULL)
				goto notnum;
				startrow = lval;
				} else {
				notnum:
				cp = adjustname(argv[i], FALSE);
				if (cp != NULL) {
				if (nfiles == 1)
				splitwind(0, 1);

				if (fisdir(cp) == TRUE) {
				(void)do_dired(cp);
				continue;
				}
				if ((curbp = findbuffer(cp)) == NULL) {
				vttidy();
				errx(1, "Can't find current buffer!");
				}
				(void)showbuffer(curbp, curwp, 0);
				if (readin(cp) != TRUE)
				killbuffer(curbp);
				else {
				/* Ensure enabled, not just toggled */
				if (init_fcn_name)
				init_fcn(FFOTHARG, 1);
				nfiles++;
				}
				if (allbro)
				curbp->b_flag \|= BFREADONLY;
				}
				}
				}

				if (nfiles > 2)
				listbuffers(0, 1);

				/* fake last flags */
				thisflag = 0;
				for (;;) {
				if (epresf == KCLEAR)
				eerase();
				if (epresf == TRUE)
				epresf = KCLEAR;
				if (winch_flag) {
				do_redraw(0, 0, TRUE);
				winch_flag = 0;
				}
				update(CMODE);
				lastflag = thisflag;
				thisflag = 0;

				switch (doin()) {
				case TRUE:
				break;
				case ABORT:
				ewprintf("Quit");
				/* FALLTHRU */
				case FALSE:
				default:
				macrodef = FALSE;
				}
				}
				}

				/*
				* Initialize default buffer and window. Default buffer is called scratch.
				*/
				static void
				edinit(struct buffer *bp)
				{
				struct mgwin *wp;

				bheadp = NULL;
				bp = bfind("scratch", TRUE); /* Text buffer. */
				if (bp == NULL)
				panic("edinit");

				wp = new_window(bp);
				if (wp == NULL)
				panic("edinit: Out of memory");

				curbp = bp; /* Current buffer. */
				wheadp = wp;
				curwp = wp;
				wp->w_wndp = NULL; /* Initialize window. */
				wp->w_linep = wp->w_dotp = bp->b_headp;
				wp->w_ntrows = nrow - 2; /* 2 = mode, echo. */
				wp->w_rflag = WFMODE \| WFFULL; /* Full. */
				}

				/*
				* Create pty for batch mode.
				*/
				static void
				pty_init(void)
				{
				struct winsize ws;
				int master;
				int slave;

				memset(&ws, 0, sizeof(ws));
				ws.ws_col = 80,
				ws.ws_row = 24;

				openpty(&master, &slave, NULL, NULL, &ws);
				login_tty(slave);

				return;
				}

				/*
				* Quit command. If an argument, always quit. Otherwise confirm if a buffer
				* has been changed and not written out. Normally bound to "C-x C-c".
				*/
				/* ARGSUSED */
				int
				quit(int f, int n)
				{
				int s;

				if ((s = anycb(FALSE)) == ABORT)
				return (ABORT);
				if (s == FIOERR \|\| s == UERROR)
				return (FALSE);
				if (s == FALSE
				\|\| eyesno("Modified buffers exist; really exit") == TRUE) {
				vttidy();
				closetags();
				exit(0);
				}
				return (TRUE);
				}

				/*
				* User abort. Should be called by any input routine that sees a C-g to abort
				* whatever C-g is aborting these days. Currently does nothing.
				*/
				/* ARGSUSED */
				int
				ctrlg(int f, int n)
				{
				return (ABORT);
				}