~drizzle-trunk/drizzle/development

/* Copyright (C) 2000-2006 MySQL AB

   This program 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; version 2 of the License.

   This program 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 this program; if not, write to the Free Software

   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */

/* Show databases, tables or columns */

#define SHOW_VERSION "9.10"

#include "client_priv.h"

#include <my_sys.h>

#include <m_string.h>

#include <mysql.h>

#include <mysqld_error.h>

#include <signal.h>

#include <stdarg.h>

static char * host=0, *opt_password=0, *user=0;

static my_bool opt_show_keys= 0, opt_compress= 0, opt_count=0, opt_status= 0;

static my_bool tty_password= 0, opt_table_type= 0;

static my_bool debug_info_flag= 0, debug_check_flag= 0;

static uint my_end_arg= 0;

static uint opt_verbose=0;

static char *default_charset= (char*) MYSQL_DEFAULT_CHARSET_NAME;

#ifdef HAVE_SMEM 

static char *shared_memory_base_name=0;

#endif

static uint opt_protocol=0;

static void get_options(int *argc,char ***argv);

static uint opt_mysql_port=0;

static int list_dbs(MYSQL *mysql,const char *wild);

static int list_tables(MYSQL *mysql,const char *db,const char *table);

static int list_table_status(MYSQL *mysql,const char *db,const char *table);

static int list_fields(MYSQL *mysql,const char *db,const char *table,

		       const char *field);

static void print_header(const char *header,uint head_length,...);

static void print_row(const char *header,uint head_length,...);

static void print_trailer(uint length,...);

static void print_res_header(MYSQL_RES *result);

static void print_res_top(MYSQL_RES *result);

static void print_res_row(MYSQL_RES *result,MYSQL_ROW cur);

static const char *load_default_groups[]= { "mysqlshow","client",0 };

static char * opt_mysql_unix_port=0;

int main(int argc, char **argv)

{

  int error;

  my_bool first_argument_uses_wildcards=0;

  char *wild;

  MYSQL mysql;

  MY_INIT(argv[0]);

  load_defaults("my",load_default_groups,&argc,&argv);

  get_options(&argc,&argv);

  wild=0;

  if (argc)

  {

    char *pos= argv[argc-1], *to;

    for (to= pos ; *pos ; pos++, to++)

    {

      switch (*pos) {

      case '*':

	*pos= '%';

	first_argument_uses_wildcards= 1;

	break;

      case '?':

	*pos= '_';

	first_argument_uses_wildcards= 1;

	break;

      case '%':

      case '_':

	first_argument_uses_wildcards= 1;

	break;

      case '\\':

	pos++;

      default: break;

      }

      *to= *pos;

    }    

    *to= *pos; /* just to copy a '\0'  if '\\' was used */

  }

  if (first_argument_uses_wildcards)

    wild= argv[--argc];

  else if (argc == 3)			/* We only want one field */

    wild= argv[--argc];

  if (argc > 2)

  {

    fprintf(stderr,"%s: Too many arguments\n",my_progname);

    exit(1);

  }

  mysql_init(&mysql);

  if (opt_compress)

    mysql_options(&mysql,MYSQL_OPT_COMPRESS,NullS);

#ifdef HAVE_OPENSSL

  if (opt_use_ssl)

    mysql_ssl_set(&mysql, opt_ssl_key, opt_ssl_cert, opt_ssl_ca,

		  opt_ssl_capath, opt_ssl_cipher);

  mysql_options(&mysql,MYSQL_OPT_SSL_VERIFY_SERVER_CERT,

                (char*)&opt_ssl_verify_server_cert);

#endif

  if (opt_protocol)

    mysql_options(&mysql,MYSQL_OPT_PROTOCOL,(char*)&opt_protocol);

#ifdef HAVE_SMEM

  if (shared_memory_base_name)

    mysql_options(&mysql,MYSQL_SHARED_MEMORY_BASE_NAME,shared_memory_base_name);

#endif

  mysql_options(&mysql, MYSQL_SET_CHARSET_NAME, default_charset);

  if (!(mysql_real_connect(&mysql,host,user,opt_password,

			   (first_argument_uses_wildcards) ? "" :

                           argv[0],opt_mysql_port,opt_mysql_unix_port,

			   0)))

  {

    fprintf(stderr,"%s: %s\n",my_progname,mysql_error(&mysql));

    exit(1);

  }

  mysql.reconnect= 1;

  switch (argc) {

  case 0:  error=list_dbs(&mysql,wild); break;

  case 1:

    if (opt_status)

      error=list_table_status(&mysql,argv[0],wild);

    else

      error=list_tables(&mysql,argv[0],wild);

    break;

  default:

    if (opt_status && ! wild)

      error=list_table_status(&mysql,argv[0],argv[1]);

    else

      error=list_fields(&mysql,argv[0],argv[1],wild);

    break;

  }

  mysql_close(&mysql);			/* Close & free connection */

  if (opt_password)

    my_free(opt_password,MYF(0));

#ifdef HAVE_SMEM

  my_free(shared_memory_base_name,MYF(MY_ALLOW_ZERO_PTR));

#endif

  my_end(my_end_arg);

  exit(error ? 1 : 0);

  return 0;				/* No compiler warnings */

}

