File: sql_normalizer.cpp

package info (click to toggle)
mysql-workbench 6.3.8%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: stretch
  • size: 113,932 kB
  • ctags: 87,814
  • sloc: ansic: 955,521; cpp: 427,465; python: 59,728; yacc: 59,129; xml: 54,204; sql: 7,091; objc: 965; makefile: 638; sh: 613; java: 237; perl: 30; ruby: 6; php: 1
file content (107 lines) | stat: -rw-r--r-- 3,149 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
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
/* 
 * Copyright (c) 2007, 2014, Oracle and/or its affiliates. All rights reserved.
 *
 * 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., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301  USA
 */

#include "sql_normalizer.h"

struct normalized_string_builder
{
    bool skip_next;
    bool check_comment;
    std::string& result;
    std::string::value_type quote;
    std::string::value_type skip_until;
    normalized_string_builder(std::string& result_str):result(result_str)
    {
        skip_next = false;
        check_comment = false;
        quote = 0;
        skip_until = 0;
    }

    void operator() (const std::string::value_type c)
    {
        if (quote)
        {
            if(skip_next)//previos char was \ this one is ecaped
                skip_next = false;
            else if(c == '\\')//Skip next char as escaped
                skip_next = true;
            else if(c == quote)//end of quoted string
                quote = 0;
            result+=c;
        }else
        {
            if (skip_until)
            {
                if (check_comment)
                {
                    if(c == '/')
                    {
                        check_comment = false;
                        skip_until = 0;
                        return;
                    }

                }
                if (c == skip_until)
                {
                    if(c == '*')
                      check_comment = true;
                    else
                      skip_until = 0;
                }
              return;
            }
            if (c == '#')
            {
                skip_until = '\n';
                return;
            }
            if(c == '/')
            {
                check_comment = true;
                return;
            }
            if (check_comment)
            {
                if(c == '*')
                {
                    skip_until = '*';
                    return;
                }
                else
                    result+= '/';
            }
            if((c == '\'') || (c == '"') || (c == '`')) //quoted string started
                quote = c;
            if (!std::isspace(c))
                result+= c;//std::toupper(c); doesn't work since chema.table will go uppercase, while `schema`.`table` won't
        }    
    }

};

std::string Sql_normalizer::remove_inter_token_spaces(const std::string &text)
{
    std::string result;
    result.reserve(text.size());
    normalized_string_builder builder(result);
    std::for_each(text.begin(),text.end(),builder);
    return result;
}