/*
 * Copyright (c) 1992, Brian Berliner and Jeff Polk
 * Copyright (c) 1989-1992, Brian Berliner
 * 
 * You may distribute under the terms of the GNU General Public License as
 * specified in the README file that comes with the CVS source distribution.
 * 
 * The functions in this file provide an interface for performing 
 * operations directly on RCS files. 
 */

#include "cvs.h"
#include <stdio.h>
#include "diffrun.h"

/* This file, rcs.h, and rcs.c, together sometimes known as the "RCS
   library", are intended to define our interface to RCS files.

   Whether there will also be a version of RCS which uses this
   library, or whether the library will be packaged for uses beyond
   CVS or RCS (many people would like such a thing) is an open
   question.  Some considerations:

   1.  An RCS library for CVS must have the capabilities of the
   existing CVS code which accesses RCS files.  In particular, simple
   approaches will often be slow.

   2.  An RCS library should not use code from the current RCS
   (5.7 and its ancestors).  The code has many problems.  Too few
   comments, too many layers of abstraction, too many global variables
   (the correct number for a library is zero), too much intricately
   interwoven functionality, and too many clever hacks.  Paul Eggert,
   the current RCS maintainer, agrees.

   3.  More work needs to be done in terms of separating out the RCS
   library from the rest of CVS (for example, cvs_output should be
   replaced by a callback, and the declarations should be centralized
   into rcs.h, and probably other such cleanups).

   4.  To be useful for RCS and perhaps for other uses, the library
   may need features beyond those needed by CVS.

   5.  Any changes to the RCS file format *must* be compatible.  Many,
   many tools (not just CVS and RCS) can at least import this format.
   RCS and CVS must preserve the current ability to import/export it
   (preferably improved--magic branches are currently a roadblock).
   See doc/RCSFILES in the CVS distribution for documentation of this
   file format.

   On a related note, see the comments at diff_exec, later in this file,
   for more on the diff library.  */

static void RCS_output_diff_options (const char *opts, const char *rev1, const char *rev2, const char *workfile);


/* Stuff to deal with passing arguments the way libdiff.a wants to deal
   with them.  This is a crufty interface; there is no good reason for it
   to resemble a command line rather than something closer to "struct
   log_data" in log.c.  */

/* First call call_diff_setup to setup any initial arguments.  The
   argument will be parsed into whitespace separated words and added
   to the global call_diff_argv list.

   Then, optionally, call call_diff_arg for each additional argument
   that you'd like to pass to the diff library.

   Finally, call call_diff or call_diff3 to produce the diffs.  */

static char **call_diff_argv;
static int call_diff_argc;
static int call_diff_argc_allocated;

static void call_diff_add_arg(const char *);
static void call_diff_setup(const char *prog);
static int call_diff(const char *out);
static int call_diff3(const char *out);

static void call_diff_write_output(const char *, size_t);
static void call_diff_flush_output();
static void call_diff_write_stdout(const char *);
static void call_diff_error(const char *, const char *, const char *);
static void call_diff_cvs_trace();

/* VARARGS */
static void call_diff_setup (const char *prog)
{
    char *cp;
    int i;
    char *call_diff_prog;

    /* clean out any malloc'ed values from call_diff_argv */
    for (i = 0; i < call_diff_argc; i++)
    {
	if (call_diff_argv[i])
	{
	    xfree (call_diff_argv[i]);
	    call_diff_argv[i] = (char *) 0;
	}
    }
    call_diff_argc = 0;

    call_diff_prog = xstrdup (prog);

    /* put each word into call_diff_argv, allocating it as we go */
    for (cp = strtok (call_diff_prog, " \t");
	 cp != NULL;
	 cp = strtok ((char *) NULL, " \t"))
	call_diff_add_arg (cp);
    xfree (call_diff_prog);
}

static void call_diff_arg (const char *s)
{
    call_diff_add_arg (s);
}