static struct my_option my_long_options[] =

{

#ifdef __NETWARE__

  {"autoclose", OPT_AUTO_CLOSE, "Auto close the screen on exit for Netware.",

   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},

#endif

  {"character-sets-dir", 'c', "Directory where character sets are.",

   (uchar**) &charsets_dir, (uchar**) &charsets_dir, 0, GET_STR, REQUIRED_ARG, 0,

   0, 0, 0, 0, 0},

  {"default-character-set", OPT_DEFAULT_CHARSET,

   "Set the default character set.", (uchar**) &default_charset,

   (uchar**) &default_charset, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},

  {"count", OPT_COUNT,

   "Show number of rows per table (may be slow for not MyISAM tables)",

   (uchar**) &opt_count, (uchar**) &opt_count, 0, GET_BOOL, NO_ARG, 0, 0, 0,

   0, 0, 0},

  {"compress", 'C', "Use compression in server/client protocol.",

   (uchar**) &opt_compress, (uchar**) &opt_compress, 0, GET_BOOL, NO_ARG, 0, 0, 0,

   0, 0, 0},

  {"debug", '#', "Output debug log. Often this is 'd:t:o,filename'.",

   0, 0, 0, GET_STR, OPT_ARG, 0, 0, 0, 0, 0, 0},

  {"debug-check", OPT_DEBUG_CHECK, "Check memory and open file usage at exit.",

   (uchar**) &debug_check_flag, (uchar**) &debug_check_flag, 0,

   GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},

  {"debug-info", OPT_DEBUG_INFO, "Print some debug info at exit.",

   (uchar**) &debug_info_flag, (uchar**) &debug_info_flag,

   0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},

  {"help", '?', "Display this help and exit.", 0, 0, 0, GET_NO_ARG, NO_ARG,

   0, 0, 0, 0, 0, 0},

  {"host", 'h', "Connect to host.", (uchar**) &host, (uchar**) &host, 0, GET_STR,

   REQUIRED_ARG, 0, 0, 0, 0, 0, 0},

  {"status", 'i', "Shows a lot of extra information about each table.",

   (uchar**) &opt_status, (uchar**) &opt_status, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0,

   0, 0},

  {"keys", 'k', "Show keys for table.", (uchar**) &opt_show_keys,

   (uchar**) &opt_show_keys, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},

  {"password", 'p',

   "Password to use when connecting to server. If password is not given it's asked from the tty.",

   0, 0, 0, GET_STR, OPT_ARG, 0, 0, 0, 0, 0, 0},

