# Copyright (c) 2009, 2019, 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, version 2.0,
# as published by the Free Software Foundation.
#
# This program is designed to work with certain software (including
# but not limited to OpenSSL) that is licensed under separate terms, as
# designated in a particular file or component or in included license
# documentation.  The authors of MySQL hereby grant you an additional
# permission to link the program and your derivative works with the
# separately licensed software that they have either included with
# the program or referenced in the documentation.
# 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, version 2.0, 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

# import the wb module
from wb import *
# import the grt module
import grt
# import the mforms module for GUI stuff
import mforms



# define this Python module as a GRT module
ModuleInfo = DefineModule(name= "TextUtils", author= "Oracle Corp.", version="1.0")




#@ModuleInfo.plugin("wb.text.doStuff", caption= "Do Stuff", input= [wbinputs.currentQueryBuffer()], pluginMenu= "Text")
#@ModuleInfo.export(grt.INT, grt.classes.db_query_QueryBuffer)
#def doStuff(qbuffer):
#
#  return 0


@ModuleInfo.exportFilter("wb.text.sort", "Sort Selection")
def sortText(text):
  lines = text.split("\n")
  lines.sort()
  return "\n".join(lines)


@ModuleInfo.exportFilter("wb.text.quoteString", "Quote and Escape String")
def quoteString(text):
  escaped= ""
  for c in text:
    escape= None
    if ord(c) == 0:
      escape= '0'
    elif c == '\n':
      escape= 'n'
    elif c == '\r':
      escape= 'r'
    elif c == '\\':
      escape= '\\'
    elif c == "'":
      escape= "'"
    elif ord(c) == 0o32:
      escape= 'Z' 

    if escape is not None:
      escaped += "\\"+escape
    else:
      escaped += c
  return "'"+escaped+"'"



