/* cygpath.cc -- convert pathnames between Windows and Unix format
   Copyright 1998, 1999, 2000, 2001, 2002 Red Hat, Inc.

This file is part of Cygwin.

This software is a copyrighted work licensed under the terms of the
Cygwin license.  Please consult the file "CYGWIN_LICENSE" for
details. */

#define NOCOMATTRIBUTE

#include <shlobj.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <limits.h>
#include <getopt.h>
#include <windows.h>
#include <io.h>
#include <sys/fcntl.h>
#include <sys/cygwin.h>
#include <ctype.h>

static char *prog_name;
static char *file_arg;
static char *close_arg;
static int path_flag, unix_flag, windows_flag, absolute_flag;
static int shortname_flag, ignore_flag, allusers_flag, output_flag;

static struct option long_options[] =
{
  { (char *) "help", no_argument, NULL, 'h' },
  { (char *) "absolute", no_argument, NULL, 'a'},
  { (char *) "option", no_argument, NULL, 'o'},
  { (char *) "path", no_argument, NULL, 'p' },
  { (char *) "close", required_argument, (int *) &close_arg, 'c'},
  { (char *) "unix", no_argument, NULL, 'u' },
  { (char *) "file", required_argument, (int *) &file_arg, 'f'},
  { (char *) "version", no_argument, NULL, 'v' },
  { (char *) "windows", no_argument, NULL, 'w' },
  { (char *) "short-name", no_argument, NULL, 's' },
  { (char *) "windir", no_argument, NULL, 'W' },
  { (char *) "sysdir", no_argument, NULL, 'S' },
  { (char *) "ignore", no_argument, NULL, 'i' },
  { (char *) "allusers", no_argument, NULL, 'A' },
  { (char *) "desktop", no_argument, NULL, 'D' },
  { (char *) "smprograms", no_argument, NULL, 'P' },
  { 0, no_argument, 0, 0 }
};

static void
usage (FILE *stream, int status)
{
  if (!ignore_flag || !status)
    fprintf (stream, "\
Usage: %s [-p|--path] (-u|--unix)|(-w|--windows [-s|--short-name]) filename\n\
  -a|--absolute		output absolute path\n\
  -c|--close handle	close handle (for use in captured process)\n\
  -f|--file file	read file for input path information\n\
  -i|--ignore		ignore missing argument\n\
  -p|--path		filename argument is a path\n\
  -s|--short-name	print Windows short form of filename\n\
  -u|--unix		print Unix form of filename\n\
  -v|--version		output version information and exit\n\
  -w|--windows		print Windows form of filename\n\
  -A|--allusers		use `All Users' instead of current user for -D, -P\n\
  -D|--desktop		output `Desktop' directory and exit\n\
  -P|--smprograms	output Start Menu `Programs' directory and exit\n\
  -S|--sysdir		output system directory and exit\n\
  -W|--windir		output `Windows' directory and exit\n",
	   prog_name);
  exit (ignore_flag ? 0 : status);
}

static char *
get_short_paths (char *path)
{
  char *sbuf;
  char *sptr;
  char *next;
  char *ptr = path;
  char *end = strrchr (path, 0);
  DWORD acc = 0;
  DWORD len;

  while (ptr != NULL)
  {
    next = ptr;
    ptr = strchr (ptr, ';');
    if (ptr)
      *ptr++ = 0;
    len = GetShortPathName (next, NULL, 0);
    if (len == ERROR_INVALID_PARAMETER)
    {
      fprintf (stderr, "%s: cannot create short name of %s\n", prog_name, next);
      exit (2);
    }
    acc += len+1;
  }
  sptr = sbuf = (char *) malloc(acc+1);
  if (sbuf == NULL)
  {
    fprintf (stderr, "%s: out of memory\n", prog_name);
    exit (1);
  }
  ptr = path;
  for(;;)
  {
    if (GetShortPathName (ptr, sptr, acc) == ERROR_INVALID_PARAMETER)
    {
      fprintf (stderr, "%s: cannot create short name of %s\n", prog_name, ptr);
      exit (2);
    }

    ptr = strrchr (ptr, 0);
    sptr = strrchr (sptr, 0);
    if (ptr == end)
      break;
    *sptr = ';';
    ++ptr, ++sptr;
  }
  return sbuf;
}