  {"port", 'P', "Port number to use for connection or 0 for default to, in "

   "order of preference, my.cnf, $MYSQL_TCP_PORT, "

#if MYSQL_PORT_DEFAULT == 0

   "/etc/services, "

#endif

   "built-in default (" STRINGIFY_ARG(MYSQL_PORT) ").",

   (uchar**) &opt_mysql_port,

   (uchar**) &opt_mysql_port, 0, GET_UINT, REQUIRED_ARG, 0, 0, 0, 0, 0,

   0},

#ifdef __WIN__

  {"pipe", 'W', "Use named pipes to connect to server.", 0, 0, 0, GET_NO_ARG,

   NO_ARG, 0, 0, 0, 0, 0, 0},

#endif

  {"protocol", OPT_MYSQL_PROTOCOL, "The protocol of connection (tcp,socket,pipe,memory).",

   0, 0, 0, GET_STR,  REQUIRED_ARG, 0, 0, 0, 0, 0, 0},

#ifdef HAVE_SMEM

  {"shared-memory-base-name", OPT_SHARED_MEMORY_BASE_NAME,

   "Base name of shared memory.", (uchar**) &shared_memory_base_name, (uchar**) &shared_memory_base_name,

   0, GET_STR_ALLOC, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},

#endif

  {"show-table-type", 't', "Show table type column.",

   (uchar**) &opt_table_type, (uchar**) &opt_table_type, 0, GET_BOOL,

   NO_ARG, 0, 0, 0, 0, 0, 0},

  {"socket", 'S', "Socket file to use for connection.",

   (uchar**) &opt_mysql_unix_port, (uchar**) &opt_mysql_unix_port, 0, GET_STR,

   REQUIRED_ARG, 0, 0, 0, 0, 0, 0},

#ifndef DONT_ALLOW_USER_CHANGE

  {"user", 'u', "User for login if not current user.", (uchar**) &user,

   (uchar**) &user, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},

#endif

  {"verbose", 'v',

   "More verbose output; You can use this multiple times to get even more verbose output.",

   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},

  {"version", 'V', "Output version information and exit.", 0, 0, 0, GET_NO_ARG,

   NO_ARG, 0, 0, 0, 0, 0, 0},

  {0, 0, 0, 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0}

};

#include <help_start.h>

static void print_version(void)

{

  printf("%s  Ver %s Distrib %s, for %s (%s)\n",my_progname,SHOW_VERSION,

	 MYSQL_SERVER_VERSION,SYSTEM_TYPE,MACHINE_TYPE);

  NETWARE_SET_SCREEN_MODE(1);

}

static void usage(void)

