/* Specific flags and argument handling of the front-end of the 
   GNU compiler for the Java(TM) language.
   Copyright (C) 1996, 97-99, 2000 Free Software Foundation, Inc.

This file is part of GNU CC.

GNU CC is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2, or (at your option)
any later version.

GNU CC is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with GNU CC; see the file COPYING.  If not, write to
the Free Software Foundation, 59 Temple Place - Suite 330,
Boston, MA 02111-1307, USA. 

Java and all Java-based marks are trademarks or registered trademarks
of Sun Microsystems, Inc. in the United States and other countries.
The Free Software Foundation is independent of Sun Microsystems, Inc.  */

#include "config.h"
#include "system.h"
#include "gcc.h"

/* Name of spec file.  */
#define SPEC_FILE "libgcj.spec"

/* This bit is set if we saw a `-xfoo' language specification.  */
#define LANGSPEC	(1<<1)
/* True if this arg is a parameter to the previous option-taking arg. */
#define PARAM_ARG	(1<<2)
/* True if this arg is a .java input file name. */
#define JAVA_FILE_ARG	(1<<3)
/* True if this arg is a .class input file name. */
#define CLASS_FILE_ARG	(1<<4)

static char *find_spec_file	PARAMS ((const char *));

static const char *main_class_name = NULL;
int lang_specific_extra_outfiles = 0;

/* Once we have the proper support in jc1 (and gcc.c) working,
   set COMBINE_INPUTS to one.  This enables combining multiple *.java
   and *.class input files to be passed to a single jc1 invocation. */
#define COMBINE_INPUTS 0

const char jvgenmain_spec[] =
  "jvgenmain %{D*} %i %{!pipe:%umain.i} |\n\
   cc1 %{!pipe:%Umain.i} %1 \
		   %{!Q:-quiet} -dumpbase %b.c %{d*} %{m*} %{a*}\
		   %{g*} %{O*} \
		   %{v:-version} %{pg:-p} %{p} %{f*}\
		   %{aux-info*}\
		   %{pg:%{fomit-frame-pointer:%e-pg and -fomit-frame-pointer are incompatible}}\
		   %{S:%W{o*}%{!o*:-o %b.s}}%{!S:-o %{|!pipe:%Umain.s}} |\n\
              %{!S:as %a %Y -o %d%w%umain%O %{!pipe:%Umain.s} %A\n }";

/* Return full path name of spec file if it is in DIR, or NULL if
   not.  */
static char *
find_spec_file (dir)
     const char *dir;
{
  char *spec;
  int x;
  struct stat sb;

  spec = (char *) xmalloc (strlen (dir) + sizeof (SPEC_FILE)
			   + sizeof ("-specs=") + 4);
  strcpy (spec, "-specs=");
  x = strlen (spec);
  strcat (spec, dir);
  strcat (spec, "/");
  strcat (spec, SPEC_FILE);
  if (! stat (spec + x, &sb))
    return spec;
  free (spec);
  return NULL;
}