static char *
get_short_name (const char *filename)
{
  char *sbuf;
  DWORD len = GetShortPathName (filename, NULL, 0);
  if (len == ERROR_INVALID_PARAMETER)
  {
    fprintf (stderr, "%s: cannot create short name of %s\n", prog_name, filename);
    exit (2);
  }
  sbuf = (char *) malloc(++len);
  if (sbuf == NULL)
  {
    fprintf (stderr, "%s: out of memory\n", prog_name);
    exit (1);
  }
  if (GetShortPathName (filename, sbuf, len) == ERROR_INVALID_PARAMETER)
  {
    fprintf (stderr, "%s: cannot create short name of %s\n", prog_name, filename);
    exit (2);
  }
  return sbuf;
}

static void
doit (char *filename)
{
  char *buf;
  size_t len;
  int retval;
  int (*conv_func)(const char *, char *);

  if (path_flag)
    {
      if (cygwin_posix_path_list_p (filename)
	  ? unix_flag
	  : windows_flag)
	{
	  /* The path is already in the right format.  */
	  puts (filename);
	  exit (0);
	}
    }

  if (! path_flag)
    {
      len = strlen (filename) + 100;
      if (len == 100)
        {
          if (!ignore_flag)
            {
              fprintf(stderr, "%s: can't convert empty path\n", prog_name);
              exit (1);
            }
          else
            exit (0);
        }
    }
  else
    {
      if (unix_flag)
	len = cygwin_win32_to_posix_path_list_buf_size (filename);
      else
	len = cygwin_posix_to_win32_path_list_buf_size (filename);
    }

  if (len < PATH_MAX)
    len = PATH_MAX;

  buf = (char *) malloc (len);
  if (buf == NULL)
    {
      fprintf (stderr, "%s: out of memory\n", prog_name);
      exit (1);
    }

  if (path_flag)
    {
      if (unix_flag)
	cygwin_win32_to_posix_path_list (filename, buf);
      else
      {
	cygwin_posix_to_win32_path_list (filename, buf);
	if (shortname_flag)
	  buf = get_short_paths (buf);
      }
    }
  else
    {
      if (unix_flag)
	conv_func = (absolute_flag ? cygwin_conv_to_full_posix_path : 
                     cygwin_conv_to_posix_path);
      else
        conv_func = (absolute_flag ? cygwin_conv_to_full_win32_path :
                     cygwin_conv_to_win32_path);
      retval = conv_func (filename, buf);
      if (retval < 0)
        {
          fprintf (stderr, "%s: error converting \"%s\"\n",
                   prog_name, filename);
          exit (1);
        }
      if (!unix_flag && shortname_flag)
        buf = get_short_name (buf);
    }

  puts (buf);
}