{

  print_version();

  puts("Copyright (C) 2000-2006 MySQL AB");

  puts("This software comes with ABSOLUTELY NO WARRANTY. This is free software,\nand you are welcome to modify and redistribute it under the GPL license\n");

  puts("Shows the structure of a mysql database (databases,tables and columns)\n");

  printf("Usage: %s [OPTIONS] [database [table [column]]]\n",my_progname);

  puts("\n\

If last argument contains a shell or SQL wildcard (*,?,% or _) then only\n\

what\'s matched by the wildcard is shown.\n\

If no database is given then all matching databases are shown.\n\

If no table is given then all matching tables in database are shown\n\

If no column is given then all matching columns and columntypes in table\n\

are shown");

  print_defaults("my",load_default_groups);

  my_print_help(my_long_options);

  my_print_variables(my_long_options);

}

#include <help_end.h>

static my_bool

get_one_option(int optid, const struct my_option *opt __attribute__((unused)),

	       char *argument)

{

  switch(optid) {

#ifdef __NETWARE__

  case OPT_AUTO_CLOSE:

    setscreenmode(SCR_AUTOCLOSE_ON_EXIT);

    break;

#endif

  case 'v':

    opt_verbose++;

    break;

  case 'p':

    if (argument)

    {

      char *start=argument;

      my_free(opt_password,MYF(MY_ALLOW_ZERO_PTR));

      opt_password=my_strdup(argument,MYF(MY_FAE));

      while (*argument) *argument++= 'x';		/* Destroy argument */

      if (*start)

	start[1]=0;				/* Cut length of argument */

      tty_password= 0;

    }

    else

      tty_password=1;

    break;

  case OPT_MYSQL_PROTOCOL:

    opt_protocol= find_type_or_exit(argument, &sql_protocol_typelib,

                                    opt->name);

    break;

  case '#':

    DBUG_PUSH(argument ? argument : "d:t:o");

    debug_check_flag= 1;

    break;

  case 'V':

    print_version();

    exit(0);

    break;

  case '?':

  case 'I':					/* Info */

    usage();

    exit(0);

  }

  return 0;

}

static void

get_options(int *argc,char ***argv)

{

  int ho_error;

  if ((ho_error=handle_options(argc, argv, my_long_options, get_one_option)))

    exit(ho_error);

  if (tty_password)

    opt_password=get_tty_password(NullS);

  if (opt_count)

  {

    /*

      We need to set verbose to 2 as we need to change the output to include

      the number-of-rows column

    */

    opt_verbose= 2;

  }

  if (debug_info_flag)

    my_end_arg= MY_CHECK_ERROR | MY_GIVE_INFO;

  if (debug_check_flag)

    my_end_arg= MY_CHECK_ERROR;

  return;

}

static int

list_dbs(MYSQL *mysql,const char *wild)

{

  const char *header;

  uint length, counter = 0;

  ulong rowcount = 0L;

  char tables[NAME_LEN+1], rows[NAME_LEN+1];

  char query[255];

  MYSQL_FIELD *field;

  MYSQL_RES *result;

  MYSQL_ROW row= NULL, rrow;

  if (!(result=mysql_list_dbs(mysql,wild)))

  {

    fprintf(stderr,"%s: Cannot list databases: %s\n",my_progname,

	    mysql_error(mysql));

    return 1;

  }

  /*

    If a wildcard was used, but there was only one row and it's name is an

    exact match, we'll assume they really wanted to see the contents of that

    database. This is because it is fairly common for database names to

    contain the underscore (_), like INFORMATION_SCHEMA.

   */

  if (wild && mysql_num_rows(result) == 1)

  {

    row= mysql_fetch_row(result);

    if (!my_strcasecmp(&my_charset_latin1, row[0], wild))

    {

      mysql_free_result(result);

      if (opt_status)

        return list_table_status(mysql, wild, NULL);

      else

        return list_tables(mysql, wild, NULL);

    }

  }

  if (wild)

    printf("Wildcard: %s\n",wild);

  header="Databases";

  length=(uint) strlen(header);

  field=mysql_fetch_field(result);

  if (length < field->max_length)

    length=field->max_length;

  if (!opt_verbose)

    print_header(header,length,NullS);

  else if (opt_verbose == 1)

    print_header(header,length,"Tables",6,NullS);

  else

    print_header(header,length,"Tables",6,"Total Rows",12,NullS);

  /* The first row may have already been read up above. */

  while (row || (row= mysql_fetch_row(result)))

  {

    counter++;

    if (opt_verbose)

    {

      if (!(mysql_select_db(mysql,row[0])))

      {

	MYSQL_RES *tresult = mysql_list_tables(mysql,(char*)NULL);

	if (mysql_affected_rows(mysql) > 0)

	{

	  sprintf(tables,"%6lu",(ulong) mysql_affected_rows(mysql));

	  rowcount = 0;

	  if (opt_verbose > 1)

	  {

            /* Print the count of tables and rows for each database */

            MYSQL_ROW trow;

	    while ((trow = mysql_fetch_row(tresult)))

	    {

	      sprintf(query,"SELECT COUNT(*) FROM `%s`",trow[0]);

	      if (!(mysql_query(mysql,query)))

	      {

		MYSQL_RES *rresult;

		if ((rresult = mysql_store_result(mysql)))

		{

		  rrow = mysql_fetch_row(rresult);

		  rowcount += (ulong) strtoull(rrow[0], (char**) 0, 10);

		  mysql_free_result(rresult);

		}

	      }

	    }

	    sprintf(rows,"%12lu",rowcount);

	  }

	}

	else

	{

	  sprintf(tables,"%6d",0);

	  sprintf(rows,"%12d",0);

	}

	mysql_free_result(tresult);

      }

      else

      {

	strmov(tables,"N/A");

	strmov(rows,"N/A");

      }

    }

    if (!opt_verbose)

      print_row(row[0],length,0);

    else if (opt_verbose == 1)

      print_row(row[0],length,tables,6,NullS);

    else

      print_row(row[0],length,tables,6,rows,12,NullS);

    row= NULL;

  }

  print_trailer(length,

		(opt_verbose > 0 ? 6 : 0),

		(opt_verbose > 1 ? 12 :0),

		0);

  if (counter && opt_verbose)

    printf("%u row%s in set.\n",counter,(counter > 1) ? "s" : "");

  mysql_free_result(result);

  return 0;

}

static int

list_tables(MYSQL *mysql,const char *db,const char *table)

{

  const char *header;

  uint head_length, counter = 0;

  char query[255], rows[NAME_LEN], fields[16];

  MYSQL_FIELD *field;

  MYSQL_RES *result;

  MYSQL_ROW row, rrow;

  if (mysql_select_db(mysql,db))

  {

    fprintf(stderr,"%s: Cannot connect to db %s: %s\n",my_progname,db,

	    mysql_error(mysql));

    return 1;

  }

  if (table)

  {

    /*

      We just hijack the 'rows' variable for a bit to store the escaped

      table name

    */

    mysql_real_escape_string(mysql, rows, table, (unsigned long)strlen(table));

    my_snprintf(query, sizeof(query), "show%s tables like '%s'",

                opt_table_type ? " full" : "", rows);

  }

  else

    my_snprintf(query, sizeof(query), "show%s tables",

                opt_table_type ? " full" : "");

  if (mysql_query(mysql, query) || !(result= mysql_store_result(mysql)))

  {

    fprintf(stderr,"%s: Cannot list tables in %s: %s\n",my_progname,db,

	    mysql_error(mysql));

    exit(1);

  }

  printf("Database: %s",db);

  if (table)

    printf("  Wildcard: %s",table);

  putchar('\n');

  header="Tables";

  head_length=(uint) strlen(header);

  field=mysql_fetch_field(result);

  if (head_length < field->max_length)

    head_length=field->max_length;

  if (opt_table_type)

  {

    if (!opt_verbose)

      print_header(header,head_length,"table_type",10,NullS);

    else if (opt_verbose == 1)

      print_header(header,head_length,"table_type",10,"Columns",8,NullS);

    else

    {

      print_header(header,head_length,"table_type",10,"Columns",8,

		   "Total Rows",10,NullS);

    }

  }

  else

  {

    if (!opt_verbose)

      print_header(header,head_length,NullS);

    else if (opt_verbose == 1)

      print_header(header,head_length,"Columns",8,NullS);

    else

      print_header(header,head_length,"Columns",8, "Total Rows",10,NullS);

  }

  while ((row = mysql_fetch_row(result)))

  {

    counter++;

    if (opt_verbose > 0)

    {

      if (!(mysql_select_db(mysql,db)))

      {

	MYSQL_RES *rresult = mysql_list_fields(mysql,row[0],NULL);

	ulong rowcount=0L;

	if (!rresult)

	{

	  strmov(fields,"N/A");

	  strmov(rows,"N/A");

	}

	else

	{

	  sprintf(fields,"%8u",(uint) mysql_num_fields(rresult));

	  mysql_free_result(rresult);

	  if (opt_verbose > 1)

	  {

            /* Print the count of rows for each table */

	    sprintf(query,"SELECT COUNT(*) FROM `%s`",row[0]);

	    if (!(mysql_query(mysql,query)))

	    {

	      if ((rresult = mysql_store_result(mysql)))

	      {

		rrow = mysql_fetch_row(rresult);

		rowcount += (unsigned long) strtoull(rrow[0], (char**) 0, 10);

		mysql_free_result(rresult);

	      }

	      sprintf(rows,"%10lu",rowcount);

	    }

	    else

	      sprintf(rows,"%10d",0);

	  }

	}

      }

      else

      {

	strmov(fields,"N/A");

	strmov(rows,"N/A");

      }

    }

    if (opt_table_type)

    {

      if (!opt_verbose)

	print_row(row[0],head_length,row[1],10,NullS);

      else if (opt_verbose == 1)

	print_row(row[0],head_length,row[1],10,fields,8,NullS);

      else

	print_row(row[0],head_length,row[1],10,fields,8,rows,10,NullS);

    }

    else

    {

      if (!opt_verbose)

	print_row(row[0],head_length,NullS);

      else if (opt_verbose == 1)

	print_row(row[0],head_length, fields,8, NullS);

      else

	print_row(row[0],head_length, fields,8, rows,10, NullS);

    }

  }

  print_trailer(head_length,

		(opt_table_type ? 10 : opt_verbose > 0 ? 8 : 0),

		(opt_table_type ? (opt_verbose > 0 ? 8 : 0) 

		 : (opt_verbose > 1 ? 10 :0)),

		!opt_table_type ? 0 : opt_verbose > 1 ? 10 :0,

		0);

  if (counter && opt_verbose)

    printf("%u row%s in set.\n\n",counter,(counter > 1) ? "s" : "");

  mysql_free_result(result);

  return 0;

}

static int

list_table_status(MYSQL *mysql,const char *db,const char *wild)

{

  char query[1024],*end;

  MYSQL_RES *result;

  MYSQL_ROW row;

  end=strxmov(query,"show table status from `",db,"`",NullS);

  if (wild && wild[0])

    strxmov(end," like '",wild,"'",NullS);

  if (mysql_query(mysql,query) || !(result=mysql_store_result(mysql)))

  {

    fprintf(stderr,"%s: Cannot get status for db: %s, table: %s: %s\n",

	    my_progname,db,wild ? wild : "",mysql_error(mysql));

    if (mysql_errno(mysql) == ER_PARSE_ERROR)

      fprintf(stderr,"This error probably means that your MySQL server doesn't support the\n\'show table status' command.\n");

    return 1;

  }

  printf("Database: %s",db);

  if (wild)

    printf("  Wildcard: %s",wild);

  putchar('\n');

  print_res_header(result);

  while ((row=mysql_fetch_row(result)))

    print_res_row(result,row);

  print_res_top(result);

  mysql_free_result(result);

  return 0;

}

/*

  list fields uses field interface as an example of how to parse

  a MYSQL FIELD

*/

static int

list_fields(MYSQL *mysql,const char *db,const char *table,

	    const char *wild)

{

  char query[1024],*end;

  MYSQL_RES *result;

  MYSQL_ROW row;

  ulong rows= 0;

  if (mysql_select_db(mysql,db))

  {

    fprintf(stderr,"%s: Cannot connect to db: %s: %s\n",my_progname,db,

	    mysql_error(mysql));

    return 1;

  }

  if (opt_count)

  {

    sprintf(query,"select count(*) from `%s`", table);

    if (mysql_query(mysql,query) || !(result=mysql_store_result(mysql)))

    {

      fprintf(stderr,"%s: Cannot get record count for db: %s, table: %s: %s\n",

              my_progname,db,table,mysql_error(mysql));

      return 1;

    }

    row= mysql_fetch_row(result);

    rows= (ulong) strtoull(row[0], (char**) 0, 10);

    mysql_free_result(result);

  }

  end=strmov(strmov(strmov(query,"show /*!32332 FULL */ columns from `"),table),"`");

  if (wild && wild[0])

    strxmov(end," like '",wild,"'",NullS);

  if (mysql_query(mysql,query) || !(result=mysql_store_result(mysql)))

  {

    fprintf(stderr,"%s: Cannot list columns in db: %s, table: %s: %s\n",

	    my_progname,db,table,mysql_error(mysql));

    return 1;

  }

  printf("Database: %s  Table: %s", db, table);

  if (opt_count)

    printf("  Rows: %lu", rows);

  if (wild && wild[0])