shellAuto.c 53.0 KB
Newer Older
A
Alex Duan 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * 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.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#define __USE_XOPEN

#include "shellInt.h"
#include "shellAuto.h"
A
Alex Duan 已提交
20
#include "shellTire.h"
A
Alex Duan 已提交
21 22 23 24 25 26 27 28
#include "tthread.h"

//
// ------------- define area  ---------------
//
#define UNION_ALL " union all "

// extern function
29 30 31 32
void shellClearScreen(int32_t ecmd_pos, int32_t cursor_pos);
void shellGetPrevCharSize(const char* str, int32_t pos, int32_t* size, int32_t* width);
void shellShowOnScreen(SShellCmd* cmd);
void shellInsertChar(SShellCmd* cmd, char* c, int size);
A
Alex Duan 已提交
33
void shellInsertStr(SShellCmd* cmd, char* str, int size);
34
bool appendAfterSelect(TAOS* con, SShellCmd* cmd, char* p, int32_t len);
35
char* tireSearchWord(int type, char* pre);
A
Alex Duan 已提交
36
bool updateTireValue(int type, bool autoFill) ;
A
Alex Duan 已提交
37 38 39

typedef struct SAutoPtr {
  STire* p;
40 41
  int    ref;
} SAutoPtr;
A
Alex Duan 已提交
42

43 44 45 46 47 48 49
typedef struct SWord {
  int           type;  // word type , see WT_ define
  char*         word;
  int32_t       len;
  struct SWord* next;
  bool          free;  // if true need free
} SWord;
A
Alex Duan 已提交
50 51

typedef struct {
52 53
  char*   source;
  int32_t source_len;  // valid data length in source
A
Alex Duan 已提交
54 55 56
  int32_t count;
  SWord*  head;
  // matched information
57 58 59
  int32_t matchIndex;  // matched word index in words
  int32_t matchLen;    // matched length at matched word
} SWords;
A
Alex Duan 已提交
60 61

SWords shellCommands[] = {
62 63 64
    {"alter database <db_name> <alter_db_options> <anyword> <alter_db_options> <anyword> <alter_db_options> <anyword> "
     "<alter_db_options> <anyword> <alter_db_options> <anyword> ;",
     0, 0, NULL},
65 66 67 68 69 70
    {"alter dnode <dnode_id> \"resetlog\";", 0, 0, NULL},
    {"alter dnode <dnode_id> \"debugFlag\" \"141\";", 0, 0, NULL},
    {"alter dnode <dnode_id> \"monitor\" \"0\";", 0, 0, NULL},
    {"alter dnode <dnode_id> \"monitor\" \"1\";", 0, 0, NULL},
    {"alter all dnodes \"resetlog\";", 0, 0, NULL},
    {"alter all dnodes \"debugFlag\" \"141\";", 0, 0, NULL},
A
Alex Duan 已提交
71
    {"alter all dnodes \"monitor\" \"0\";", 0, 0, NULL},
72
    {"alter all dnodes \"monitor\" \"1\";", 0, 0, NULL},
73 74
    {"alter table <tb_name> <tb_actions> <anyword> ;", 0, 0, NULL},
    {"alter table modify column", 0, 0, NULL},
75 76 77 78 79 80
    {"alter local \"resetlog\";", 0, 0, NULL},
    {"alter local \"DebugFlag\" \"143\";", 0, 0, NULL},
    {"alter local \"cDebugFlag\" \"143\";", 0, 0, NULL},
    {"alter local \"uDebugFlag\" \"143\";", 0, 0, NULL},
    {"alter local \"rpcDebugFlag\" \"143\";", 0, 0, NULL},
    {"alter local \"tmrDebugFlag\" \"143\";", 0, 0, NULL},
81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111
    {"alter topic", 0, 0, NULL},
    {"alter user <user_name> <user_actions> <anyword> ;", 0, 0, NULL},
    // 20
    {"create table <anyword> using <stb_name> tags(", 0, 0, NULL},
    {"create database <anyword> <db_options> <anyword> <db_options> <anyword> <db_options> <anyword> <db_options> "
     "<anyword> <db_options> <anyword> <db_options> <anyword> <db_options> <anyword> <db_options> <anyword> "
     "<db_options> <anyword> <db_options> <anyword> ;",
     0, 0, NULL},
    {"create dnode ", 0, 0, NULL},
    {"create index ", 0, 0, NULL},
    {"create mnode on dnode <dnode_id> ;", 0, 0, NULL},
    {"create qnode on dnode <dnode_id> ;", 0, 0, NULL},
    {"create stream <anyword> into <anyword> as select", 0, 0, NULL},  // 26 append sub sql
    {"create topic <anyword> as select", 0, 0, NULL},                  // 27 append sub sql
    {"create function ", 0, 0, NULL},
    {"create user <anyword> pass <anyword> sysinfo 0;", 0, 0, NULL},
    {"create user <anyword> pass <anyword> sysinfo 1;", 0, 0, NULL},
    {"describe <all_table>", 0, 0, NULL},
    {"delete from <all_table> where ", 0, 0, NULL},
    {"drop database <db_name>", 0, 0, NULL},
    {"drop table <all_table>", 0, 0, NULL},
    {"drop dnode <dnode_id>", 0, 0, NULL},
    {"drop mnode on dnode <dnode_id> ;", 0, 0, NULL},
    {"drop qnode on dnode <dnode_id> ;", 0, 0, NULL},
    {"drop user <user_name> ;", 0, 0, NULL},
    // 40
    {"drop function", 0, 0, NULL},
    {"drop consumer group <anyword> on ", 0, 0, NULL},
    {"drop topic <topic_name> ;", 0, 0, NULL},
    {"drop stream <stream_name> ;", 0, 0, NULL},
    {"explain select", 0, 0, NULL},  // 44 append sub sql
112
    {"flush database <db_name> ;", 0, 0, NULL},
113
    {"help;", 0, 0, NULL},
114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154
    {"grant all on <anyword> to <user_name> ;", 0, 0, NULL},
    {"grant read on <anyword> to <user_name> ;", 0, 0, NULL},
    {"grant write on <anyword> to <user_name> ;", 0, 0, NULL},
    {"kill connection <anyword> ;", 0, 0, NULL},
    {"kill query ", 0, 0, NULL},
    {"kill transaction ", 0, 0, NULL},
    {"merge vgroup ", 0, 0, NULL},
    {"reset query cache;", 0, 0, NULL},
    {"revoke all on <anyword> from <user_name> ;", 0, 0, NULL},
    {"revoke read on <anyword> from <user_name> ;", 0, 0, NULL},
    {"revoke write on <anyword> from <user_name> ;", 0, 0, NULL},
    {"select * from <all_table>", 0, 0, NULL},
    {"select client_version();", 0, 0, NULL},
    // 60
    {"select current_user();", 0, 0, NULL},
    {"select database();", 0, 0, NULL},
    {"select server_version();", 0, 0, NULL},
    {"select server_status();", 0, 0, NULL},
    {"select now();", 0, 0, NULL},
    {"select today();", 0, 0, NULL},
    {"select timezone();", 0, 0, NULL},
    {"set max_binary_display_width ", 0, 0, NULL},
    {"show apps;", 0, 0, NULL},
    {"show create database <db_name> \\G;", 0, 0, NULL},
    {"show create stable <stb_name> \\G;", 0, 0, NULL},
    {"show create table <tb_name> \\G;", 0, 0, NULL},
    {"show connections;", 0, 0, NULL},
    {"show cluster;", 0, 0, NULL},
    {"show databases;", 0, 0, NULL},
    {"show dnodes;", 0, 0, NULL},
    {"show dnode <dnode_id> variables;", 0, 0, NULL},
    {"show functions;", 0, 0, NULL},
    {"show mnodes;", 0, 0, NULL},
    {"show queries;", 0, 0, NULL},
    // 80
    {"show query <anyword> ;", 0, 0, NULL},
    {"show qnodes;", 0, 0, NULL},
    {"show stables;", 0, 0, NULL},
    {"show stables like ", 0, 0, NULL},
    {"show streams;", 0, 0, NULL},
    {"show scores;", 0, 0, NULL},
A
Alex Duan 已提交
155
    {"show snodes;", 0, 0, NULL},
156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216
    {"show subscriptions;", 0, 0, NULL},
    {"show tables;", 0, 0, NULL},
    {"show tables like", 0, 0, NULL},
    {"show table distributed <all_table>", 0, 0, NULL},
    {"show tags from <tb_name>", 0, 0, NULL},
    {"show tags from <db_name>", 0, 0, NULL},
    {"show topics;", 0, 0, NULL},
    {"show transactions;", 0, 0, NULL},
    {"show users;", 0, 0, NULL},
    {"show variables;", 0, 0, NULL},
    {"show local variables;", 0, 0, NULL},
    {"show vnodes <dnode_id>", 0, 0, NULL},
    {"show vgroups;", 0, 0, NULL},
    {"show consumers;", 0, 0, NULL},
    {"show grants;", 0, 0, NULL},
    {"split vgroup ", 0, 0, NULL},
    {"insert into <tb_name> values(", 0, 0, NULL},
    {"insert into <tb_name> using <stb_name> tags(", 0, 0, NULL},
    {"insert into <tb_name> using <stb_name> <anyword> values(", 0, 0, NULL},
    {"insert into <tb_name> file ", 0, 0, NULL},
    {"trim database <db_name>", 0, 0, NULL},
    {"use <db_name>", 0, 0, NULL},
    {"quit", 0, 0, NULL}};

char* keywords[] = {
    "and ",         "asc ",      "desc ",     "from ",    "fill(",         "limit ",    "where ",
    "interval(",    "order by ", "order by ", "offset ",  "or ",           "group by ", "now()",
    "session(",     "sliding ",  "slimit ",   "soffset ", "state_window(", "today() ",  "union all select ",
    "partition by "};