static void call_diff_add_arg (const char *s)
{
    /* allocate more argv entries if we've run out */
    if (call_diff_argc >= call_diff_argc_allocated)
    {
	call_diff_argc_allocated += 50;
	call_diff_argv = (char **)
	    xrealloc ((char *) call_diff_argv,
		      call_diff_argc_allocated * sizeof (char **));
    }

    if (s)
	call_diff_argv[call_diff_argc++] = xstrdup (s);
    else
	/* Not post-incremented on purpose!  */
	call_diff_argv[call_diff_argc] = (char *) 0;
}

/* Callback function for the diff library to write data to the output
   file.  This is used when we are producing output to stdout.  */

static void call_diff_write_output (const char *text, size_t len)
{
    if (len > 0)
	cvs_output (text, len);
}

/* Call back function for the diff library to flush the output file.
   This is used when we are producing output to stdout.  */

static void call_diff_flush_output ()
{
    cvs_flushout ();
}

/* Call back function for the diff library to write to stdout.  */

static void call_diff_write_stdout (const char *text)
{
    cvs_output (text, 0);
}

/* Call back function for the diff library to write to stderr.  */

static void call_diff_error (const char *format, const char *a1, const char *a2)
{
    /* FIXME: Should we somehow indicate that this error is coming from
       the diff library?  */
    error (0, 0, format, a1, a2);
}

static void call_diff_cvs_trace (int level, const char *fmt,...)
{
    va_list va;
    char str[1536];

    va_start(va, fmt);
    vsnprintf(str, 1024, fmt, va);
    va_end(va);
    TRACE (level, str);
}

static FILE *call_diff_fopen(const char *file, const char *mode)
{
	return CVS_FOPEN(file,mode);
}

static int call_diff_open(const char *file, int mode, ...)
{
	return CVS_OPEN(file,mode);
}

static int call_diff_stat(const char *fn, struct stat *st)
{
	return CVS_STAT(fn,st);
}

/* This set of callback functions is used if we are sending the diff
   to stdout.  */

static struct diff_callbacks call_diff_stdout_callbacks =
{
    call_diff_write_output,
    call_diff_flush_output,
    call_diff_write_stdout,
    call_diff_error,
	call_diff_cvs_trace,
	call_diff_fopen,
	call_diff_open,
	call_diff_stat
};

/* This set of callback functions is used if we are sending the diff
   to a file.  */

static struct diff_callbacks call_diff_file_callbacks =
{
    (void (*)(const char *, size_t)) NULL,
    (void (*)()) NULL,
    call_diff_write_stdout,
    call_diff_error,
	call_diff_cvs_trace,
	call_diff_fopen,
	call_diff_open,
	call_diff_stat
};

static int call_diff (const char *out)
{
    if (out == RUN_TTY)
		return diff_run (call_diff_argc, call_diff_argv, NULL,
			 &call_diff_stdout_callbacks);
    else
		return diff_run (call_diff_argc, call_diff_argv, out,
			 &call_diff_file_callbacks);
}

static int call_diff3 (const char *out)
{
    if (out == RUN_TTY)
	return diff3_run (call_diff_argc, call_diff_argv, NULL,
			  &call_diff_stdout_callbacks);
    else
	return diff3_run (call_diff_argc, call_diff_argv, out,
			  &call_diff_file_callbacks);
}

/* Merge revisions REV1 and REV2. */

