/*	$OpenBSD: co.c,v 1.26 2005/11/03 15:40:40 niallo Exp $	*/
/*
 * Copyright (c) 2005 Joris Vink <joris@openbsd.org>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. The name of the author may not be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
 * AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
 * THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL  DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <sys/param.h>
#include <sys/stat.h>

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include "log.h"
#include "rcs.h"
#include "rcsprog.h"

static int checkout_state(RCSFILE *, RCSNUM *, const char *, int,
    const char *, const char *);

int
checkout_main(int argc, char **argv)
{
	int i, ch;
	int flags;
	RCSNUM *frev, *rev;
	RCSFILE *file;
	char fpath[MAXPATHLEN], buf[16];
	char *username;
	const char *state = NULL;

	flags = 0;
	rev = RCS_HEAD_REV;
	frev = NULL;

	if ((username = getlogin()) == NULL) {
		cvs_log(LP_ERRNO, "failed to get username");
		exit (1);
	}

	while ((ch = rcs_getopt(argc, argv, "f::l::M::p::qr::s:u::V")) != -1) {
		switch (ch) {
		case 'f':
			rcs_set_rev(rcs_optarg, &rev);
			flags |= FORCE;
			break;
		case 'l':
			rcs_set_rev(rcs_optarg, &rev);
			flags |= CO_LOCK;
			break;
		case 'M':
			rcs_set_rev(rcs_optarg, &rev);
			flags |= CO_REVDATE;
			break;
		case 'p':
			rcs_set_rev(rcs_optarg, &rev);
			pipeout = 1;
			break;
		case 'q':
			verbose = 0;
			break;
		case 'r':
			rcs_set_rev(rcs_optarg, &rev);
			break;
		case 's':
			if ((state = strdup(rcs_optarg)) == NULL) {
				cvs_log(LP_ERRNO, "out of memory");
				exit(1);
			}
			flags |= CO_STATE;
			break;
		case 'u':
			rcs_set_rev(rcs_optarg, &rev);
			flags |= CO_UNLOCK;
			break;
		case 'V':
			printf("%s\n", rcs_version);
			exit(0);
		default:
			(usage)();
			exit(1);
		}
	}

	argc -= rcs_optind;
	argv += rcs_optind;

	if (argc == 0) {
		cvs_log(LP_ERR, "no input file");
		(usage)();
		exit (1);
	}

	for (i = 0; i < argc; i++) {
		if (rcs_statfile(argv[i], fpath, sizeof(fpath)) < 0)
			continue;

		if (verbose == 1)
			printf("%s  -->  %s\n", fpath,
			    (pipeout == 1) ? "standard output" : argv[i]);

		if ((file = rcs_open(fpath, RCS_RDWR)) == NULL)
			continue;

		if (rev == RCS_HEAD_REV)
			frev = file->rf_head;
		else
			frev = rev;

		rcsnum_tostr(frev, buf, sizeof(buf));

		if (flags & CO_STATE) {
			if (checkout_state(file, frev, argv[i], flags,
			    username, state) < 0) {
				rcs_close(file);
				continue;
			}
		}
		else {
			if (checkout_rev(file, frev, argv[i], flags,
				username) < 0) {
				rcs_close(file);
				continue;
			}
		}

		rcs_close(file);
	}

	if (rev != RCS_HEAD_REV)
		rcsnum_free(frev);

	return (0);
}

void
checkout_usage(void)
{
	fprintf(stderr,
	    "usage: co [-qV] [-l[rev]] [-p[rev]] [-r[rev]] [-sstate]\n"
            "          [-u[rev]] file ...\n");
}

/*
 * Checkout revision <rev> from RCSFILE <file>, writing it to the path <dst>
 * <lkmode> is either LOCK_LOCK or LOCK_UNLOCK or something else
 * (which has no effect).
 * In the case of LOCK_LOCK, a lock is set for <username> if it is not NULL.
 * In the case of LOCK_UNLOCK, all locks are removed for that revision.
 *
 * Returns 0 on success, -1 on failure.
 */