char* functions[] = {
    "count(",         "sum(",
    "avg(",           "last(",
    "last_row(",      "top(",
    "interp(",        "max(",
    "min(",           "now()",
    "today()",        "percentile(",
    "tail(",          "pow(",
    "abs(",           "atan(",
    "acos(",          "asin(",
    "apercentile(",   "bottom(",
    "cast(",          "ceil(",
    "char_length(",   "cos(",
    "concat(",        "concat_ws(",
    "csum(",          "diff(",
    "derivative(",    "elapsed(",
    "first(",         "floor(",
    "hyperloglog(",   "histogram(",
    "irate(",         "leastsquares(",
    "length(",        "log(",
    "lower(",         "ltrim(",
    "mavg(",          "mode(",
    "tan(",           "round(",
    "rtrim(",         "sample(",
    "sin(",           "spread(",
    "substr(",        "statecount(",
    "stateduration(", "stddev(",
    "sqrt(",          "timediff(",
    "timezone(",      "timetruncate(",
    "twa(",           "to_unixtimestamp(",
    "unique(",        "upper(",
A
Alex Duan 已提交
217 218
};

219 220 221
char* tb_actions[] = {
    "add column ", "modify column ", "drop column ", "rename column ", "add tag ",
    "modify tag ", "drop tag ",      "rename tag ",  "set tag ",
A
Alex Duan 已提交
222 223
};

224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250
char* user_actions[] = {"pass ", "enable ", "sysinfo "};

char* tb_options[] = {"comment ", "watermark ", "max_delay ", "ttl ", "rollup(", "sma("};

char* db_options[] = {"keep ",
                      "replica ",
                      "precision ",
                      "strict ",
                      "buffer ",
                      "cachemodel ",
                      "cachesize ",
                      "comp ",
                      "duration ",
                      "wal_fsync_period",
                      "maxrows ",
                      "minrows ",
                      "pages ",
                      "pagesize ",
                      "retentions ",
                      "wal_level ",
                      "vgroups ",
                      "single_stable ",
                      "wal_retention_period ",
                      "wal_roll_period ",
                      "wal_retention_size ",
                      "wal_segment_size "};

251
char* alter_db_options[] = {"cachemodel ", "replica ", "keep ",  "cachesize ", "wal_fsync_period ", "wal_level "};
252 253 254 255 256 257 258 259 260 261 262 263 264

char* data_types[] = {"timestamp",    "int",
                      "int unsigned", "varchar(16)",
                      "float",        "double",
                      "binary(16)",   "nchar(16)",
                      "bigint",       "bigint unsigned",
                      "smallint",     "smallint unsigned",
                      "tinyint",      "tinyint unsigned",
                      "bool",         "json"};

char* key_tags[] = {"tags("};

char* key_select[] = {"select "};
A
Alex Duan 已提交
265

266
char* key_systable[] = {
A
Alex Duan 已提交
267 268 269 270 271 272
    "ins_dnodes",        "ins_mnodes",     "ins_modules",      "ins_qnodes",  "ins_snodes",          "ins_cluster",
    "ins_databases",     "ins_functions",  "ins_indexes",      "ins_stables", "ins_tables",          "ins_tags",
    "ins_users",         "ins_grants",     "ins_vgroups",      "ins_configs", "ins_dnode_variables", "ins_topics",
    "ins_subscriptions", "ins_streams",    "ins_stream_tasks", "ins_vnodes",  "ins_user_privileges", "perf_connections",
    "perf_queries",      "perf_consumers", "perf_trans",       "perf_apps"};

273

A
Alex Duan 已提交
274 275 276
//
//  ------- gobal variant define ---------
//
277 278 279 280 281
int32_t firstMatchIndex = -1;  // first match shellCommands index
int32_t lastMatchIndex = -1;   // last match shellCommands index
int32_t curMatchIndex = -1;    // current match shellCommands index
int32_t lastWordBytes = -1;    // printShow last word length
bool    waitAutoFill = false;
A
Alex Duan 已提交
282 283 284 285

//
//   ----------- global var array define -----------
//
286 287 288 289 290 291 292 293 294 295 296 297
#define WT_VAR_DBNAME         0
#define WT_VAR_STABLE         1
#define WT_VAR_TABLE          2
#define WT_VAR_DNODEID        3
#define WT_VAR_USERNAME       4
#define WT_VAR_TOPIC          5
#define WT_VAR_STREAM         6
#define WT_VAR_ALLTABLE       7
#define WT_VAR_FUNC           8
#define WT_VAR_KEYWORD        9
#define WT_VAR_TBACTION       10
#define WT_VAR_DBOPTION       11
A
Alex Duan 已提交
298
#define WT_VAR_ALTER_DBOPTION 12
299 300 301 302 303 304
#define WT_VAR_DATATYPE       13
#define WT_VAR_KEYTAGS        14
#define WT_VAR_ANYWORD        15
#define WT_VAR_TBOPTION       16
#define WT_VAR_USERACTION     17
#define WT_VAR_KEYSELECT      18
305
#define WT_VAR_SYSTABLE       19
A
Alex Duan 已提交
306

307
#define WT_VAR_CNT 20
A
Alex Duan 已提交
308

309 310
#define WT_FROM_DB_MAX 6  // max get content from db
#define WT_FROM_DB_CNT (WT_FROM_DB_MAX + 1)
A
Alex Duan 已提交
311

312
#define WT_TEXT 0xFF
A
Alex Duan 已提交
313

314
char dbName[256] = "";  // save use database name;
A
Alex Duan 已提交
315
// tire array
316
STire*        tires[WT_VAR_CNT];
A
Alex Duan 已提交
317
TdThreadMutex tiresMutex;
318
// save thread handle obtain var name from db server
A
Alex Duan 已提交
319 320
TdThread* threads[WT_FROM_DB_CNT];
// obtain var name  with sql from server
321 322 323 324
char varTypes[WT_VAR_CNT][64] = {"<db_name>",    "<stb_name>",    "<tb_name>",          "<dnode_id>",   "<user_name>",
                                 "<topic_name>", "<stream_name>", "<all_table>",        "<function>",   "<keyword>",
                                 "<tb_actions>", "<db_options>",  "<alter_db_options>", "<data_types>", "<key_tags>",
                                 "<anyword>",    "<tb_options>",  "<user_actions>",     "<key_select>"};
A
Alex Duan 已提交
325

326 327
char varSqls[WT_FROM_DB_CNT][64] = {"show databases;", "show stables;", "show tables;", "show dnodes;",
                                    "show users;",     "show topics;",  "show streams;"};
A
Alex Duan 已提交
328 329

// var words current cursor, if user press any one key except tab, cursorVar can be reset to -1
330 331
int  cursorVar = -1;
bool varMode = false;  // enter var names list mode
A
Alex Duan 已提交
332

333 334
TAOS*      varCon = NULL;
SShellCmd* varCmd = NULL;
335
bool       varRunOnce = false;
336 337
SMatch*    lastMatch = NULL;  // save last match result
int        cntDel = 0;        // delete byte count after next press tab
A
Alex Duan 已提交
338 339 340

// show auto tab introduction
void printfIntroduction() {
341 342 343 344 345 346 347 348 349 350 351 352 353
  printf("  ******************************  Tab Completion  *************************************\n");
  printf("  *   The TDengine CLI supports tab completion for a variety of items,                *\n");
  printf("  *   including database names, table names, function names and keywords.             *\n");
  printf("  *   The full list of shortcut keys is as follows:                                   *\n");
  printf("  *    [ TAB ]        ......  complete the current word                               *\n");
  printf("  *                   ......  if used on a blank line, display all supported commands *\n");
  printf("  *    [ Ctrl + A ]   ......  move cursor to the st[A]rt of the line                  *\n");
  printf("  *    [ Ctrl + E ]   ......  move cursor to the [E]nd of the line                    *\n");
  printf("  *    [ Ctrl + W ]   ......  move cursor to the middle of the line                   *\n");
  printf("  *    [ Ctrl + L ]   ......  clear the entire screen                                 *\n");
  printf("  *    [ Ctrl + K ]   ......  clear the screen after the cursor                       *\n");
  printf("  *    [ Ctrl + U ]   ......  clear the screen before the cursor                      *\n");
  printf("  *************************************************************************************\n\n");
A
Alex Duan 已提交
354 355 356
}

