text: #!/usr/bin/python
  # -*- coding: utf-8 -*-
  # Copyright: (c) 2017, Ansible Project
  # GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
  
  from __future__ import absolute_import, division, print_function
  __metaclass__ = type
  
  DOCUMENTATION = '''
  ---
  module: proxysql_query_rules
  author: "Ben Mildren (@bmildren)"
  short_description: Modifies query rules using the proxysql admin interface
  description:
     - The M(community.proxysql.proxysql_query_rules) module modifies query rules using the
       proxysql admin interface.
  options:
    rule_id:
      description:
        - The unique id of the rule. Rules are processed in rule_id order.
      type: int
    active:
      description:
        - A rule with I(active) set to C(False) will be tracked in the database,
          but will be never loaded in the in-memory data structures.
      type: bool
    username:
      description:
        - Filtering criteria matching username.  If I(username) is non-NULL, a
          query will match only if the connection is made with the correct
          username.
      type: str
    schemaname:
      description:
        - Filtering criteria matching schemaname. If I(schemaname) is non-NULL, a
          query will match only if the connection uses schemaname as its default
          schema.
      type: str
    flagIN:
      description:
        - Used in combination with I(flagOUT) and I(apply) to create chains of
          rules.
      type: int
    client_addr:
      description:
        - Match traffic from a specific source.
      type: str
    proxy_addr:
      description:
        - Match incoming traffic on a specific local IP.
      type: str
    proxy_port:
      description:
        - Match incoming traffic on a specific local port.
      type: int
    digest:
      description:
        - Match queries with a specific digest, as returned by
          stats_mysql_query_digest.digest.
      type: str
    match_digest:
      description:
        - Regular expression that matches the query digest. The dialect of
          regular expressions used is that of re2 - U(https://github.com/google/re2).
      type: str
    match_pattern:
      description:
        - Regular expression that matches the query text. The dialect of regular
          expressions used is that of re2 - U(https://github.com/google/re2).
      type: str
    negate_match_pattern:
      description:
        - If I(negate_match_pattern) is set to C(True), only queries not matching
          the query text will be considered as a match. This acts as a NOT
          operator in front of the regular expression matching against
          match_pattern.
      type: bool
    re_modifiers:
      description:
        - Comma separated list of options to modify the behavior of the RE engine.
          With C(CASELESS) the match is case insensitive. With C(GLOBAL) the replace
          is global (replaces all matches and not just the first).
          For backward compatibility, only C(CASELESS) is the enabled by default.
      type: str
      version_added: "1.3.0"
    flagOUT:
      description:
        - Used in combination with I(flagIN) and apply to create chains of rules.
          When set, I(flagOUT) signifies the I(flagIN) to be used in the next
          chain of rules.
      type: int
    replace_pattern:
      description:
        - This is the pattern with which to replace the matched pattern. Note
          that this is optional, and when omitted, the query processor will only
          cache, route, or set other parameters without rewriting.
      type: str
    destination_hostgroup:
      description:
        - Route matched queries to this hostgroup. This happens unless there is a
          started transaction and the logged in user has
          I(transaction_persistent) set to C(True) (see M(community.proxysql.proxysql_mysql_users)).
      type: int
    cache_ttl:
      description:
        - The number of milliseconds for which to cache the result of the query.
          Note in ProxySQL 1.1 I(cache_ttl) was in seconds.
      type: int
    cache_empty_result:
      description:
        - Controls if resultset without rows will be cached or not.
      type: bool
      version_added: '1.1.0'
    multiplex:
      description:
        - If C(0), multiplex will be disabled.
        - If C(1), try to enable multiplex. There can be other conditions preventing this (for example, user variables or transactions).
        - If C(2), multiplexing is not disabled for just the current query.
        - By default, does not change multiplexing policies.
      type: int
      choices: [0, 1, 2]
      version_added: '1.1.0'
    timeout:
      description:
        - The maximum timeout in milliseconds with which the matched or rewritten
          query should be executed. If a query run for longer than the specific
          threshold, the query is automatically killed. If timeout is not
          specified, the global variable mysql-default_query_timeout applies.
      type: int
    retries:
      description:
        - The maximum number of times a query needs to be re-executed in case of
          detected failure during the execution of the query. If retries is not
          specified, the global variable mysql-query_retries_on_failure applies.
      type: int
    delay:
      description:
        - Number of milliseconds to delay the execution of the query. This is
          essentially a throttling mechanism and QoS, and allows a way to give
          priority to queries over others. This value is added to the
          mysql-default_query_delay global variable that applies to all queries.
      type: int
    next_query_flagIN:
      description:
        - When is set, its value will become the I(flagIN) value for the next queries.
      type: int
      version_added: "1.3.0"
    mirror_flagOUT:
      description:
        - Enables query mirroring. If set I(mirror_flagOUT) can be used to
          evaluates the mirrored query against the specified chain of rules.
      type: int
    mirror_hostgroup:
      description:
        - Enables query mirroring. If set I(mirror_hostgroup) can be used to
          mirror queries to the same or different hostgroup.
      type: int
    OK_msg:
      description:
        - The specified message will be returned for a query that uses the
          defined rule.
      type: str
      version_added: '1.1.0'
    error_msg:
      description:
        - Query will be blocked, and the specified error_msg will be returned to
          the client.
      type: str
    log:
      description:
        - Query will be logged.
      type: bool
    apply:
      description:
        - Used in combination with I(flagIN) and I(flagOUT) to create chains of
          rules. Setting apply to True signifies the last rule to be applied.
      type: bool
    comment:
      description:
        - Free form text field, usable for a descriptive comment of the query
          rule.
      type: str
    state:
      description:
        - When C(present) - adds the rule, when C(absent) - removes the rule.
      type: str
      choices: [ "present", "absent" ]
      default: present
    force_delete:
      description:
        - By default we avoid deleting more than one schedule in a single batch,
          however if you need this behaviour and you are not concerned about the
          schedules deleted, you can set I(force_delete) to C(True).
      type: bool
      default: False
  extends_documentation_fragment:
  - community.proxysql.proxysql.managing_config
  - community.proxysql.proxysql.connectivity
  notes:
  - Supports C(check_mode).
  '''
  
  EXAMPLES = '''
  ---
  # This example adds a rule to redirect queries from a specific user to another
  # hostgroup, it saves the mysql query rule config to disk, but avoids loading
  # the mysql query config config to runtime (this might be because several
  # rules are being added and the user wants to push the config to runtime in a
  # single batch using the community.general.proxysql_manage_config module). It
  # uses supplied credentials to connect to the proxysql admin interface.
  
  - name: Add a rule
    community.proxysql.proxysql_query_rules:
      login_user: admin
      login_password: admin
      username: 'guest_ro'
      match_pattern: "^SELECT.*"
      destination_hostgroup: 1
      active: 1
      retries: 3
      state: present
      load_to_runtime: False
  
  # This example demonstrates the situation, if your application tries to set a
  # variable that will disable multiplexing, and you think it can be filtered out,
  # you can create a filter that returns OK without executing the request.
  
  - name: Add a filter rule
    community.proxysql.proxysql_query_rules:
      login_user: admin
      login_password: admin
      match_digest: '^SET @@wait_timeout = ?'
      active: 1
      OK_msg: 'The wait_timeout variable is ignored'
  
  # This example adds a caching rule for a query that matches the digest.
  # The query digest can be obtained from the `stats_mysql_query_digest`
  # table. `cache_ttl` is specified in milliseconds. Empty responses are
  # not cached.
  
  - name: Add a cache rule
    community.proxysql.proxysql_query_rules:
      login_user: admin
      login_password: admin
      rule_id: 1
      digest: 0xECA450EA500A9A55
      cache_ttl: 30000
      cache_empty_result: no
      destination_hostgroup: 1
      active: yes
      state: present
      save_to_disk: yes
      load_to_runtime: yes
  
  # This example demonstrates how to prevent disabling multiplexing for
  # situations where a request contains @.
  
  - name: Add a rule with multiplex
    community.proxysql.proxysql_query_rules:
      login_user: admin
      login_password: admin
      rule_id: 1
      active: 1
      match_digest: '^SELECT @@max_allowed_packet'
      multiplex: 2
  
  # This example demonstrates how to use next_query_flagIN argument. It allows
  # ProxySQL query rules to be chained. The examples shows how you can have SELECTS
  # immediately follow INSERT/UPDATE/DELETE statements to query the primary hostgroup
  # and avoid replication lag
  
  - name: Add insert query rule
    proxysql_query_rules:
      match_digest: "^INSERT"
      destination_hostgroup: 1,
      next_query_flagIN: 1
  
  - name: Add update query rule
    proxysql_query_rules:
      match_digest: "^UPDATE"
      destination_hostgroup: 1,
      next_query_flagIN: 1
  
  - name: Add delete query rules
    proxysql_query_rules:
      match_digest: "^DELETE"
      destination_hostgroup: 1,
      next_query_flagIN: 1
  
  - name: Add insert query rules
    proxysql_query_rules:
      match_digest: ".*"
      destination_hostgroup: 1,
      next_query_flagIN: 1
      comment: Match every queries after an INSERT/UPDATE/DELETE query
  
  # This example removes all rules that use the username 'guest_ro', saves the
  # mysql query rule config to disk, and dynamically loads the mysql query rule
  # config to runtime.  It uses credentials in a supplied config file to connect
  # to the proxysql admin interface.
  
  - name: Remove rules
    community.proxysql.proxysql_query_rules:
      config_file: '~/proxysql.cnf'
      username: 'guest_ro'
      state: absent
      force_delete: true
  '''
  
  RETURN = '''
  stdout:
      description: The mysql user modified or removed from proxysql.
      returned: On create/update will return the newly modified rule, in all
                other cases will return a list of rules that match the supplied
                criteria.
      type: dict
      "sample": {
          "changed": true,
          "msg": "Added rule to mysql_query_rules",
          "rules": [
              {
                  "active": "0",
                  "apply": "0",
                  "cache_ttl": null,
                  "cache_empty_result": null,
                  "multiplex": null,
                  "client_addr": null,
                  "comment": null,
                  "delay": null,
                  "destination_hostgroup": 1,
                  "digest": null,
                  "OK_msg": null,
                  "error_msg": null,
                  "flagIN": "0",
                  "flagOUT": null,
                  "log": null,
                  "match_digest": null,
                  "match_pattern": null,
                  "mirror_flagOUT": null,
                  "mirror_hostgroup": null,
                  "negate_match_pattern": "0",
                  "proxy_addr": null,
                  "proxy_port": null,
                  "reconnect": null,
                  "replace_pattern": null,
                  "retries": null,
                  "rule_id": "1",
                  "schemaname": null,
                  "timeout": null,
                  "username": "guest_ro"
              }
          ],
          "state": "present"
      }
  '''
  
  from ansible.module_utils.basic import AnsibleModule
  from ansible_collections.community.proxysql.plugins.module_utils.mysql import (
      mysql_connect,
      mysql_driver,
      proxysql_common_argument_spec,
      save_config_to_disk,
      load_config_to_runtime,
  )
  from ansible.module_utils.six import iteritems
  from ansible.module_utils._text import to_native
  
  # ===========================================
  # proxysql module specific support methods.
  #
  
  
  class ProxyQueryRule(object):
  
      def __init__(self, module, version):
          self.state = module.params["state"]
          self.force_delete = module.params["force_delete"]
          self.save_to_disk = module.params["save_to_disk"]
          self.load_to_runtime = module.params["load_to_runtime"]
  
          config_data_keys = ["rule_id",
                              "active",
                              "username",
                              "schemaname",
                              "flagIN",
                              "client_addr",
                              "proxy_addr",
                              "proxy_port",
                              "digest",
                              "match_digest",
                              "match_pattern",
                              "negate_match_pattern",
                              "re_modifiers",
                              "flagOUT",
                              "replace_pattern",
                              "destination_hostgroup",
                              "cache_ttl",
                              "multiplex",
                              "timeout",
                              "retries",
                              "delay",
                              "next_query_flagIN",
                              "mirror_flagOUT",
                              "mirror_hostgroup",
                              "error_msg",
                              "OK_msg",
                              "multiplex",
                              "log",
                              "apply",
                              "comment"]
  
          if version.get('major') >= 2:
              config_data_keys.append("cache_empty_result")
  
          self.config_data = dict((k, module.params[k])
                                  for k in config_data_keys)
  
      def check_rule_pk_exists(self, cursor):
          query_string = \
              """SELECT count(*) AS `rule_count`
                 FROM mysql_query_rules
                 WHERE rule_id = %s"""
  
          query_data = \
              [self.config_data["rule_id"]]
  
          cursor.execute(query_string, query_data)
          check_count = cursor.fetchone()
          return (int(check_count['rule_count']) > 0)
  
      def check_rule_cfg_exists(self, cursor):
          query_string = \
              """SELECT count(*) AS `rule_count`
                 FROM mysql_query_rules"""
  
          cols = 0
          query_data = []
  
          for col, val in iteritems(self.config_data):
              if val is not None:
                  cols += 1
                  query_data.append(val)
                  if cols == 1:
                      query_string += "\n WHERE " + col + " = %s"
                  else:
                      query_string += "\n  AND " + col + " = %s"
  
          if cols > 0:
              cursor.execute(query_string, query_data)
          else:
              cursor.execute(query_string)
          check_count = cursor.fetchone()
          return int(check_count['rule_count'])
  
      def get_rule_config(self, cursor, created_rule_id=None):
          query_string = \
              """SELECT *
                 FROM mysql_query_rules"""
  
          if created_rule_id:
              query_data = [created_rule_id, ]
              query_string += "\nWHERE rule_id = %s"
  
              cursor.execute(query_string, query_data)
              rule = cursor.fetchone()
          else:
              cols = 0
              query_data = []
  
              for col, val in iteritems(self.config_data):
                  if val is not None:
                      cols += 1
                      query_data.append(val)
                      if cols == 1:
                          query_string += "\n WHERE " + col + " = %s"
                      else:
                          query_string += "\n  AND " + col + " = %s"
  
              if cols > 0:
                  cursor.execute(query_string, query_data)
              else:
                  cursor.execute(query_string)
              rule = cursor.fetchall()
  
          return rule
  
      def create_rule_config(self, cursor):
          query_string = \
              """INSERT INTO mysql_query_rules ("""
  
          cols = 0
          query_data = []
  
          for col, val in iteritems(self.config_data):
              if val is not None:
                  cols += 1
                  query_data.append(val)
                  query_string += "\n" + col + ","
  
          query_string = query_string[:-1]
  
          query_string += \
              (")\n" +
               "VALUES (" +
               "%s ," * cols)
  
          query_string = query_string[:-2]
          query_string += ")"
  
          cursor.execute(query_string, query_data)
          new_rule_id = cursor.lastrowid
          return True, new_rule_id
  
      def update_rule_config(self, cursor):
          query_string = """UPDATE mysql_query_rules"""
  
          cols = 0
          query_data = []
  
          for col, val in iteritems(self.config_data):
              if val is not None and col != "rule_id":
                  cols += 1
                  query_data.append(val)
                  if cols == 1:
                      query_string += "\nSET " + col + "= %s,"
                  else:
                      query_string += "\n    " + col + " = %s,"
  
          query_string = query_string[:-1]
          query_string += "\nWHERE rule_id = %s"
  
          query_data.append(self.config_data["rule_id"])
  
          cursor.execute(query_string, query_data)
          return True
  
      def delete_rule_config(self, cursor):
          query_string = \
              """DELETE FROM mysql_query_rules"""
  
          cols = 0
          query_data = []
  
          for col, val in iteritems(self.config_data):
              if val is not None:
                  cols += 1
                  query_data.append(val)
                  if cols == 1:
                      query_string += "\n WHERE " + col + " = %s"
                  else:
                      query_string += "\n  AND " + col + " = %s"
  
          if cols > 0:
              cursor.execute(query_string, query_data)
          else:
              cursor.execute(query_string)
          check_count = cursor.rowcount
          return True, int(check_count)
  
      def manage_config(self, cursor, state):
          if state:
              if self.save_to_disk:
                  save_config_to_disk(cursor, "QUERY RULES")
              if self.load_to_runtime:
                  load_config_to_runtime(cursor, "QUERY RULES")
  
      def create_rule(self, check_mode, result, cursor):
          if not check_mode:
              result['changed'], new_rule_id = \
                  self.create_rule_config(cursor)
              result['msg'] = "Added rule to mysql_query_rules"
              self.manage_config(cursor,
                                 result['changed'])
              result['rules'] = \
                  self.get_rule_config(cursor, new_rule_id)
          else:
              result['changed'] = True
              result['msg'] = ("Rule would have been added to" +
                               " mysql_query_rules, however" +
                               " check_mode is enabled.")
  
      def update_rule(self, check_mode, result, cursor):
          if not check_mode:
              result['changed'] = \
                  self.update_rule_config(cursor)
              result['msg'] = "Updated rule in mysql_query_rules"
              self.manage_config(cursor,
                                 result['changed'])
              result['rules'] = \
                  self.get_rule_config(cursor)
          else:
              result['changed'] = True
              result['msg'] = ("Rule would have been updated in" +
                               " mysql_query_rules, however" +
                               " check_mode is enabled.")
  
      def delete_rule(self, check_mode, result, cursor):
          if not check_mode:
              result['rules'] = \
                  self.get_rule_config(cursor)
              result['changed'], result['rows_affected'] = \
                  self.delete_rule_config(cursor)
              result['msg'] = "Deleted rule from mysql_query_rules"
              self.manage_config(cursor,
                                 result['changed'])
          else:
              result['changed'] = True
              result['msg'] = ("Rule would have been deleted from" +
                               " mysql_query_rules, however" +
                               " check_mode is enabled.")
  
  # ===========================================
  # Module execution.
  #
  
  
  def main():
      argument_spec = proxysql_common_argument_spec()
      argument_spec.update(
          rule_id=dict(type='int'),
          active=dict(type='bool'),
          username=dict(type='str'),
          schemaname=dict(type='str'),
          flagIN=dict(type='int'),
          client_addr=dict(type='str'),
          proxy_addr=dict(type='str'),
          proxy_port=dict(type='int'),
          digest=dict(type='str'),
          match_digest=dict(type='str'),
          match_pattern=dict(type='str'),
          negate_match_pattern=dict(type='bool'),
          re_modifiers=dict(type='str'),
          flagOUT=dict(type='int'),
          replace_pattern=dict(type='str'),
          destination_hostgroup=dict(type='int'),
          cache_ttl=dict(type='int'),
          cache_empty_result=dict(type='bool'),
          multiplex=dict(type='int', choices=[0, 1, 2]),
          timeout=dict(type='int'),
          retries=dict(type='int'),
          delay=dict(type='int'),
          next_query_flagIN=dict(type='int'),
          mirror_flagOUT=dict(type='int'),
          mirror_hostgroup=dict(type='int'),
          OK_msg=dict(type='str'),
          error_msg=dict(type='str'),
          log=dict(type='bool'),
          apply=dict(type='bool'),
          comment=dict(type='str'),
          state=dict(default='present', choices=['present',
                                                 'absent']),
          force_delete=dict(default=False, type='bool'),
          save_to_disk=dict(default=True, type='bool'),
          load_to_runtime=dict(default=True, type='bool')
      )
  
      module = AnsibleModule(
          supports_check_mode=True,
          argument_spec=argument_spec
      )
  
      login_user = module.params["login_user"]
      login_password = module.params["login_password"]
      config_file = module.params["config_file"]
  
      cursor = None
      try:
          cursor, db_conn, version = mysql_connect(module,
                                                   login_user,
                                                   login_password,
                                                   config_file,
                                                   cursor_class='DictCursor')
      except mysql_driver.Error as e:
          module.fail_json(
              msg="unable to connect to ProxySQL Admin Module.. %s" % to_native(e)
          )
  
      proxysql_query_rule = ProxyQueryRule(module, version)
      result = {}
  
      result['state'] = proxysql_query_rule.state
  
      if proxysql_query_rule.state == "present":
          try:
              if not proxysql_query_rule.check_rule_cfg_exists(cursor):
                  if proxysql_query_rule.config_data["rule_id"] and \
                     proxysql_query_rule.check_rule_pk_exists(cursor):
                      proxysql_query_rule.update_rule(module.check_mode,
                                                      result,
                                                      cursor)
                  else:
                      proxysql_query_rule.create_rule(module.check_mode,
                                                      result,
                                                      cursor)
              else:
                  result['changed'] = False
                  result['msg'] = ("The rule already exists in" +
                                   " mysql_query_rules and doesn't need to be" +
                                   " updated.")
                  result['rules'] = \
                      proxysql_query_rule.get_rule_config(cursor)
  
          except mysql_driver.Error as e:
              module.fail_json(
                  msg="unable to modify rule.. %s" % to_native(e)
              )
  
      elif proxysql_query_rule.state == "absent":
          try:
              existing_rules = proxysql_query_rule.check_rule_cfg_exists(cursor)
              if existing_rules > 0:
                  if existing_rules == 1 or \
                     proxysql_query_rule.force_delete:
                      proxysql_query_rule.delete_rule(module.check_mode,
                                                      result,
                                                      cursor)
                  else:
                      module.fail_json(
                          msg=("Operation would delete multiple rules" +
                               " use force_delete to override this")
                      )
              else:
                  result['changed'] = False
                  result['msg'] = ("The rule is already absent from the" +
                                   " mysql_query_rules memory configuration")
          except mysql_driver.Error as e:
              module.fail_json(
                  msg="unable to remove rule.. %s" % to_native(e)
              )
  
      module.exit_json(**result)
  
  
  if __name__ == '__main__':
      main()
  