int
checkout_rev(RCSFILE *file, RCSNUM *frev, const char *dst, int flags,
    const char *username)
{
	char buf[16], yn;
	mode_t mode = 0444;
	BUF *bp;
	struct stat st;
	char *content;

	/*
	 * Check out the latest revision if <frev> is greater than HEAD
	 */
	if (rcsnum_cmp(frev, file->rf_head, 0) == -1)
		frev = file->rf_head;

	rcsnum_tostr(frev, buf, sizeof(buf));

	if (verbose == 1)
		printf("revision %s", buf);

	if ((bp = rcs_getrev(file, frev)) == NULL) {
		cvs_log(LP_ERR, "cannot find revision `%s'", buf);
		return (-1);
	}

	if (flags & CO_LOCK) {
		if ((username != NULL)
		    && (rcs_lock_add(file, username, frev) < 0)) {
			if ((rcs_errno != RCS_ERR_DUPENT) && (verbose == 1))
				cvs_log(LP_ERR, "failed to lock '%s'", buf);
		}

		mode = 0644;
		if (verbose == 1)
			printf(" (locked)");
	} else if (flags & CO_UNLOCK) {
		if (rcs_lock_remove(file, frev) < 0) {
			if (rcs_errno != RCS_ERR_NOENT)
				cvs_log(LP_ERR,
				    "failed to remove lock '%s'", buf);
		}

		mode = 0444;
		if (verbose == 1)
			printf(" (unlocked)");
	}

	if (verbose == 1)
		printf("\n");

	if ((stat(dst, &st) != -1) && !(flags & FORCE)) {
		if (st.st_mode & S_IWUSR) {
			yn = 0;
			if (verbose == 0) {
				cvs_log(LP_ERR,
				    "writeable %s exists; checkout aborted",
				    dst);
				return (-1);
			}
			while (yn != 'y' && yn != 'n') {
				printf("writeable %s exists; ", dst);
				printf("remove it? [ny](n): ");
				fflush(stdout);
				yn = getchar();
			}

			if (yn == 'n') {
				cvs_log(LP_ERR, "checkout aborted");
				return (-1);
			}
		}
	}

	if (pipeout == 1) {
		cvs_buf_putc(bp, '\0');
		content = cvs_buf_release(bp);
		printf("%s", content);
		free(content);
	} else {
		if (cvs_buf_write(bp, dst, mode) < 0) {
			cvs_log(LP_ERR, "failed to write revision to file");
			cvs_buf_free(bp);
			return (-1);
		}
		cvs_buf_free(bp);
		if (flags & CO_REVDATE) {
			struct timeval tv[2];
			bzero(&tv, sizeof(tv));
			tv[0].tv_sec = (long)rcs_rev_getdate(file, frev);
			tv[1].tv_sec = tv[0].tv_sec;
			if (utimes(dst, (const struct timeval *)&tv) < 0)
				cvs_log(LP_ERRNO, "error setting utimes");
		}

		if (verbose == 1)
			printf("done\n");
	}

	return (0);
}

/*
 * checkout_state()
 *
 * Search from supplied revision backwards until we find one
 * with state <state> and check that out.
 *
 * Returns 0 on success, -1 on checkout_rev failure.
 */
static int
checkout_state(RCSFILE *rfp, RCSNUM *rev, const char *dst, int flags,
    const char *username, const char *state)
{
	const char *tstate;
	
	if (rev == NULL) {
		cvs_log(LP_ERR, "%s: No revision on branch has state %s",
		    rfp->rf_path, state);
		return (-1);
	}
	else {
		if (((tstate = rcs_state_get(rfp, rev)) != NULL)
		    && (strcmp(state, tstate) == 0))
			return (checkout_rev(rfp, rev, dst, flags, username));
		else
			rev = rcsnum_dec(rev);
		return (checkout_state(rfp, rev, dst, flags, username, state));
	}
}