int RCS_merge(RCSNode *rcs, const char *path, const char *workfile, const char *options, const char *rev1, const char *rev2, int conflict_3way, mode_t *mode)
{
    char *xrev1, *xrev2;
    char *tmp1, *tmp2;
    char *diffout = NULL;
    int retval;
	kflag kf;
	int is_unicode, is_bom, is_testing;
	char global_buffer[MAX_PATH];
	char *wf;
	mode_t lmode1, lmode2;

	TRACE(2,"rcs_merge(%s,%s,%s,%s,%s,%s)",PATCH_NULL(rcs->path),PATCH_NULL(path),PATCH_NULL(workfile),PATCH_NULL(options),PATCH_NULL(rev1),PATCH_NULL(rev2));

    if(!CGlobalSettings::GetGlobalValue("cvsnt","PServer","ServerDiffNoUnlink",global_buffer,sizeof(global_buffer)))
    {
		is_testing = atoi(global_buffer);
		TRACE(3,"RCS_merge - is testing");
    }
    else
		is_testing = 0;

	/* Calculate numeric revision numbers from rev1 and rev2 (may be
       symbolic). */
    xrev1 = RCS_gettag (rcs, rev1, 0, NULL);
    xrev2 = RCS_gettag (rcs, rev2, 0, NULL);

	/* They are the same symbolic revision - do nothing */
	if(!strcmp(xrev1,xrev2))
	{
		xfree(xrev1);
		xfree(xrev2);
		return 3; /* No difference */
	}

	cvs_output ("RCS file: ", 0);
    cvs_output (fn_root(rcs->path), 0);
    cvs_output ("\n", 1);

    /* Check out chosen revisions.  The error message when RCS_checkout
       fails is not very informative -- it is taken verbatim from RCS 5.7,
       and relies on RCS_checkout saying something intelligent upon failure. */
    cvs_output ("retrieving revision ", 0);
    cvs_output (xrev1, 0);
    cvs_output ("\n", 1);

    tmp1 = cvs_temp_name();
	RCS_get_kflags(options, true, kf);
	is_bom=0;
	is_unicode = 0;
	if(kf.flags&KFLAG_ENCODED)
	{
		/* even though on the server the Unicode has been converted to UTF-8 it still has BOM */
		if(kf.encoding.bom)
		{
			is_bom = 1;
			TRACE(2,"Unicode file has a BOM.");
		}
		if(!server_active) /* The server already automatically ignores -u */
		{
			static char newopts[128];
			is_unicode = 1;
			kf.flags&=~KFLAG_ENCODED;
			TRACE(3,"Unicode and not server - so rebuilding options without KFLAG_ENCODED.");
			options = RCS_rebuild_options(&kf, newopts);
		}
	}

	if(!options || !*options)
		options="kV";
	if (RCS_checkout (rcs, NULL, xrev1, rev1, options, tmp1,
		      (RCSCHECKOUTPROC)0, NULL, &lmode1))
    {
	cvs_outerr ("rcsmerge: co failed\n", 0);
	error_exit();
    }

    cvs_output ("retrieving revision ", 0);
    cvs_output (xrev2, 0);
    cvs_output ("\n", 1);

    tmp2 = cvs_temp_name();
    if (RCS_checkout (rcs, NULL, xrev2, rev2, options, tmp2,
		      (RCSCHECKOUTPROC)0, NULL, &lmode2))
    {
	cvs_outerr ("rcsmerge: co failed\n", 0);
	error_exit();
    }

	/* getting mode from 2nd revision (target revision) */
	*mode = lmode2;

    /* Merge changes. */
    cvs_output ("Merging differences between ", 0);
    cvs_output (xrev1, 0);
    cvs_output (" and ", 0);
    cvs_output (xrev2, 0);
    cvs_output (" into ", 0);
    cvs_output (workfile, 0);
    cvs_output ("\n", 1);

	if(is_unicode)
	{
		char *unibuf=NULL;
		size_t unisiz,unilen;
		FILE *f;
	    wf = cvs_temp_name();
		TRACE(2,"Converting Unicode %s to UTF-8 %s",PATCH_NULL(workfile),PATCH_NULL(wf));
		get_file(workfile,workfile,kf.flags&KFLAG_BINARY?"rb":"r",&unibuf,&unisiz,&unilen,kf);
		if((f=fopen(wf,kf.flags&KFLAG_BINARY?"wb":"w"))==NULL)
			error(1,errno,"Couldn't create temporary unicode file %s",wf);
		if(fwrite(unibuf,1,unilen,f)!=unilen)
			error(1,errno,"Couldn't write temporary unicode file %s",wf);
		fclose(f);
		xfree(unibuf);
	}
	else
		wf = xstrdup(workfile);

    /* Remember that the first word in the `call_diff_setup' string is used now
       only for diagnostic messages -- CVS no longer forks to run diff3. */
    diffout = cvs_temp_name();
    call_diff_setup ("diff3");
	if (is_testing)
		call_diff_arg ("--nounlink");
	call_diff_arg (conflict_3way?"-A":"-E");
    call_diff_arg ("-am");

    call_diff_arg ("-L");
    call_diff_arg (wf);
    call_diff_arg ("-L");
    call_diff_arg (xrev1);
    call_diff_arg ("-L");
    call_diff_arg (xrev2);

    call_diff_arg (wf);
    call_diff_arg (tmp1);
    call_diff_arg (tmp2);

	TRACE(2,"Calling diff3 with %s, %s, %s",PATCH_NULL(wf),PATCH_NULL(tmp1),PATCH_NULL(tmp2));
    retval = call_diff3 (diffout);
	TRACE(2,"diff3 returned %d",retval);

    if (retval == 1)
		cvs_outerr ("rcsmerge: warning: conflicts during merge\n", 0);
    else if (retval == 2)
		error_exit();
/*	else if (retval == 3)   
		cvs_outerr ("rcsmerge: warning: work file unchanged\n", 0); */

    if (diffout)
	{
		LineType crlf=CRLF_DEFAULT;
		if(kf.flags&KFLAG_MAC)
			crlf=ltCr;
		else if(kf.flags&KFLAG_DOS)
			crlf=ltCrLf;
		else if(kf.flags&(KFLAG_BINARY|KFLAG_UNIX))
			crlf=ltLf;
		if(is_unicode || (crlf!=CRLF_DEFAULT && !server_active))
		{
			CCodepage cdp;
			char *unibuf=NULL;
			size_t unisiz,unilen;
			FILE *f;
			TRACE(2,"Transcoding %s to %s",PATCH_NULL(wf),PATCH_NULL(workfile));
			get_file(diffout,diffout,kf.flags&KFLAG_BINARY?"rb":"r",&unibuf,&unisiz,&unilen,kf);
			if((f=fopen(workfile,kf.flags&KFLAG_BINARY?"wb":"w"))==NULL)
				error(1,errno,"Couldn't create unicode file %s",workfile);
			cdp.BeginEncoding(CCodepage::NullEncoding,kf.encoding);
			if(cdp.OutputAsEncoded(fileno(f),unibuf,unilen,crlf))
				error(1,errno,"Couldn't expand temporary file to unicode %s",workfile);
			cdp.EndEncoding();
			fclose(f);
			xfree(unibuf);
		}
		else
			copy_file (diffout, workfile, 1, 1);
	}

    /* Clean up. */
    {
	int save_noexec = noexec;
	noexec = 0;
	if (unlink_file (tmp1) < 0)
	{
	    if (!existence_error (errno))
		error (0, errno, "cannot remove temp file %s", tmp1);
	}
	xfree (tmp1);
	if (unlink_file (tmp2) < 0)
	{
	    if (!existence_error (errno))
		error (0, errno, "cannot remove temp file %s", tmp2);
	}
	xfree (tmp2);
	if (is_unicode)
	{
		if (unlink_file (wf) < 0)
		{
			if (!existence_error (errno))
			error (0, errno, "cannot remove temp file %s", wf);
		}
	}
	xfree(wf);
	if (diffout)
	{
	    if (unlink_file (diffout) < 0)
	    {
		if (!existence_error (errno))
		    error (0, errno, "cannot remove temp file %s", diffout);
	    }
	    xfree (diffout);
	}
	xfree (xrev1);
	xfree (xrev2);
	noexec = save_noexec;
    }

    return retval;
}