int
main (int argc, char **argv)
{
  int c, o;
  int options_from_file_flag;
  char *filename;
  char buf[MAX_PATH], buf2[MAX_PATH];
  WIN32_FIND_DATA w32_fd;
  LPITEMIDLIST id;

  prog_name = strrchr (argv[0], '/');
  if (prog_name == NULL)
    prog_name = strrchr (argv[0], '\\');
  if (prog_name == NULL)
    prog_name = argv[0];
  else
    prog_name++;

  path_flag = 0;
  unix_flag = 0;
  windows_flag = 0;
  shortname_flag = 0;
  ignore_flag = 0;
  options_from_file_flag = 0;
  allusers_flag = 0;
  output_flag = 0;
  while ((c = getopt_long (argc, argv, (char *) "hac:f:opsSuvwWiDPA", long_options, (int *) NULL))
	 != EOF)
    {
      switch (c)
	{
	case 'a':
	  absolute_flag = 1;
	  break;

	case 'c':
	  CloseHandle ((HANDLE) strtoul (optarg, NULL, 16));
	  break;

	case 'f':
	  file_arg = optarg;
	  break;

	case 'o':
	  options_from_file_flag = 1;
	  break;

	case 'p':
	  path_flag = 1;
	  break;

	case 'u':
	  if (unix_flag || windows_flag)
	    usage (stderr, 1);
	  unix_flag = 1;
	  break;

	case 'w':
	  if (unix_flag || windows_flag)
	    usage (stderr, 1);
	  windows_flag = 1;
	  break;

	case 's':
	  if (unix_flag)
	    usage (stderr, 1);
	  shortname_flag = 1;
	  break;

	case 'A':
	  allusers_flag = 1;
	  break;

	case 'D':
	  if (output_flag)
	    usage (stderr, 1);
	  output_flag = 1;
	  o = 'D';
	  break;

	case 'P':
	  if (output_flag)
	    usage (stderr, 1);
	  output_flag = 1;
	  o = 'P';
          break;

	case 'S':
	  if (output_flag)
	    usage (stderr, 1);
	  output_flag = 1;
	  o = 'S';
          break;

	case 'W':
	  if (output_flag)
	    usage (stderr, 1);
	  output_flag = 1;
	  o = 'W';
	  break;

	case 'i':
	  ignore_flag = 1;
	  break;

	case 'h':
	  usage (stdout, 0);
	  break;

	case 'v':
	  printf ("Cygwin path conversion version 1.2\n");
	  printf ("Copyright 1998-2002 Red Hat, Inc.\n");
	  exit (0);

	default:
	  usage (stderr, 1);
	  break;
	}

    }

  if (output_flag) 
    {
      switch(o) {
	case 'D':
	  if (!allusers_flag)
            SHGetSpecialFolderLocation(NULL, CSIDL_DESKTOPDIRECTORY,&id);
	  else
            SHGetSpecialFolderLocation(NULL, CSIDL_COMMON_DESKTOPDIRECTORY,&id);
          SHGetPathFromIDList(id, buf);
          /* This if clause is a Fix for Win95 without any "All Users" */
          if ( strlen(buf) == 0 ) {
            SHGetSpecialFolderLocation(NULL, CSIDL_DESKTOPDIRECTORY,&id);
            SHGetPathFromIDList(id, buf);
          }
	  if (!windows_flag)
	    cygwin_conv_to_posix_path(buf, buf2);
	  else
	    strcpy(buf2, buf);
          printf("%s\n", buf2);
          exit(0);

	case 'P':
	  if (!allusers_flag)
            SHGetSpecialFolderLocation(NULL, CSIDL_PROGRAMS, &id);
	  else
            SHGetSpecialFolderLocation(NULL, CSIDL_COMMON_PROGRAMS, &id);
          SHGetPathFromIDList(id, buf);
          /* This if clause is a Fix for Win95 without any "All Users" */
          if ( strlen(buf) == 0 ) {
            SHGetSpecialFolderLocation(NULL, CSIDL_PROGRAMS, &id);
            SHGetPathFromIDList(id, buf);
          }
	  if (!windows_flag)
	    cygwin_conv_to_posix_path(buf, buf2);
	  else
	    strcpy(buf2, buf);
          printf("%s\n", buf2);
          exit(0);
 
	case 'S':
	  GetSystemDirectory(buf, MAX_PATH);
	  FindFirstFile(buf, &w32_fd);
	  strcpy(strrchr(buf, '\\')+1, w32_fd.cFileName);
	  if (!windows_flag)
	    cygwin_conv_to_posix_path(buf, buf2);
	  else
	    strcpy(buf2, buf);
          printf("%s\n", buf2);
          exit(0);

	case 'W':
	  GetWindowsDirectory(buf, MAX_PATH);
	  if (!windows_flag)
	    cygwin_conv_to_posix_path(buf, buf2);
	  else
	    strcpy(buf2, buf);
          printf("%s\n", buf2);
          exit(0);
 
	default:
	  fprintf(stderr, "ERROR: main: switch(o)!\n");
      }
  }

  if (options_from_file_flag && !file_arg)
    usage (stderr, 1);

  if (! unix_flag && ! windows_flag && !options_from_file_flag)
    usage (stderr, 1);

  if (!file_arg)
    {
      if (optind != argc - 1)
	usage (stderr, 1);

      filename = argv[optind];
      doit (filename);
    }
  else
    {
      FILE *fp;
      char buf[PATH_MAX * 2 + 1];

      if (argv[optind])
	usage (stderr, 1);

      if (strcmp (file_arg, "-") != 0)
	fp = fopen (file_arg, "rt");
      else
	{
	  fp = stdin;
	  setmode (0, O_TEXT);
	}
      if (fp == NULL)
	{
	  perror ("cygpath");
	  exit (1);
	}

      setbuf (stdout, NULL);
      while (fgets (buf, sizeof (buf), fp) != NULL)
	{
	  char *s = buf;
	  char *p = strchr (s, '\n');
	  if (p)
	    *p = '\0';
	  if (options_from_file_flag && *s == '-')
	    {
	      char c;
	      for (c = *++s; c && !isspace (c); c = *++s)
		switch (c)
		  {
		  case 'a':
		    absolute_flag = 1;
		    break;
		  case 'i':
		    ignore_flag = 1;
		    break;
		  case 's':
		    shortname_flag = 1;
		    break;
		  case 'w':
		    unix_flag = 0;
		    windows_flag = 1;
		    break;
		  case 'u':
		    windows_flag = 0;
		    unix_flag = 1;
		    break;
		  case 'p':
		    path_flag = 1;
		  }
	      if (*s)
		do
		  s++;
		while (*s && isspace (*s));
	    }
	  if (*s)
	    doit (s);
	}
    }

  exit (0);
}