void
lang_specific_driver (in_argc, in_argv, in_added_libraries)
     int *in_argc;
     char ***in_argv;
     int *in_added_libraries;
{
  int i, j;

  /* If non-zero, the user gave us the `-v' flag.  */ 
  int saw_verbose_flag = 0;

  /* This will be 0 if we encounter a situation where we should not
     link in libgcj.  */
  int library = 1;

#if COMBINE_INPUTS
  /* This will be 1 if multiple input files (.class and/or .java)
     should be passed to a single jc1 invocation. */
  int combine_inputs = 0;

  /* Index of last .java or .class argument. */
  int last_input_index;

  /* A buffer containing the concatenation of the inputs files
     (e.g. "foo.java&bar.class&baz.class"). if combine_inputs. */
  char* combined_inputs_buffer;

  /* Next available location in combined_inputs_buffer. */
  int combined_inputs_pos;

  /* Number of .java and .class source file arguments seen. */
  int java_files_count = 0;
  int class_files_count = 0;

  /* Cumulative length of the  .java and .class source file names. */
  int java_files_length = 0;
  int class_files_length = 0;
#endif

  /* The number of arguments being added to what's in argv, other than
     libraries.  We use this to track the number of times we've inserted
     -xc++/-xnone.  */
  int added = 2;

  /* Used to track options that take arguments, so we don't go wrapping
     those with -xc++/-xnone.  */
  const char *quote = NULL;

  /* The new argument list will be contained in this.  */
  char **real_arglist;
  const char **arglist;

  /* Non-zero if we saw a `-xfoo' language specification on the
     command line.  Used to avoid adding our own -xc++ if the user
     already gave a language for the file.  */
  int saw_speclang = 0;

  /* "-lm" or "-lmath" if it appears on the command line.  */
  const char *saw_math ATTRIBUTE_UNUSED = 0;

  /* "-lc" if it appears on the command line.  */
  const char *saw_libc ATTRIBUTE_UNUSED = 0;

  /* "-lgcjgc" if it appears on the command line.  */
  const char *saw_gc ATTRIBUTE_UNUSED = 0;

  /* Saw `-l' option for the thread library.  */
  const char *saw_threadlib ATTRIBUTE_UNUSED = 0;

  /* Saw `-lgcj' on command line.  */
  int saw_libgcj ATTRIBUTE_UNUSED = 0;

  /* Saw -C or -o option, respectively. */
  int saw_C = 0;
  int saw_o = 0;

  /* Saw some -O* or -g* option, respectively. */
  int saw_O = 0;
  int saw_g = 0;

  /* Saw a `-D' option.  */
  int saw_D = 0;

  /* An array used to flag each argument that needs a bit set for
     LANGSPEC, MATHLIB, WITHLIBC, or GCLIB.  */
  int *args;

  /* The total number of arguments with the new stuff.  */
  int argc;

  /* The argument list.  */
  char **argv;

  /* The number of libraries added in.  */
  int added_libraries;

  /* The total number of arguments with the new stuff.  */
  int num_args = 1;

  /* Non-zero if linking is supposed to happen.  */
  int will_link = 1;

  /* Non-zero if we want to find the spec file.  */
  int want_spec_file = 1;

  /* The argument we use to specify the spec file.  */
  char *spec_file = NULL;

  argc = *in_argc;
  argv = *in_argv;
  added_libraries = *in_added_libraries;

  args = (int *) xcalloc (argc, sizeof (int));

  for (i = 1; i < argc; i++)
    {
      /* If the previous option took an argument, we swallow it here.  */
      if (quote)
	{
	  quote = NULL;
	  args[i] |= PARAM_ARG;
	  continue;
	}

      /* We don't do this anymore, since we don't get them with minus
	 signs on them.  */
      if (argv[i][0] == '\0' || argv[i][1] == '\0')
	continue;

      if (argv[i][0] == '-')
	{
	  if (library != 0 && (strcmp (argv[i], "-nostdlib") == 0
			       || strcmp (argv[i], "-nodefaultlibs") == 0))
	    {
	      library = 0;
	    }
	  else if (strncmp (argv[i], "-fmain=", 7) == 0)
	    {
	      main_class_name = argv[i] + 7;
	      added--;
	    }
	  else if (strcmp (argv[i], "-fhelp") == 0)
	    want_spec_file = 0;
	  else if (strcmp (argv[i], "-v") == 0)
	    {
	      saw_verbose_flag = 1;
	      if (argc == 2)
		{
		  /* If they only gave us `-v', don't try to link
		     in libgcj.  */ 
		  library = 0;
		}
	    }
	  else if (strncmp (argv[i], "-x", 2) == 0)
	    saw_speclang = 1;
	  else if (strcmp (argv[i], "-C") == 0)
	    {
	      saw_C = 1;
	      want_spec_file = 0;
#if COMBINE_INPUTS
	      combine_inputs = 1;
#endif
	      if (library != 0)
		added -= 2;
	      library = 0;
	      will_link = 0;
	    }
	  else if (argv[i][1] == 'D')
	    saw_D = 1;
	  else if (argv[i][1] == 'g')
	    saw_g = 1;
	  else if (argv[i][1] == 'O')
	    saw_O = 1;
	  else if (((argv[i][2] == '\0'
		     && (char *)strchr ("bBVDUoeTuIYmLiA", argv[i][1]) != NULL)
		    || strcmp (argv[i], "-Tdata") == 0))
	    {
	      if (strcmp (argv[i], "-o") == 0)
		saw_o = 1;
	      quote = argv[i];
	    }
	  else if (strcmp(argv[i], "-classpath") == 0
		   || strcmp(argv[i], "-CLASSPATH") == 0)
	    {
	      quote = argv[i];
	      added -= 1;
	    }
	  else if (library != 0 
		   && ((argv[i][2] == '\0'
			&& (char *) strchr ("cSEM", argv[i][1]) != NULL)
		       || strcmp (argv[i], "-MM") == 0))
	    {
	      /* Don't specify libraries if we won't link, since that would
		 cause a warning.  */
	      library = 0;
	      added -= 2;

	      /* Remember this so we can confirm -fmain option.  */
	      will_link = 0;
	    }
	  else if (strcmp (argv[i], "-d") == 0)
	    {
	      /* `-d' option is for javac compatibility.  */
	      quote = argv[i];
	      added -= 1;
	    }
	  else if (strcmp (argv[i], "-fsyntax-only") == 0
		   || strcmp (argv[i], "--syntax-only") == 0)
	    {
	      want_spec_file = 0;
	      library = 0;
	      will_link = 0;
	      continue;
	    }
	  else
	    /* Pass other options through.  */
	    continue;
	}
      else
	{
#if COMBINE_INPUTS
	  int len; 
#endif

	  if (saw_speclang)
	    {
	      saw_speclang = 0;
	      continue;
	    }

#if COMBINE_INPUTS
	  len = strlen (argv[i]);
	  if (len > 5 && strcmp (argv[i] + len - 5, ".java") == 0)
	    {
	      args[i] |= JAVA_FILE_ARG;
	      java_files_count++;
	      java_files_length += len;
	      last_input_index = i;
	    }
	  if (len > 6 && strcmp (argv[i] + len - 6, ".class") == 0)
	    {
	      args[i] |= CLASS_FILE_ARG;
	      class_files_count++;
	      class_files_length += len;
	      last_input_index = i;
	    }
#endif
	}
    }

  if (quote)
    fatal ("argument to `%s' missing\n", quote);

  if (saw_D && ! main_class_name)
    fatal ("can't specify `-D' without `--main'\n");

  num_args = argc + added;
  if (saw_C)
    {
      num_args += 3;
#if COMBINE_INPUTS
      class_files_length = 0;
      num_args -= class_files_count;
      num_args += 2;  /* For -o NONE. */
#endif
      if (saw_o)
	fatal ("cannot specify both -C and -o");
    }
#if COMBINE_INPUTS
  if (saw_o && java_files_count + (saw_C ? 0 : class_files_count) > 1)
    combine_inputs = 1;

  if (combine_inputs)
    {
      int len = java_files_length + java_files_count - 1;
      num_args -= java_files_count;
      num_args++;  /* Add one for the combined arg. */
      if (class_files_length > 0)
	{
	  len += class_files_length + class_files_count - 1;
	  num_args -= class_files_count;
	}
      combined_inputs_buffer = (char*) xmalloc (len);
      combined_inputs_pos = 0;
    }
  /* If we know we don't have to do anything, bail now.  */
#endif
#if 0
  if (! added && ! library && main_class_name == NULL && ! saw_C)
    {
      free (args);
      return;
    }
#endif

  if (main_class_name)
    {
      lang_specific_extra_outfiles++;
    }
  if (saw_g + saw_O == 0)
    num_args++;
  num_args++;
  arglist = (const char **)
    (real_arglist = (char **) xmalloc ((num_args + 1) * sizeof (char *)));

  for (i = 0, j = 0; i < argc; i++, j++)
    {
      arglist[j] = argv[i];

      if ((args[i] & PARAM_ARG) || i == 0)
	continue;

      if (strcmp (argv[i], "-classpath") == 0
	  || strcmp (argv[i], "-CLASSPATH") == 0)
	{
	  char* patharg
	    = (char*) xmalloc (strlen (argv[i]) + strlen (argv[i+1]) + 3);
	  sprintf (patharg, "-f%s=%s", argv[i]+1, argv[i+1]);
	  arglist[j] = patharg;
	  i++;
	  continue;
	}

      if (strcmp (argv[i], "-d") == 0)
	{
	  char *patharg = (char *) xmalloc (sizeof ("-foutput-class-dir=")
					    + strlen (argv[i + 1]) + 1);
	  sprintf (patharg, "-foutput-class-dir=%s", argv[i + 1]);
	  arglist[j] = patharg;
	  ++i;
	  continue;
	}

      if (spec_file == NULL && strncmp (argv[i], "-L", 2) == 0)
	spec_file = find_spec_file (argv[i] + 2);

      if (strncmp (argv[i], "-fmain=", 7) == 0)
	{
	  if (! will_link)
	    fatal ("cannot specify `main' class when not linking");
	  --j;
	  continue;
	}

      if ((args[i] & CLASS_FILE_ARG) && saw_C)
	{
	  --j;
	  continue;
	}

#if COMBINE_INPUTS
      if (combine_inputs && (args[i] & (CLASS_FILE_ARG|JAVA_FILE_ARG)) != 0)
	{
	  if (combined_inputs_pos > 0)
	    combined_inputs_buffer[combined_inputs_pos++] = '&';
	  strcpy (&combined_inputs_buffer[combined_inputs_pos], argv[i]);
	  combined_inputs_pos += strlen (argv[i]);
	  --j;
	  continue;
	}
#endif
  }

#if COMBINE_INPUTS
  if (combine_inputs)
    {
      combined_inputs_buffer[combined_inputs_pos] = '\0';
#if 0
      if (! saw_C)
#endif
      arglist[j++] = combined_inputs_buffer;
    }
#endif

  /* If we saw no -O or -g option, default to -g1, for javac compatibility. */
  if (saw_g + saw_O == 0)
    arglist[j++] = "-g1";

  /* Read the specs file corresponding to libgcj.
     If we didn't find the spec file on the -L path, then we hope it
     is somewhere in the standard install areas.  */
  if (want_spec_file)
    arglist[j++] = spec_file == NULL ? "-specs=libgcj.spec" : spec_file;

  if (saw_C)
    {
      arglist[j++] = "-fsyntax-only";
      arglist[j++] = "-femit-class-files";
      arglist[j++] = "-S";
#if COMBINE_INPUTS
      arglist[j++] = "-o";
      arglist[j++] = "NONE";
#endif
    }

  arglist[j] = NULL;

  *in_argc = j;
  *in_argv = real_arglist;
  *in_added_libraries = added_libraries;
}

int
lang_specific_pre_link ()
{
  if (main_class_name == NULL)
    return 0;
  input_filename = main_class_name;
  input_filename_length = strlen (main_class_name);
  return do_spec (jvgenmain_spec);
}