/* Diff revisions and/or files.  OPTS controls the format of the diff
   (it contains options such as "-w -c", &c), or "" for the default.
   OPTIONS controls keyword expansion, as a string starting with "-k",
   or "" to use the default.  REV1 is the first revision to compare
   against; it must be non-NULL.  If REV2 is non-NULL, compare REV1
   and REV2; if REV2 is NULL compare REV1 with the file in the working
   directory, whose name is WORKFILE.  LABEL1 and LABEL2 are default
   file labels, and (if non-NULL) should be added as -L options
   to diff.  Output goes to stdout.

   Return value is 0 for success, -1 for a failure which set errno,
   or positive for a failure which printed a message on stderr.

   This used to exec rcsdiff, but now calls RCS_checkout and diff_exec.

   An issue is what timezone is used for the dates which appear in the
   diff output.  rcsdiff uses the -z flag, which is not presently
   processed by CVS diff, but I'm not sure exactly how hard to worry
   about this--any such features are undocumented in the context of
   CVS, and I'm not sure how important to users.  */
int RCS_exec_rcsdiff (RCSNode *rcsfile, const char *opts, const char *options, const char *rev1,
                      const char *rev2, const char *label1, const char *label2, const char *workfile)
{
    char *tmpfile1;
    char *tmpfile2;
    const char *use_file2;
    int status, retval;

    tmpfile1 = cvs_temp_name ();
    tmpfile2 = NULL;

    cvs_output ("\
===================================================================\n\
RCS file: ", 0);
	cvs_output (fn_root(rcsfile->path), 0);
    cvs_output ("\n", 1);

    /* Historically, `cvs diff' has expanded the $Name keyword to the
       empty string when checking out revisions.  This is an accident,
       but no one has considered the issue thoroughly enough to determine
       what the best behavior is.  Passing NULL for the `nametag' argument
       preserves the existing behavior. */

    cvs_output ("retrieving revision ", 0);
    cvs_output (rev1, 0);
    cvs_output ("\n", 1);
    status = RCS_checkout (rcsfile, NULL, (char*)rev1, NULL, (char*)options, tmpfile1,
			   (RCSCHECKOUTPROC)0, NULL, NULL);
    if (status > 0)
    {
	retval = status;
	goto error_return;
    }
    else if (status < 0)
    {
	error (0, errno,
	       "cannot check out revision %s of %s", rev1, rcsfile->path);
	retval = 1;
	goto error_return;
    }

    if (rev2 == NULL)
    {
	assert (workfile != NULL);
	use_file2 = workfile;
    }
    else
    {
	tmpfile2 = cvs_temp_name ();
	cvs_output ("retrieving revision ", 0);
	cvs_output (rev2, 0);
	cvs_output ("\n", 1);
	status = RCS_checkout (rcsfile, NULL, (char*)rev2, NULL, (char*)options,
			       tmpfile2, (RCSCHECKOUTPROC)0, NULL, NULL);
	if (status > 0)
	{
	    retval = status;
	    goto error_return;
	}
	else if (status < 0)
	{
	    error (0, errno,
		   "cannot check out revision %s of %s", rev2, rcsfile->path);
	    return 1;
	}
	use_file2 = tmpfile2;
    }

    RCS_output_diff_options (opts, rev1, rev2, workfile);
    status = diff_execv (tmpfile1, use_file2, label1, label2, opts, RUN_TTY);
    if (status >= 0)
    {
	retval = status;
	goto error_return;
    }
    else if (status < 0)
    {
	error (0, errno,
	       "cannot diff %s and %s", tmpfile1, use_file2);
	retval = 1;
	goto error_return;
    }

 error_return:
    {
	int save_noexec = noexec;
	noexec = 0;
	if (unlink_file (tmpfile1) < 0)
	{
	    if (!existence_error (errno))
		error (0, errno, "cannot remove temp file %s", tmpfile1);
	}
	noexec = save_noexec;
    }
    xfree (tmpfile1);
    if (tmpfile2 != NULL)
    {
	int save_noexec = noexec;
	noexec = 0;
	if (unlink_file (tmpfile2) < 0)
	{
	    if (!existence_error (errno))
		error (0, errno, "cannot remove temp file %s", tmpfile2);
	}
	noexec = save_noexec;
	xfree (tmpfile2);
    }

    return retval;
}