void showHelp() {
A
Alex Duan 已提交
357
  printf("\nThe TDengine CLI supports the following commands:");
358 359
  printf(
      "\n\
A
Alex Duan 已提交
360 361
  ----- A ----- \n\
    alter database <db_name> <db_options> \n\
A
Alex Duan 已提交
362 363 364
    alter dnode <dnode_id> 'resetlog';\n\
    alter dnode <dnode_id> 'monitor' '0';\n\
    alter dnode <dnode_id> 'monitor' \"1\";\n\
365 366 367 368 369
    alter dnode <dnode_id> \"debugflag\" \"143\";\n\
    alter all dnodes \"monitor\" \"0\";\n\
    alter all dnodes \"monitor\" \"1\";\n\
    alter all dnodes \"resetlog\";\n\
    alter all dnodes \"debugFlag\" \n\
A
Alex Duan 已提交
370 371
    alter table <tb_name> <tb_actions> ;\n\
    alter table modify column\n\
372 373
    alter local \"resetlog\";\n\
    alter local \"DebugFlag\" \"143\";\n\
A
Alex Duan 已提交
374 375
    alter topic\n\
    alter user <user_name> <user_actions> ...\n\
A
Alex Duan 已提交
376 377
  ----- C ----- \n\
    create table <tb_name> using <stb_name> tags ...\n\
A
Alex Duan 已提交
378
    create database <db_name> <db_options>  ...\n\
379
    create dnode \"fqdn:port\" ...\n\
A
Alex Duan 已提交
380 381 382 383 384 385 386
    create index ...\n\
    create mnode on dnode <dnode_id> ;\n\
    create qnode on dnode <dnode_id> ;\n\
    create stream <stream_name> into <stb_name> as select ...\n\
    create topic <topic_name> as select ...\n\
    create function ...\n\
    create user <user_name> pass <password> ...\n\
A
Alex Duan 已提交
387
  ----- D ----- \n\
A
Alex Duan 已提交
388 389
    describe <all_table>\n\
    delete from <all_table> where ...\n\
A
Alex Duan 已提交
390 391 392
    drop database <db_name>;\n\
    drop table <all_table>;\n\
    drop dnode <dnode_id>;\n\
A
Alex Duan 已提交
393 394 395 396 397 398 399 400 401
    drop mnode on dnode <dnode_id> ;\n\
    drop qnode on dnode <dnode_id> ;\n\
    drop user <user_name> ;\n\
    drop function <function_name>;\n\
    drop consumer group ... \n\
    drop topic <topic_name> ;\n\
    drop stream <stream_name> ;\n\
  ----- E ----- \n\
    explain select clause ...\n\
402
  ----- F ----- \n\
A
Alex Duan 已提交
403
    flush database <db_name>;\n\
404 405
  ----- H ----- \n\
    help;\n\
A
Alex Duan 已提交
406 407 408 409 410 411 412
  ----- I ----- \n\
    insert into <tb_name> values(...) ;\n\
    insert into <tb_name> using <stb_name> tags(...) values(...) ;\n\
  ----- G ----- \n\
    grant all   on <priv_level> to <user_name> ;\n\
    grant read  on <priv_level> to <user_name> ;\n\
    grant write on <priv_level> to <user_name> ;\n\
A
Alex Duan 已提交
413 414 415
  ----- K ----- \n\
    kill connection <connection_id>; \n\
    kill query <query_id>; \n\
A
Alex Duan 已提交
416 417 418 419 420 421 422 423
    kill transaction <transaction_id>;\n\
  ----- M ----- \n\
    merge vgroup ...\n\
  ----- R ----- \n\
    reset query cache;\n\
    revoke all   on <priv_level> from <user_name> ;\n\
    revoke read  on <priv_level> from <user_name> ;\n\
    revoke write on <priv_level> from <user_name> ;\n\
A
Alex Duan 已提交
424 425 426 427
  ----- S ----- \n\
    select * from <all_table> where ... \n\
    select client_version();\n\
    select current_user();\n\
A
Alex Duan 已提交
428
    select database();\n\
A
Alex Duan 已提交
429
    select server_version();\n\
A
Alex Duan 已提交
430 431 432 433 434 435
    select server_status();\n\
    select now();\n\
    select today();\n\
    select timezone();\n\
    set max_binary_display_width ...\n\
    show apps;\n\
A
Alex Duan 已提交
436 437 438 439
    show create database <db_name>;\n\
    show create stable <stb_name>;\n\
    show create table <tb_name>;\n\
    show connections;\n\
A
Alex Duan 已提交
440
    show cluster;\n\
A
Alex Duan 已提交
441 442
    show databases;\n\
    show dnodes;\n\
A
Alex Duan 已提交
443
    show dnode <dnode_id> variables;\n\
A
Alex Duan 已提交
444 445 446
    show functions;\n\
    show mnodes;\n\
    show queries;\n\
A
Alex Duan 已提交
447 448 449
    show query <query_id> ;\n\
    show qnodes;\n\
    show snodes;\n\
A
Alex Duan 已提交
450
    show stables;\n\
A
Alex Duan 已提交
451
    show stables like \n\
A
Alex Duan 已提交
452 453
    show streams;\n\
    show scores;\n\
A
Alex Duan 已提交
454
    show subscriptions;\n\
A
Alex Duan 已提交
455
    show tables;\n\
A
Alex Duan 已提交
456 457 458 459 460 461
    show tables like\n\
    show table distributed <all_table>;\n\
    show tags from <tb_name>\n\
    show tags from <db_name>\n\
    show topics;\n\
    show transactions;\n\
A
Alex Duan 已提交
462 463
    show users;\n\
    show variables;\n\
A
Alex Duan 已提交
464 465
    show local variables;\n\
    show vnodes <dnode_id>\n\
A
Alex Duan 已提交
466
    show vgroups;\n\
A
Alex Duan 已提交
467 468 469 470 471
    show consumers;\n\
    show grants;\n\
    split vgroup ...\n\
  ----- T ----- \n\
    trim database <db_name>;\n\
A
Alex Duan 已提交
472 473 474 475
  ----- U ----- \n\
    use <db_name>;");

  printf("\n\n");
476 477 478 479

  // define in getDuration() function
  printf(
      "\
A
Alex Duan 已提交
480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501
  Timestamp expression Format:\n\
    b - nanosecond \n\
    u - microsecond \n\
    a - millisecond \n\
    s - second \n\
    m - minute \n\
    h - hour \n\
    d - day \n\
    w - week \n\
    now - current time \n\
  Example : \n\
    select * from t1 where ts > now - 2w + 3d and ts <= now - 1w -2h ;\n");
  printf("\n");
}

//
//  -------------------  parse words --------------------------
//

#define SHELL_COMMAND_COUNT() (sizeof(shellCommands) / sizeof(SWords))

// get at
502 503
SWord* atWord(SWords* command, int32_t index) {
  SWord* word = command->head;
A
Alex Duan 已提交
504
  for (int32_t i = 0; i < index; i++) {
505
    if (word == NULL) return NULL;
A
Alex Duan 已提交
506 507 508 509 510 511 512 513 514 515
    word = word->next;
  }

  return word;
}

#define MATCH_WORD(x) atWord(x, x->matchIndex)

int wordType(const char* p, int32_t len) {
  for (int i = 0; i < WT_VAR_CNT; i++) {
516
    if (strncmp(p, varTypes[i], len) == 0) return i;
A
Alex Duan 已提交
517 518 519 520 521
  }
  return WT_TEXT;
}

// add word
522 523
SWord* addWord(const char* p, int32_t len, bool pattern) {
  SWord* word = (SWord*)taosMemoryMalloc(sizeof(SWord));
A
Alex Duan 已提交
524
  memset(word, 0, sizeof(SWord));
525 526
  word->word = (char*)p;
  word->len = len;
A
Alex Duan 已提交
527 528 529 530 531 532 533 534 535 536 537 538

  // check format
  if (pattern) {
    word->type = wordType(p, len);
  } else {
    word->type = WT_TEXT;
  }

  return word;
}

// parse one command
539 540
void parseCommand(SWords* command, bool pattern) {
  char*   p = command->source;
A
Alex Duan 已提交
541
  int32_t start = 0;
542
  int32_t size = command->source_len > 0 ? command->source_len : strlen(p);
A
Alex Duan 已提交
543 544 545 546 547 548 549

  bool lastBlank = false;
  for (int i = 0; i <= size; i++) {
    if (p[i] == ' ' || i == size) {
      // check continue blank like '    '
      if (p[i] == ' ') {
        if (lastBlank) {
550
          start++;
A
Alex Duan 已提交
551 552
          continue;
        }
553
        if (i == 0) {  // first blank
A
Alex Duan 已提交
554
          lastBlank = true;
555
          start++;
A
Alex Duan 已提交
556 557 558
          continue;
        }
        lastBlank = true;
559
      }
A
Alex Duan 已提交
560 561 562 563 564 565

      // found split or string end , append word
      if (command->head == NULL) {
        command->head = addWord(p + start, i - start, pattern);
        command->count = 1;
      } else {
566
        SWord* word = command->head;
A
Alex Duan 已提交
567 568 569 570
        while (word->next) {
          word = word->next;
        }
        word->next = addWord(p + start, i - start, pattern);
571
        command->count++;
A
Alex Duan 已提交
572 573 574 575 576 577 578 579 580
      }
      start = i + 1;
    } else {
      lastBlank = false;
    }
  }
}

// free SShellCmd
581
void freeCommand(SWords* command) {
A
Alex Duan 已提交
582
  SWord* item = command->head;
583
  command->head = NULL;
584
  // loop
A
Alex Duan 已提交
585 586 587
  while (item) {
    SWord* tmp = item;
    item = item->next;
A
Alex Duan 已提交
588
    // if malloc need free
589
    if (tmp->free && tmp->word) taosMemoryFree(tmp->word);
A
Alex Duan 已提交
590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608
    taosMemoryFree(tmp);
  }
}

void GenerateVarType(int type, char** p, int count) {
  STire* tire = createTire(TIRE_LIST);
  for (int i = 0; i < count; i++) {
    insertWord(tire, p[i]);
  }

  taosThreadMutexLock(&tiresMutex);
  tires[type] = tire;
  taosThreadMutexUnlock(&tiresMutex);
}

//
//  -------------------- shell auto ----------------
//

609
// init shell auto funciton , shell start call once
A
Alex Duan 已提交
610 611 612
bool shellAutoInit() {
  // command
  int32_t count = SHELL_COMMAND_COUNT();
613
  for (int32_t i = 0; i < count; i++) {
A
Alex Duan 已提交
614 615 616 617 618 619 620 621 622 623 624
    parseCommand(shellCommands + i, true);
  }

  // tires
  memset(tires, 0, sizeof(STire*) * WT_VAR_CNT);
  taosThreadMutexInit(&tiresMutex, NULL);

  // threads
  memset(threads, 0, sizeof(TdThread*) * WT_FROM_DB_CNT);

  // generate varType
625 626 627 628 629 630 631 632 633 634
  GenerateVarType(WT_VAR_FUNC, functions, sizeof(functions) / sizeof(char*));
  GenerateVarType(WT_VAR_KEYWORD, keywords, sizeof(keywords) / sizeof(char*));
  GenerateVarType(WT_VAR_DBOPTION, db_options, sizeof(db_options) / sizeof(char*));
  GenerateVarType(WT_VAR_ALTER_DBOPTION, alter_db_options, sizeof(alter_db_options) / sizeof(char*));
  GenerateVarType(WT_VAR_TBACTION, tb_actions, sizeof(tb_actions) / sizeof(char*));
  GenerateVarType(WT_VAR_DATATYPE, data_types, sizeof(data_types) / sizeof(char*));
  GenerateVarType(WT_VAR_KEYTAGS, key_tags, sizeof(key_tags) / sizeof(char*));
  GenerateVarType(WT_VAR_TBOPTION, tb_options, sizeof(tb_options) / sizeof(char*));
  GenerateVarType(WT_VAR_USERACTION, user_actions, sizeof(user_actions) / sizeof(char*));
  GenerateVarType(WT_VAR_KEYSELECT, key_select, sizeof(key_select) / sizeof(char*));
635
  GenerateVarType(WT_VAR_SYSTABLE, key_systable, sizeof(key_systable) / sizeof(char*));
A
Alex Duan 已提交
636 637 638 639 640

  return true;
}

// set conn
641 642 643
void shellSetConn(TAOS* conn, bool runOnce) {
  varCon = conn;
  varRunOnce = runOnce;
A
Alex Duan 已提交
644
  // init database and stable
645
  if (!runOnce) updateTireValue(WT_VAR_DBNAME, false);
A
Alex Duan 已提交
646
}
A
Alex Duan 已提交
647 648 649 650 651

// exit shell auto funciton, shell exit call once
void shellAutoExit() {
  // free command
  int32_t count = SHELL_COMMAND_COUNT();
652
  for (int32_t i = 0; i < count; i++) {
A
Alex Duan 已提交
653 654 655 656 657 658 659 660 661
    freeCommand(shellCommands + i);
  }

  // free tires
  taosThreadMutexLock(&tiresMutex);
  for (int32_t i = 0; i < WT_VAR_CNT; i++) {
    if (tires[i]) {
      freeTire(tires[i]);
      tires[i] = NULL;
662
    }
A
Alex Duan 已提交
663 664 665 666 667 668
  }
  taosThreadMutexUnlock(&tiresMutex);
  // destory
  taosThreadMutexDestroy(&tiresMutex);

  // free threads
A
Alex Duan 已提交
669
  for (int32_t i = 0; i < WT_FROM_DB_CNT; i++) {
A
Alex Duan 已提交
670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686
    if (threads[i]) {
      taosDestroyThread(threads[i]);
      threads[i] = NULL;
    }
  }

  // free lastMatch
  if (lastMatch) {
    freeMatch(lastMatch);
    lastMatch = NULL;
  }
}

//
//  -------------------  auto ptr for tires --------------------------
//
bool setNewAuotPtr(int type, STire* pNew) {
687
  if (pNew == NULL) return false;
A
Alex Duan 已提交
688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721

  taosThreadMutexLock(&tiresMutex);
  STire* pOld = tires[type];
  if (pOld != NULL) {
    // previous have value, release self ref count
    if (--pOld->ref == 0) {
      freeTire(pOld);
    }
  }

  // set new
  tires[type] = pNew;
  tires[type]->ref = 1;
  taosThreadMutexUnlock(&tiresMutex);

  return true;
}

// get ptr
STire* getAutoPtr(int type) {
  if (tires[type] == NULL) {
    return NULL;
  }

  taosThreadMutexLock(&tiresMutex);
  tires[type]->ref++;
  taosThreadMutexUnlock(&tiresMutex);

  return tires[type];
}

// put back tire to tires[type], if tire not equal tires[type].p, need free tire
void putBackAutoPtr(int type, STire* tire) {
  if (tire == NULL) {
722
    return;
A
Alex Duan 已提交
723 724 725 726
  }

  taosThreadMutexLock(&tiresMutex);
  if (tires[type] != tire) {
727
    // update by out,  can't put back , so free
A
Alex Duan 已提交
728 729 730 731
    if (--tire->ref == 1) {
      // support multi thread getAuotPtr
      freeTire(tire);
    }
732

A
Alex Duan 已提交
733 734 735 736 737 738
  } else {
    tires[type]->ref--;
    assert(tires[type]->ref > 0);
  }
  taosThreadMutexUnlock(&tiresMutex);

739
  return;
A
Alex Duan 已提交
740 741 742 743 744 745
}

//
//  -------------------  var Word --------------------------
//

746
#define MAX_CACHED_CNT 100000  // max cached rows 10w
A
Alex Duan 已提交
747 748
// write sql result to var name, return write rows cnt
int writeVarNames(int type, TAOS_RES* tres) {
749
  // fetch row
A
Alex Duan 已提交
750 751 752 753 754
  TAOS_ROW row = taos_fetch_row(tres);
  if (row == NULL) {
    return 0;
  }

755
  TAOS_FIELD* fields = taos_fetch_fields(tres);
A
Alex Duan 已提交
756
  // create new tires
757
  char   tireType = type == WT_VAR_TABLE ? TIRE_TREE : TIRE_LIST;
A
Alex Duan 已提交
758 759 760 761
  STire* tire = createTire(tireType);

  // enum rows
  char name[1024];
762
  int  numOfRows = 0;
A
Alex Duan 已提交
763 764
  do {
    int32_t* lengths = taos_fetch_lengths(tres);
765 766 767
    int32_t  bytes = lengths[0];
    if (fields[0].type == TSDB_DATA_TYPE_INT) {
      sprintf(name, "%d", *(int16_t*)row[0]);
A
Alex Duan 已提交
768 769 770
    } else {
      memcpy(name, row[0], bytes);
    }
771 772

    name[bytes] = 0;  // set string end
A
Alex Duan 已提交
773 774 775
    // insert to tire
    insertWord(tire, name);

776
    if (++numOfRows > MAX_CACHED_CNT) {
A
Alex Duan 已提交
777 778 779 780 781 782 783 784 785 786 787 788
      break;
    }

    row = taos_fetch_row(tres);
  } while (row != NULL);

  // replace old tire
  setNewAuotPtr(type, tire);

  return numOfRows;
}

789 790 791 792 793 794
void setThreadNull(int type) {
  taosThreadMutexLock(&tiresMutex);
  threads[type] = NULL;
  taosThreadMutexUnlock(&tiresMutex);
}

795
bool firstMatchCommand(TAOS* con, SShellCmd* cmd);
A
Alex Duan 已提交
796
//
797
//  thread obtain var thread from db server
A
Alex Duan 已提交
798 799
//
void* varObtainThread(void* param) {
800
  int type = *(int*)param;
A
Alex Duan 已提交
801 802 803 804 805 806 807 808 809
  taosMemoryFree(param);

  if (varCon == NULL || type > WT_FROM_DB_MAX) {
    return NULL;
  }

  TAOS_RES* pSql = taos_query(varCon, varSqls[type]);
  if (taos_errno(pSql)) {
    taos_free_result(pSql);
810
    setThreadNull(type);
A
Alex Duan 已提交
811 812 813 814 815 816 817 818 819
    return NULL;
  }

  // write var names from pSql
  int cnt = writeVarNames(type, pSql);

  // free sql
  taos_free_result(pSql);

820
  // check need call auto tab
A
Alex Duan 已提交
821 822 823 824 825
  if (cnt > 0 && waitAutoFill) {
    // press tab key by program
    firstMatchCommand(varCon, varCmd);
  }

826
  setThreadNull(type);
A
Alex Duan 已提交
827 828 829
  return NULL;
}

A
Alex Duan 已提交
830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861
// return true is need update value by async
bool updateTireValue(int type, bool autoFill) {
  // TYPE CONTEXT GET FROM DB
  taosThreadMutexLock(&tiresMutex);

  // check need obtain from server
  if (tires[type] == NULL) {
    waitAutoFill = autoFill;
    // need async obtain var names from db sever
    if (threads[type] != NULL) {
      if (taosThreadRunning(threads[type])) {
        // thread running , need not obtain again, return
        taosThreadMutexUnlock(&tiresMutex);
        return NULL;
      }
      // destroy previous thread handle for new create thread handle
      taosDestroyThread(threads[type]);
      threads[type] = NULL;
    }

    // create new
    void* param = taosMemoryMalloc(sizeof(int));
    *((int*)param) = type;
    threads[type] = taosCreateThread(varObtainThread, param);
    taosThreadMutexUnlock(&tiresMutex);
    return true;
  }
  taosThreadMutexUnlock(&tiresMutex);

  return false;
}

A
Alex Duan 已提交
862 863 864
// only match next one word from all match words, return valuue must free by caller
char* matchNextPrefix(STire* tire, char* pre) {
  SMatch* match = NULL;
A
Alex Duan 已提交
865
  if(tire == NULL) return NULL;
A
Alex Duan 已提交
866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881

  // re-use last result
  if (lastMatch) {
    if (strcmp(pre, lastMatch->pre) == 0) {
      // same pre
      match = lastMatch;
    }
  }

  if (match == NULL) {
    // not same with last result
    if (pre[0] == 0) {
      // EMPTY PRE
      match = enumAll(tire);
    } else {
      // NOT EMPTY
882 883 884
      match = (SMatch*)taosMemoryMalloc(sizeof(SMatch));
      memset(match, 0, sizeof(SMatch));
      matchPrefix(tire, pre, match);
A
Alex Duan 已提交
885
    }
886

A
Alex Duan 已提交
887 888
    // save to lastMatch
    if (match) {
889
      if (lastMatch) freeMatch(lastMatch);
A
Alex Duan 已提交
890 891 892 893 894 895 896
      lastMatch = match;
    }
  }

  // check valid
  if (match == NULL || match->head == NULL) {
    // no one matched
897
    return NULL;
A
Alex Duan 已提交
898 899 900 901 902
  }

  if (cursorVar == -1) {
    // first
    cursorVar = 0;
903
    return strdup(match->head->word);
A
Alex Duan 已提交
904 905 906
  }

  // according to cursorVar , calculate next one
907
  int         i = 0;
A
Alex Duan 已提交
908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943
  SMatchNode* item = match->head;
  while (item) {
    if (i == cursorVar + 1) {
      // found next position ok
      if (item->next == NULL) {
        // match last item, reset cursorVar to head
        cursorVar = -1;
      } else {
        cursorVar = i;
      }

      return strdup(item->word);
    }

    // check end item
    if (item->next == NULL) {
      // if cursorVar > var list count, return last and reset cursorVar
      cursorVar = -1;

      return strdup(item->word);
    }

    // move next
    item = item->next;
    i++;
  }

  return NULL;
}

// search pre word from tire tree, return value must free by caller
char* tireSearchWord(int type, char* pre) {
  if (type == WT_TEXT) {
    return NULL;
  }

944
  if (type > WT_FROM_DB_MAX) {
A
Alex Duan 已提交
945 946
    // NOT FROM DB , tires[type] alwary not null
    STire* tire = tires[type];
947 948
    if (tire == NULL) return NULL;
    return matchNextPrefix(tire, pre);
A
Alex Duan 已提交
949 950
  }

951
  if(updateTireValue(type, true)) {
A
Alex Duan 已提交
952 953 954
    return NULL;
  }

955 956 957
  // can obtain var names from local
  STire* tire = getAutoPtr(type);
  if (tire == NULL) {
A
Alex Duan 已提交
958 959 960
    return NULL;
  }

A
Alex Duan 已提交
961 962 963 964 965 966 967
  char* str = matchNextPrefix(tire, pre);
  // used finish, put back pointer to autoptr array
  putBackAutoPtr(type, tire);

  return str;
}

968
// match var word, word1 is pattern , word2 is input from shell
A
Alex Duan 已提交
969
bool matchVarWord(SWord* word1, SWord* word2) {
970
  // search input word from tire tree
A
Alex Duan 已提交
971 972 973 974 975 976 977 978 979 980
  char pre[512];
  memcpy(pre, word2->word, word2->len);
  pre[word2->len] = 0;

  char* str = NULL;
  if (word1->type == WT_VAR_ALLTABLE) {
    // ALL_TABLE
    str = tireSearchWord(WT_VAR_STABLE, pre);
    if (str == NULL) {
      str = tireSearchWord(WT_VAR_TABLE, pre);
981
      if (str == NULL) return false;
A
Alex Duan 已提交
982 983 984 985 986 987 988 989 990 991 992
    }
  } else {
    // OTHER
    str = tireSearchWord(word1->type, pre);
    if (str == NULL) {
      // not found or word1->type variable list not obtain from server, return not match
      return false;
    }
  }

  // free previous malloc
993
  if (word1->free && word1->word) {
A
Alex Duan 已提交
994 995 996 997 998
    taosMemoryFree(word1->word);
  }

  // save
  word1->word = str;
999 1000 1001
  word1->len = strlen(str);
  word1->free = true;  // need free

A
Alex Duan 已提交
1002 1003 1004 1005 1006 1007 1008
  return true;
}

//
//  -------------------  match words --------------------------
//

1009 1010 1011 1012
// compare command cmdPattern come from shellCommands , cmdInput come from user input
int32_t compareCommand(SWords* cmdPattern, SWords* cmdInput) {
  SWord* wordPattern = cmdPattern->head;
  SWord* wordInput = cmdInput->head;
A
Alex Duan 已提交
1013

1014
  if (wordPattern == NULL || wordInput == NULL) {
A
Alex Duan 已提交
1015 1016 1017
    return -1;
  }

1018 1019
  for (int32_t i = 0; i < cmdPattern->count; i++) {
    if (wordPattern->type == WT_TEXT) {
A
Alex Duan 已提交
1020
      // WT_TEXT match
1021 1022 1023
      if (wordPattern->len == wordInput->len) {
        if (strncasecmp(wordPattern->word, wordInput->word, wordPattern->len) != 0) return -1;
      } else if (wordPattern->len < wordInput->len) {
A
Alex Duan 已提交
1024 1025
        return -1;
      } else {
1026 1027 1028 1029 1030 1031 1032 1033 1034 1035
        // wordPattern->len > wordInput->len
        if (strncasecmp(wordPattern->word, wordInput->word, wordInput->len) == 0) {
          if (i + 1 == cmdInput->count) {
            // last word return match
            cmdPattern->matchIndex = i;
            cmdPattern->matchLen = wordInput->len;
            return i;
          } else {
            return -1;
          }
A
Alex Duan 已提交
1036 1037 1038 1039 1040 1041
        } else {
          return -1;
        }
      }
    } else {
      // WT_VAR auto match any one word
1042 1043 1044 1045
      if (wordInput->next == NULL) {  // input words last one
        if (matchVarWord(wordPattern, wordInput)) {
          cmdPattern->matchIndex = i;
          cmdPattern->matchLen = wordInput->len;
A
Alex Duan 已提交
1046 1047 1048 1049 1050 1051 1052 1053
          varMode = true;
          return i;
        }
        return -1;
      }
    }

    // move next
1054 1055 1056
    wordPattern = wordPattern->next;
    wordInput = wordInput->next;
    if (wordPattern == NULL || wordInput == NULL) {
A
Alex Duan 已提交
1057 1058 1059 1060 1061 1062 1063 1064
      return -1;
    }
  }

  return -1;
}

// match command
1065
SWords* matchCommand(SWords* input, bool continueSearch) {
A
Alex Duan 已提交
1066
  int32_t count = SHELL_COMMAND_COUNT();
1067 1068
  for (int32_t i = 0; i < count; i++) {
    SWords* shellCommand = shellCommands + i;
A
Alex Duan 已提交
1069 1070 1071 1072 1073 1074 1075 1076 1077 1078
    if (continueSearch && lastMatchIndex != -1 && i <= lastMatchIndex) {
      // new match must greate than lastMatchIndex
      if (varMode && i == lastMatchIndex) {
        // do nothing, var match on lastMatchIndex
      } else {
        continue;
      }
    }

    // command is large
1079
    if (input->count > shellCommand->count) {
A
Alex Duan 已提交
1080 1081 1082 1083 1084 1085
      continue;
    }

    // compare
    int32_t index = compareCommand(shellCommand, input);
    if (index != -1) {
1086
      if (firstMatchIndex == -1) firstMatchIndex = i;
A
Alex Duan 已提交
1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100
      curMatchIndex = i;
      return &shellCommands[i];
    }
  }

  // not match
  return NULL;
}

//
//  -------------------  print screen --------------------------
//

// delete char count
1101
void deleteCount(SShellCmd* cmd, int count) {
A
Alex Duan 已提交
1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119
  int size = 0;
  int width = 0;
  int prompt_size = 6;
  shellClearScreen(cmd->endOffset + prompt_size, cmd->screenOffset + prompt_size);

  // loop delete
  while (--count >= 0 && cmd->cursorOffset > 0) {
    shellGetPrevCharSize(cmd->command, cmd->cursorOffset, &size, &width);
    memmove(cmd->command + cmd->cursorOffset - size, cmd->command + cmd->cursorOffset,
            cmd->commandSize - cmd->cursorOffset);
    cmd->commandSize -= size;
    cmd->cursorOffset -= size;
    cmd->screenOffset -= width;
    cmd->endOffset -= width;
  }
}

// show screen
1120
void printScreen(TAOS* con, SShellCmd* cmd, SWords* match) {
A
Alex Duan 已提交
1121 1122 1123
  // modify SShellCmd
  if (firstMatchIndex == -1 || curMatchIndex == -1) {
    // no match
1124
    return;
A
Alex Duan 已提交
1125 1126
  }

1127 1128 1129
  // first tab press
  const char* str = NULL;
  int         strLen = 0;
A
Alex Duan 已提交
1130 1131 1132

  if (firstMatchIndex == curMatchIndex && lastWordBytes == -1) {
    // first press tab
1133
    SWord* word = MATCH_WORD(match);
A
Alex Duan 已提交
1134 1135 1136 1137 1138
    str = word->word + match->matchLen;
    strLen = word->len - match->matchLen;
    lastMatchIndex = firstMatchIndex;
    lastWordBytes = word->len;
  } else {
1139
    if (lastWordBytes == -1) return;
A
Alex Duan 已提交
1140 1141
    deleteCount(cmd, lastWordBytes);

1142
    SWord* word = MATCH_WORD(match);
A
Alex Duan 已提交
1143 1144 1145 1146 1147 1148
    str = word->word;
    strLen = word->len;
    // set current to last
    lastMatchIndex = curMatchIndex;
    lastWordBytes = word->len;
  }
1149

A
Alex Duan 已提交
1150
  // insert new
A
Alex Duan 已提交
1151
  shellInsertStr(cmd, (char*)str, strLen);
A
Alex Duan 已提交
1152 1153 1154
}

// main key press tab , matched return true else false
1155
bool firstMatchCommand(TAOS* con, SShellCmd* cmd) {
A
Alex Duan 已提交
1156
  if(con == NULL || cmd == NULL) return false;
A
Alex Duan 已提交
1157
  // parse command
1158
  SWords* input = (SWords*)taosMemoryMalloc(sizeof(SWords));
A
Alex Duan 已提交
1159 1160 1161 1162 1163 1164
  memset(input, 0, sizeof(SWords));
  input->source = cmd->command;
  input->source_len = cmd->commandSize;
  parseCommand(input, false);

  // if have many , default match first, if press tab again , switch to next
1165
  curMatchIndex = -1;
A
Alex Duan 已提交
1166
  lastMatchIndex = -1;
1167
  SWords* match = matchCommand(input, true);
A
Alex Duan 已提交
1168 1169 1170 1171 1172 1173 1174 1175 1176
  if (match == NULL) {
    // not match , nothing to do
    freeCommand(input);
    taosMemoryFree(input);
    return false;
  }

  // print to screen
  printScreen(con, cmd, match);
wafwerar's avatar
wafwerar 已提交
1177 1178 1179 1180
#ifdef WINDOWS
  printf("\r");
  shellShowOnScreen(cmd);
#endif
A
Alex Duan 已提交
1181 1182 1183 1184 1185 1186
  freeCommand(input);
  taosMemoryFree(input);
  return true;
}

// create input source
1187
void createInputFromFirst(SWords* input, SWords* firstMatch) {
A
Alex Duan 已提交
1188 1189 1190 1191
  //
  // if next pressTabKey , input context come from firstMatch, set matched length with source_len
  //
  input->source = (char*)taosMemoryMalloc(1024);
1192
  memset((void*)input->source, 0, 1024);
A
Alex Duan 已提交
1193

1194
  SWord* word = firstMatch->head;
A
Alex Duan 已提交
1195

1196
  // source_len = full match word->len + half match with firstMatch->matchLen
A
Alex Duan 已提交
1197 1198 1199
  for (int i = 0; i < firstMatch->matchIndex && word; i++) {
    // combine source from each word
    strncpy(input->source + input->source_len, word->word, word->len);
1200 1201
    strcat(input->source, " ");          // append blank splite
    input->source_len += word->len + 1;  // 1 is blank length
A
Alex Duan 已提交
1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212
    // move next
    word = word->next;
  }
  // appand half matched word for last
  if (word) {
    strncpy(input->source + input->source_len, word->word, firstMatch->matchLen);
    input->source_len += firstMatch->matchLen;
  }
}

// user press Tabkey again is named next , matched return true else false
1213
bool nextMatchCommand(TAOS* con, SShellCmd* cmd, SWords* firstMatch) {
A
Alex Duan 已提交
1214 1215 1216
  if (firstMatch == NULL || firstMatch->head == NULL) {
    return false;
  }
1217
  SWords* input = (SWords*)taosMemoryMalloc(sizeof(SWords));
A
Alex Duan 已提交
1218 1219 1220 1221 1222 1223 1224 1225 1226
  memset(input, 0, sizeof(SWords));

  // create input from firstMatch
  createInputFromFirst(input, firstMatch);

  // parse input
  parseCommand(input, false);

  // if have many , default match first, if press tab again , switch to next
1227
  SWords* match = matchCommand(input, true);
A
Alex Duan 已提交
1228 1229 1230
  if (match == NULL) {
    // if not match , reset all index
    firstMatchIndex = -1;
1231
    curMatchIndex = -1;
A
Alex Duan 已提交
1232 1233 1234
    match = matchCommand(input, false);
    if (match == NULL) {
      freeCommand(input);
1235
      if (input->source) taosMemoryFree(input->source);
A
Alex Duan 已提交
1236 1237 1238 1239 1240 1241 1242
      taosMemoryFree(input);
      return false;
    }
  }

  // print to screen
  printScreen(con, cmd, match);
wafwerar's avatar
wafwerar 已提交
1243 1244 1245 1246
#ifdef WINDOWS
  printf("\r");
  shellShowOnScreen(cmd);
#endif
A
Alex Duan 已提交
1247 1248

  // free
A
Alex Duan 已提交
1249
  freeCommand(input);
A
Alex Duan 已提交
1250 1251 1252 1253 1254 1255 1256 1257 1258 1259
  if (input->source) {
    taosMemoryFree(input->source);
    input->source = NULL;
  }
  taosMemoryFree(input);

  return true;
}

// fill with type
1260
bool fillWithType(TAOS* con, SShellCmd* cmd, char* pre, int type) {
A
Alex Duan 已提交
1261 1262
  // get type
  STire* tire = tires[type];
1263
  char*  str = matchNextPrefix(tire, pre);
A
Alex Duan 已提交
1264 1265 1266 1267 1268
  if (str == NULL) {
    return false;
  }

  // need insert part string
1269
  char* part = str + strlen(pre);
A
Alex Duan 已提交
1270 1271 1272

  // show
  int count = strlen(part);
A
Alex Duan 已提交
1273
  shellInsertStr(cmd, part, count);
1274
  cntDel = count;  // next press tab delete current append count
A
Alex Duan 已提交
1275 1276 1277 1278 1279 1280

  taosMemoryFree(str);
  return true;
}

// fill with type
1281
bool fillTableName(TAOS* con, SShellCmd* cmd, char* pre) {
A
Alex Duan 已提交
1282
  // search stable and table
1283
  char* str = tireSearchWord(WT_VAR_STABLE, pre);
A
Alex Duan 已提交
1284 1285
  if (str == NULL) {
    str = tireSearchWord(WT_VAR_TABLE, pre);
1286
    if (str == NULL) return false;
A
Alex Duan 已提交
1287 1288 1289
  }

  // need insert part string
1290
  char* part = str + strlen(pre);
A
Alex Duan 已提交
1291 1292

  // delete autofill count last append
1293
  if (cntDel > 0) {
A
Alex Duan 已提交
1294 1295 1296 1297 1298 1299
    deleteCount(cmd, cntDel);
    cntDel = 0;
  }

  // show
  int count = strlen(part);
A
Alex Duan 已提交
1300
  shellInsertStr(cmd, part, count);
1301 1302
  cntDel = count;  // next press tab delete current append count

A
Alex Duan 已提交
1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313
  taosMemoryFree(str);
  return true;
}

//
// find last word from sql select clause
//  example :
//  1 select cou -> press tab  select count(
//  2 select count(*),su -> select count(*), sum(
//  3 select count(*), su -> select count(*), sum(
//
1314 1315 1316 1317
char* lastWord(char* p) {
  // get near from end revert find ' ' and ','
  char* p1 = strrchr(p, ' ');
  char* p2 = strrchr(p, ',');
A
Alex Duan 已提交
1318 1319

  if (p1 && p2) {
1320
    return p1 > p2 ? p1 + 1 : p2 + 1;
A
Alex Duan 已提交
1321 1322
  } else if (p1) {
    return p1 + 1;
1323
  } else if (p2) {
A
Alex Duan 已提交
1324 1325 1326
    return p2 + 1;
  } else {
    return p;
1327
  }
A
Alex Duan 已提交
1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342
}

bool fieldsInputEnd(char* sql) {
  // not in '()'
  char* p1 = strrchr(sql, '(');
  char* p2 = strrchr(sql, ')');
  if (p1 && p2 == NULL) {
    // like select count( '  '
    return false;
  } else if (p1 && p2 && p1 > p2) {
    // like select sum(age), count( ' '
    return false;
  }

  // not in ','
1343 1344 1345
  char* p3 = strrchr(sql, ',');
  char* p = p3;
  // like select ts, age,'    '
A
Alex Duan 已提交
1346 1347
  if (p) {
    ++p;
1348 1349 1350 1351
    bool  allBlank = true;  // after last ','  all char is blank
    int   cnt = 0;          // blank count , like '    ' as one blank
    char* plast = NULL;     // last blank position
    while (*p) {
A
Alex Duan 已提交
1352 1353
      if (*p == ' ') {
        plast = p;
1354
        cnt++;
A
Alex Duan 已提交
1355 1356 1357 1358 1359 1360 1361
      } else {
        allBlank = false;
      }
      ++p;
    }

    // any one word is not blank
1362
    if (allBlank) {
A
Alex Duan 已提交
1363 1364 1365 1366 1367 1368 1369 1370 1371
      return false;
    }

    // like 'select count(*),sum(age) fr' need return true
    if (plast && plast > p3 && p2 > p1 && plast > p2 && p1 > p3) {
      return true;
    }

    // if last char not ' ', then not end field, like 'select count(*), su' can fill sum(
1372
    if (sql[strlen(sql) - 1] != ' ' && cnt <= 1) {
A
Alex Duan 已提交
1373 1374 1375 1376
      return false;
    }
  }

1377 1378
  char* p4 = strrchr(sql, ' ');
  if (p4 == NULL) {
A
Alex Duan 已提交
1379 1380 1381 1382 1383 1384 1385 1386
    // only one word
    return false;
  }

  return true;
}

// need insert from
1387 1388 1389
bool needInsertFrom(char* sql, int len) {
  // last is blank
  if (sql[len - 1] != ' ') {
A
Alex Duan 已提交
1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402
    // insert from keyword
    return false;
  }

  //  select fields input is end
  if (!fieldsInputEnd(sql)) {
    return false;
  }

  // can insert from keyword
  return true;
}

A
Alex Duan 已提交
1403
// p is string following select keyword
1404
bool appendAfterSelect(TAOS* con, SShellCmd* cmd, char* sql, int32_t len) {
A
Alex Duan 已提交
1405
  char* p = strndup(sql, len);
A
Alex Duan 已提交
1406 1407

  // union all
1408
  char* p1;
A
Alex Duan 已提交
1409 1410
  do {
    p1 = strstr(p, UNION_ALL);
1411
    if (p1) {
A
Alex Duan 已提交
1412 1413 1414 1415
      p = p1 + strlen(UNION_ALL);
    }
  } while (p1);

1416 1417 1418 1419
  char* from = strstr(p, " from ");
  // last word , maybe empty string or some letters of a string
  char* last = lastWord(p);
  bool  ret = false;
A
Alex Duan 已提交
1420 1421
  if (from == NULL) {
    bool fieldEnd = fieldsInputEnd(p);
A
Alex Duan 已提交
1422
    // check fields input end then insert from keyword
1423
    if (fieldEnd && p[len - 1] == ' ') {
A
Alex Duan 已提交
1424
      shellInsertStr(cmd, "from", 4);
A
Alex Duan 已提交
1425
      taosMemoryFree(p);
A
Alex Duan 已提交
1426 1427 1428 1429
      return true;
    }

    // fill funciton
1430
    if (fieldEnd) {
A
Alex Duan 已提交
1431 1432 1433 1434 1435
      // fields is end , need match keyword
      ret = fillWithType(con, cmd, last, WT_VAR_KEYWORD);
    } else {
      ret = fillWithType(con, cmd, last, WT_VAR_FUNC);
    }
1436

A
Alex Duan 已提交
1437
    taosMemoryFree(p);
A
Alex Duan 已提交
1438 1439 1440 1441
    return ret;
  }

  // have from
1442
  char* blank = strstr(from + 6, " ");
A
Alex Duan 已提交
1443 1444 1445 1446 1447 1448 1449
  if (blank == NULL) {
    // no table name, need fill
    ret = fillTableName(con, cmd, last);
  } else {
    ret = fillWithType(con, cmd, last, WT_VAR_KEYWORD);
  }

A
Alex Duan 已提交
1450
  taosMemoryFree(p);
A
Alex Duan 已提交
1451 1452 1453
  return ret;
}

A
Alex Duan 已提交
1454 1455
int32_t searchAfterSelect(char* p, int32_t len) {
  // select * from st;
1456
  if (strncasecmp(p, "select ", 7) == 0) {
A
Alex Duan 已提交
1457
    // check nest query
1458 1459 1460 1461
    char* p1 = p + 7;
    while (1) {
      char* p2 = strstr(p1, "select ");
      if (p2 == NULL) break;
A
Alex Duan 已提交
1462 1463 1464 1465 1466 1467 1468
      p1 = p2 + 7;
    }

    return p1 - p;
  }

  // explain as select * from st;
1469
  if (strncasecmp(p, "explain select ", 15) == 0) {
A
Alex Duan 已提交
1470 1471 1472 1473
    return 15;
  }

  char* as_pos_end = strstr(p, " as select ");
1474
  if (as_pos_end == NULL) return -1;
A
Alex Duan 已提交
1475 1476 1477
  as_pos_end += 11;

  // create stream <stream_name> as select
1478 1479 1480
  if (strncasecmp(p, "create stream ", 14) == 0) {
    return as_pos_end - p;
    ;
A
Alex Duan 已提交
1481 1482 1483
  }

  // create topic <topic_name> as select
1484
  if (strncasecmp(p, "create topic ", 13) == 0) {
A
Alex Duan 已提交
1485 1486 1487 1488 1489 1490
    return as_pos_end - p;
  }

  return -1;
}

1491
bool matchSelectQuery(TAOS* con, SShellCmd* cmd) {
A
Alex Duan 已提交
1492 1493 1494 1495 1496 1497 1498
  // if continue press Tab , delete bytes by previous autofill
  if (cntDel > 0) {
    deleteCount(cmd, cntDel);
    cntDel = 0;
  }

  // match select ...
1499 1500
  int   len = cmd->commandSize;
  char* p = cmd->command;
A
Alex Duan 已提交
1501 1502 1503 1504 1505 1506 1507 1508

  // remove prefix blank
  while (p[0] == ' ' && len > 0) {
    p++;
    len--;
  }

  // special range
1509
  if (len < 7 || len > 512) {
A
Alex Duan 已提交
1510 1511 1512 1513
    return false;
  }

  // search
1514
  char*   sql_cp = strndup(p, len);
A
Alex Duan 已提交
1515 1516
  int32_t n = searchAfterSelect(sql_cp, len);
  taosMemoryFree(sql_cp);
1517 1518
  if (n == -1 || n > len) return false;
  p += n;
A
Alex Duan 已提交
1519 1520 1521 1522 1523 1524
  len -= n;

  // append
  return appendAfterSelect(con, cmd, p, len);
}

A
Alex Duan 已提交
1525
// if is input create fields or tags area, return true
1526
bool isCreateFieldsArea(char* p) {
1527 1528 1529 1530 1531 1532 1533 1534 1535 1536
  // put to while, support like create table st(ts timestamp, bin1 binary(16), bin2 + blank + TAB 
  char* p1 = strdup(p);
  bool ret = false;
  while (1) {
    char* left = strrchr(p1, '(');
    if (left == NULL) {
      // like 'create table st'
      ret = false;
      break;
    }
A
Alex Duan 已提交
1537

1538 1539 1540 1541 1542 1543
    char* right = strrchr(p1, ')');
    if (right == NULL) {
      // like 'create table st( '
      ret = true;
      break;
    }
A
Alex Duan 已提交
1544

1545 1546 1547 1548 1549 1550 1551 1552
    if (left > right) {
      // like 'create table st( ts timestamp, age int) tags(area '
      ret = true;
      break;
    }
    
    // set string end by small for next strrchr search
    *left = 0;
A
Alex Duan 已提交
1553
  }
1554
  taosMemoryFree(p1);
A
Alex Duan 已提交
1555

1556
  return ret;
A
Alex Duan 已提交
1557 1558
}

1559
bool matchCreateTable(TAOS* con, SShellCmd* cmd) {
A
Alex Duan 已提交
1560 1561 1562 1563 1564 1565 1566
  // if continue press Tab , delete bytes by previous autofill
  if (cntDel > 0) {
    deleteCount(cmd, cntDel);
    cntDel = 0;
  }

  // match select ...
1567 1568
  int   len = cmd->commandSize;
  char* p = cmd->command;
A
Alex Duan 已提交
1569 1570 1571 1572 1573 1574 1575 1576

  // remove prefix blank
  while (p[0] == ' ' && len > 0) {
    p++;
    len--;
  }

  // special range
1577
  if (len < 7 || len > 1024) {
A
Alex Duan 已提交
1578 1579 1580
    return false;
  }

1581 1582
  // select and from
  if (strncasecmp(p, "create table ", 13) != 0) {
A
Alex Duan 已提交
1583 1584 1585 1586 1587 1588 1589
    // not select query clause
    return false;
  }
  p += 13;
  len -= 13;

  char* ps = strndup(p, len);
1590 1591
  bool  ret = false;
  char* last = lastWord(ps);
A
Alex Duan 已提交
1592 1593 1594 1595 1596 1597 1598 1599 1600

  // check in create fields or tags input area
  if (isCreateFieldsArea(ps)) {
    ret = fillWithType(con, cmd, last, WT_VAR_DATATYPE);
  }

  // tags
  if (!ret) {
    // find only one ')' , can insert tags
1601
    char* p1 = strchr(ps, ')');
A
Alex Duan 已提交
1602
    if (p1) {
1603
      if (strchr(p1 + 1, ')') == NULL && strstr(p1 + 1, "tags") == NULL) {
A
Alex Duan 已提交
1604 1605 1606 1607 1608 1609
        // can insert tags keyword
        ret = fillWithType(con, cmd, last, WT_VAR_KEYTAGS);
      }
    }
  }

A
Alex Duan 已提交
1610 1611 1612
  // tb options
  if (!ret) {
    // find like create talbe st (...) tags(..)  <here is fill tb option area>
1613
    char* p1 = strchr(ps, ')');  // first ')' end
A
Alex Duan 已提交
1614
    if (p1) {
1615
      if (strchr(p1 + 1, ')')) {  // second ')' end
A
Alex Duan 已提交
1616 1617 1618 1619 1620 1621
        // here is tb options area, can insert option
        ret = fillWithType(con, cmd, last, WT_VAR_TBOPTION);
      }
    }
  }

A
Alex Duan 已提交
1622 1623 1624 1625
  taosMemoryFree(ps);
  return ret;
}

1626 1627
bool matchOther(TAOS* con, SShellCmd* cmd) {
  int   len = cmd->commandSize;
A
Alex Duan 已提交
1628 1629
  char* p = cmd->command;

A
Alex Duan 已提交
1630
  // '\\'
A
Alex Duan 已提交
1631 1632 1633
  if (p[len - 1] == '\\') {
    // append '\G'
    char a[] = "G;";
A
Alex Duan 已提交
1634
    shellInsertStr(cmd, a, 2);
A
Alex Duan 已提交
1635 1636
    return true;
  }
A
Alex Duan 已提交
1637 1638

  // too small
1639
  if (len < 8) return false;
A
Alex Duan 已提交
1640 1641 1642 1643 1644 1645 1646 1647

  // like 'from ( '
  char* sql = strndup(p, len);
  char* last = lastWord(sql);

  if (strcmp(last, "from(") == 0) {
    fillWithType(con, cmd, "", WT_VAR_KEYSELECT);
    taosMemoryFree(sql);
1648
    return true;
A
Alex Duan 已提交
1649 1650
  }
  if (strncmp(last, "(", 1) == 0) {
1651
    last += 1;
A
Alex Duan 已提交
1652 1653 1654 1655 1656 1657
  }

  char* from = strstr(sql, " from");
  // find last ' from'
  while (from) {
    char* p1 = strstr(from + 5, " from");
1658
    if (p1 == NULL) break;
A
Alex Duan 已提交
1659 1660 1661 1662 1663
    from = p1;
  }

  if (from) {
    // find next is '('
1664 1665 1666
    char* p2 = from + 5;
    bool  found = false;   // found 'from ... ( ...'  ... is any count of blank
    bool  found1 = false;  // found '('
A
Alex Duan 已提交
1667
    while (1) {
1668
      if (p2 == last || *p2 == '\0') {
A
Alex Duan 已提交
1669 1670 1671 1672 1673
        // last word or string end
        if (found1) {
          found = true;
        }
        break;
1674
      } else if (*p2 == '(') {
A
Alex Duan 已提交
1675
        found1 = true;
1676
      } else if (*p2 == ' ') {
A
Alex Duan 已提交
1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693
        // do nothing
      } else {
        // have any other char
        break;
      }

      // move next
      p2++;
    }

    if (found) {
      fillWithType(con, cmd, last, WT_VAR_KEYSELECT);
      taosMemoryFree(sql);
      return true;
    }
  }

1694
  // INSERT
A
Alex Duan 已提交
1695 1696

  taosMemoryFree(sql);
1697

A
Alex Duan 已提交
1698 1699 1700
  return false;
}

1701 1702 1703 1704 1705 1706
// last match if nothing matched
bool matchEnd(TAOS* con, SShellCmd* cmd) {
  // str dump
  bool ret = false;
  char* ps = strndup(cmd->command, cmd->commandSize);
  char* last = lastWord(ps);
A
Alex Duan 已提交
1707 1708
  char* elast = strrchr(last, '.'); // find end last
  if(elast) {
A
Alex Duan 已提交
1709
    last = elast + 1;
A
Alex Duan 已提交
1710 1711
  }

A
Alex Duan 已提交
1712 1713
  // less one char can match
  if(strlen(last) == 0 ) {
1714 1715 1716 1717
    goto _return;
  }

  // match database
A
Alex Duan 已提交
1718
  if(elast == NULL) {
A
Alex Duan 已提交
1719 1720 1721 1722 1723
    // dot need not completed with dbname
    if (fillWithType(con, cmd, last, WT_VAR_DBNAME)) {
      ret = true;
      goto _return;
    }
1724 1725
  }

1726
  if (fillWithType(con, cmd, last, WT_VAR_SYSTABLE)) {
1727
    ret = true;
1728 1729 1730 1731 1732 1733 1734 1735
    goto _return;
  }

_return:
  taosMemoryFree(ps);
  return ret;
}

A
Alex Duan 已提交
1736
// main key press tab
1737
void pressTabKey(SShellCmd* cmd) {
1738
  // check empty tab key
1739
  if (cmd->commandSize == 0) {
1740 1741 1742 1743
    // have multi line tab key
    if(cmd->bufferSize == 0) {
      showHelp();
    }
A
Alex Duan 已提交
1744
    shellShowOnScreen(cmd);
1745 1746
    return;
  }
A
Alex Duan 已提交
1747 1748 1749 1750 1751 1752 1753

  // save connection to global
  varCmd = cmd;
  bool matched = false;

  // manual match like create table st( ...
  matched = matchCreateTable(varCon, cmd);
1754
  if (matched) return;
A
Alex Duan 已提交
1755

1756
  // shellCommands match
A
Alex Duan 已提交
1757 1758 1759 1760 1761
  if (firstMatchIndex == -1) {
    matched = firstMatchCommand(varCon, cmd);
  } else {
    matched = nextMatchCommand(varCon, cmd, &shellCommands[firstMatchIndex]);
  }
1762
  if (matched) return;
A
Alex Duan 已提交
1763 1764 1765 1766

  // NOT MATCHED ANYONE
  // match other like '\G' ...
  matched = matchOther(varCon, cmd);
1767
  if (matched) return;
A
Alex Duan 已提交
1768 1769 1770

  // manual match like select * from ...
  matched = matchSelectQuery(varCon, cmd);
1771
  if (matched) return;
A
Alex Duan 已提交
1772

1773 1774 1775
  // match end
  matched = matchEnd(varCon, cmd);

1776
  return;
A
Alex Duan 已提交
1777 1778 1779 1780 1781 1782
}

// press othr key
void pressOtherKey(char c) {
  // reset global variant
  firstMatchIndex = -1;
1783 1784 1785
  lastMatchIndex = -1;
  curMatchIndex = -1;
  lastWordBytes = -1;
A
Alex Duan 已提交
1786 1787

  // var names
1788 1789
  cursorVar = -1;
  varMode = false;
A
Alex Duan 已提交
1790
  waitAutoFill = false;
1791
  cntDel = 0;
A
Alex Duan 已提交
1792 1793 1794 1795 1796 1797 1798 1799

  if (lastMatch) {
    freeMatch(lastMatch);
    lastMatch = NULL;
  }
}

// put name into name, return name length
1800 1801
int getWordName(char* p, char* name, int nameLen) {
  // remove prefix blank
A
Alex Duan 已提交
1802 1803 1804 1805 1806 1807
  while (*p == ' ') {
    p++;
  }

  // get databases name;
  int i = 0;
1808 1809
  while (p[i] != 0 && i < nameLen - 1) {
    name[i] = p[i];
A
Alex Duan 已提交
1810
    i++;
1811
    if (p[i] == ' ' || p[i] == ';' || p[i] == '(') {
A
Alex Duan 已提交
1812 1813 1814 1815 1816 1817 1818 1819 1820 1821
      // name end
      break;
    }
  }
  name[i] = 0;

  return i;
}

// deal use db, if have  'use' return true
1822 1823 1824
bool dealUseDB(char* sql) {
  // check use keyword
  if (strncasecmp(sql, "use ", 4) != 0) {
A
Alex Duan 已提交
1825 1826
    return false;
  }
1827 1828 1829

  char  db[256];
  char* p = sql + 4;
A
Alex Duan 已提交
1830
  if (getWordName(p, db, sizeof(db)) == 0) {
1831
    // no name , return
A
Alex Duan 已提交
1832 1833 1834 1835 1836
    return true;
  }

  //  dbName is previous use open db name
  if (strcasecmp(db, dbName) == 0) {
1837
    // same , no need switch
A
Alex Duan 已提交
1838 1839 1840 1841 1842 1843 1844 1845
    return true;
  }

  // switch new db
  taosThreadMutexLock(&tiresMutex);
  // STABLE set null
  STire* tire = tires[WT_VAR_STABLE];
  tires[WT_VAR_STABLE] = NULL;
1846
  if (tire) {
A
Alex Duan 已提交
1847 1848 1849 1850 1851
    freeTire(tire);
  }
  // TABLE set null
  tire = tires[WT_VAR_TABLE];
  tires[WT_VAR_TABLE] = NULL;
1852
  if (tire) {
A
Alex Duan 已提交
1853 1854 1855 1856 1857 1858 1859 1860 1861 1862
    freeTire(tire);
  }
  // save
  strcpy(dbName, db);
  taosThreadMutexUnlock(&tiresMutex);

  return true;
}

// deal create, if have 'create' return true
1863 1864 1865
bool dealCreateCommand(char* sql) {
  // check keyword
  if (strncasecmp(sql, "create ", 7) != 0) {
A
Alex Duan 已提交
1866 1867
    return false;
  }
1868 1869 1870

  char  name[1024];
  char* p = sql + 7;
A
Alex Duan 已提交
1871
  if (getWordName(p, name, sizeof(name)) == 0) {
1872
    // no name , return
A
Alex Duan 已提交
1873 1874 1875 1876 1877 1878 1879 1880
    return true;
  }

  int type = -1;
  //  dbName is previous use open db name
  if (strcasecmp(name, "database") == 0) {
    type = WT_VAR_DBNAME;
  } else if (strcasecmp(name, "table") == 0) {
1881
    if (strstr(sql, " tags") != NULL && strstr(sql, " using ") == NULL)
A
Alex Duan 已提交
1882 1883 1884 1885 1886
      type = WT_VAR_STABLE;
    else
      type = WT_VAR_TABLE;
  } else if (strcasecmp(name, "user") == 0) {
    type = WT_VAR_USERNAME;
A
Alex Duan 已提交
1887 1888 1889 1890
  } else if (strcasecmp(name, "topic") == 0) {
    type = WT_VAR_TOPIC;
  } else if (strcasecmp(name, "stream") == 0) {
    type = WT_VAR_STREAM;
A
Alex Duan 已提交
1891
  } else {
1892
    // no match , return
A
Alex Duan 已提交
1893 1894 1895 1896 1897 1898 1899 1900
    return true;
  }

  // move next
  p += strlen(name);

  // get next word , that is table name
  if (getWordName(p, name, sizeof(name)) == 0) {
1901
    // no name , return
A
Alex Duan 已提交
1902 1903 1904 1905 1906 1907 1908
    return true;
  }

  // switch new db
  taosThreadMutexLock(&tiresMutex);
  // STABLE set null
  STire* tire = tires[type];
1909
  if (tire) {
A
Alex Duan 已提交
1910 1911 1912 1913 1914 1915 1916 1917
    insertWord(tire, name);
  }
  taosThreadMutexUnlock(&tiresMutex);

  return true;
}

// deal create, if have 'drop' return true
1918 1919 1920
bool dealDropCommand(char* sql) {
  // check keyword
  if (strncasecmp(sql, "drop ", 5) != 0) {
A
Alex Duan 已提交
1921 1922
    return false;
  }
1923 1924 1925

  char  name[1024];
  char* p = sql + 5;
A
Alex Duan 已提交
1926
  if (getWordName(p, name, sizeof(name)) == 0) {
1927
    // no name , return
A
Alex Duan 已提交
1928 1929 1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940
    return true;
  }

  int type = -1;
  //  dbName is previous use open db name
  if (strcasecmp(name, "database") == 0) {
    type = WT_VAR_DBNAME;
  } else if (strcasecmp(name, "table") == 0) {
    type = WT_VAR_ALLTABLE;
  } else if (strcasecmp(name, "dnode") == 0) {
    type = WT_VAR_DNODEID;
  } else if (strcasecmp(name, "user") == 0) {
    type = WT_VAR_USERNAME;
A
Alex Duan 已提交
1941 1942 1943 1944
  } else if (strcasecmp(name, "topic") == 0) {
    type = WT_VAR_TOPIC;
  } else if (strcasecmp(name, "stream") == 0) {
    type = WT_VAR_STREAM;
A
Alex Duan 已提交
1945
  } else {
1946
    // no match , return
A
Alex Duan 已提交
1947 1948 1949 1950 1951 1952 1953 1954
    return true;
  }

  // move next
  p += strlen(name);

  // get next word , that is table name
  if (getWordName(p, name, sizeof(name)) == 0) {
1955
    // no name , return
A
Alex Duan 已提交
1956 1957 1958 1959 1960 1961
    return true;
  }

  // switch new db
  taosThreadMutexLock(&tiresMutex);
  // STABLE set null
1962
  if (type == WT_VAR_ALLTABLE) {
A
Alex Duan 已提交
1963 1964 1965
    bool del = false;
    // del in stable
    STire* tire = tires[WT_VAR_STABLE];
1966
    if (tire) del = deleteWord(tire, name);
A
Alex Duan 已提交
1967
    // del in table
1968
    if (!del) {
A
Alex Duan 已提交
1969
      tire = tires[WT_VAR_TABLE];
1970
      if (tire) del = deleteWord(tire, name);
A
Alex Duan 已提交
1971 1972 1973 1974
    }
  } else {
    // OTHER TYPE
    STire* tire = tires[type];
1975
    if (tire) deleteWord(tire, name);
A
Alex Duan 已提交
1976 1977 1978 1979 1980 1981 1982 1983
  }
  taosThreadMutexUnlock(&tiresMutex);

  return true;
}

// callback autotab module after shell sql execute
void callbackAutoTab(char* sqlstr, TAOS* pSql, bool usedb) {
1984
  char* sql = sqlstr;
A
Alex Duan 已提交
1985 1986 1987 1988 1989
  // remove prefix blank
  while (*sql == ' ') {
    sql++;
  }

1990
  if (dealUseDB(sql)) {
A
Alex Duan 已提交
1991
    // change to new db
A
Alex Duan 已提交
1992
    if (!varRunOnce) updateTireValue(WT_VAR_STABLE, false);
1993
    return;
A
Alex Duan 已提交
1994 1995 1996
  }

  // create command add name to autotab
1997 1998
  if (dealCreateCommand(sql)) {
    return;
A
Alex Duan 已提交
1999 2000 2001
  }

  // drop command remove name from autotab
2002 2003
  if (dealDropCommand(sql)) {
    return;
A
Alex Duan 已提交
2004 2005
  }

2006
  return;
A
Alex Duan 已提交
2007
}