/* Show differences between two files.  This is the start of a diff library.

   Some issues:

   * Should option parsing be part of the library or the caller?  The
   former allows the library to add options without changing the callers,
   but it causes various problems.  One is that something like --brief really
   wants special handling in CVS, and probably the caller should retain
   some flexibility in this area.  Another is online help (the library could
   have some feature for providing help, but how does that interact with
   the help provided by the caller directly?).  Another is that as things
   stand currently, there is no separate namespace for diff options versus
   "cvs diff" options like -l (that is, if the library adds an option which
   conflicts with a CVS option, it is trouble).

   * This isn't required for a first-cut diff library, but if there
   would be a way for the caller to specify the timestamps that appear
   in the diffs (rather than the library getting them from the files),
   that would clean up the kludgy utime() calls in patch.c.

   Show differences between FILE1 and FILE2.  Either one can be
   DEVNULL to indicate a nonexistent file (same as an empty file
   currently, I suspect, but that may be an issue in and of itself).
   OPTIONS is a list of diff options, or "" if none.  At a minimum,
   CVS expects that -c (update.c, patch.c) and -n (update.c) will be
   supported.  Other options, like -u, --speed-large-files, &c, will
   be specified if the user specified them.

   OUT is a filename to send the diffs to, or RUN_TTY to send them to
   stdout.  Error messages go to stderr.  Return value is 0 for
   success, -1 for a failure which set errno, 1 for success (and some
   differences were found), or >1 for a failure which printed a
   message on stderr.  */

int diff_exec (const char *file1, const char *file2, const char *label1, const char *label2, const char *options, const char *out)
{
    char *args;

	TRACE(2,"diff_exec(%s,%s,%s,%s,%s,%s)",PATCH_NULL(file1),PATCH_NULL(file2),PATCH_NULL(label1),PATCH_NULL(label2),PATCH_NULL(options),PATCH_NULL(out));

    args = (char*)xmalloc (strlen (options) + 10);
    /* The first word in this string is used only for error reporting. */
    sprintf (args, "diff %s", options);
    call_diff_setup (args);
    if (label1)
	call_diff_arg (label1);
    if (label2)
	call_diff_arg (label2);
	call_diff_arg ("--");
    call_diff_arg (file1);
    call_diff_arg (file2);
    xfree (args);

    return call_diff (out);
}

int diff_execv (const char *file1, const char *file2, const char *label1, const char *label2, const char *options, const char *out)
{
    char *args;

	TRACE(2,"diff_execv(%s,%s,%s,%s,%s,%s)",PATCH_NULL(file1),PATCH_NULL(file2),PATCH_NULL(label1),PATCH_NULL(label2),PATCH_NULL(options),PATCH_NULL(out));

    args = (char*)xmalloc (strlen (options) + 10);
    /* The first word in this string is used only for error reporting.  */
    /* I guess we are pretty confident that options starts with a space.  */
    sprintf (args, "diff%s", options);
    call_diff_setup (args);
    if (label1)
	call_diff_arg (label1);
    if (label2)
	call_diff_arg (label2);
	call_diff_arg ("--");
    call_diff_arg (file1);
    call_diff_arg (file2);
    xfree (args);

    return call_diff (out);
}

/* Print the options passed to DIFF, in the format used by rcsdiff.
   The rcsdiff code that produces this output is extremely hairy, and
   it is not clear how rcsdiff decides which options to print and
   which not to print.  The code below reproduces every rcsdiff run
   that I have seen. */

static void RCS_output_diff_options (const char *opts, const char *rev1, const char *rev2, const char *workfile)
{
    char *tmp;

    tmp = (char *) xmalloc (strlen (opts) + strlen (rev1) + 10);

    sprintf (tmp, "diff%s -r%s", opts, rev1);
    cvs_output (tmp, 0);
    xfree (tmp);

    if (rev2)
    {
	cvs_output (" -r", 3);
	cvs_output (rev2, 0);
    }
    else
    {
	assert (workfile != NULL);
	cvs_output (" ", 1);
	cvs_output (workfile, 0);
    }
    cvs_output ("\n", 1);
}