|
| 1 | +""" |
| 2 | +Uncoder IO Community Edition License |
| 3 | +----------------------------------------------------------------- |
| 4 | +Copyright (c) 2024 SOC Prime, Inc. |
| 5 | +
|
| 6 | +Licensed under the Apache License, Version 2.0 (the "License"); |
| 7 | +you may not use this file except in compliance with the License. |
| 8 | +You may obtain a copy of the License at |
| 9 | +
|
| 10 | + http://www.apache.org/licenses/LICENSE-2.0 |
| 11 | +
|
| 12 | +Unless required by applicable law or agreed to in writing, software |
| 13 | +distributed under the License is distributed on an "AS IS" BASIS, |
| 14 | +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 15 | +See the License for the specific language governing permissions and |
| 16 | +limitations under the License. |
| 17 | +----------------------------------------------------------------- |
| 18 | +""" |
| 19 | +fromapp.translator.constimportDEFAULT_VALUE_TYPE |
| 20 | +fromapp.translator.core.custom_types.valuesimportValueType |
| 21 | +fromapp.translator.core.models.platform_detailsimportPlatformDetails |
| 22 | +fromapp.translator.core.renderimportBaseFieldValueRender,PlatformQueryRender |
| 23 | +fromapp.translator.managersimportrender_manager |
| 24 | +fromapp.translator.platforms.anomali.constimportanomali_query_details |
| 25 | +fromapp.translator.platforms.anomali.mappingimportAnomaliMappings,anomali_query_mappings |
| 26 | +fromapp.translator.platforms.base.sql.str_value_managerimportsql_str_value_manager |
| 27 | + |
| 28 | + |
| 29 | +classAnomaliFieldValueRender(BaseFieldValueRender): |
| 30 | +details:PlatformDetails=anomali_query_details |
| 31 | +str_value_manager=sql_str_value_manager |
| 32 | + |
| 33 | +@staticmethod |
| 34 | +def_wrap_str_value(value:str)->str: |
| 35 | +returnf"'{value}'" |
| 36 | + |
| 37 | +defequal_modifier(self,field:str,value:DEFAULT_VALUE_TYPE)->str: |
| 38 | +ifisinstance(value,list): |
| 39 | +returnf"({self.or_token.join([self.equal_modifier(field=field,value=v)forvinvalue])})" |
| 40 | +returnf"{field} ={self._pre_process_value(field,value,wrap_str=True)}" |
| 41 | + |
| 42 | +defnot_equal_modifier(self,field:str,value:DEFAULT_VALUE_TYPE)->str: |
| 43 | +ifisinstance(value,list): |
| 44 | +returnf"({self.or_token.join([self.not_equal_modifier(field=field,value=v)forvinvalue])})" |
| 45 | +returnf"{field} !={self._pre_process_value(field,value,wrap_str=True)}" |
| 46 | + |
| 47 | +defless_modifier(self,field:str,value:DEFAULT_VALUE_TYPE)->str: |
| 48 | +returnf"{field} <{self._pre_process_value(field,value,wrap_str=True)}" |
| 49 | + |
| 50 | +defless_or_equal_modifier(self,field:str,value:DEFAULT_VALUE_TYPE)->str: |
| 51 | +returnf"{field} <={self._pre_process_value(field,value,wrap_str=True)}" |
| 52 | + |
| 53 | +defgreater_modifier(self,field:str,value:DEFAULT_VALUE_TYPE)->str: |
| 54 | +returnf"{field} >{self._pre_process_value(field,value,wrap_str=True)}" |
| 55 | + |
| 56 | +defgreater_or_equal_modifier(self,field:str,value:DEFAULT_VALUE_TYPE)->str: |
| 57 | +returnf"{field} >={self._pre_process_value(field,value,wrap_str=True)}" |
| 58 | + |
| 59 | +defcontains_modifier(self,field:str,value:DEFAULT_VALUE_TYPE)->str: |
| 60 | +ifisinstance(value,list): |
| 61 | +returnf"({self.or_token.join(self.contains_modifier(field=field,value=v)forvinvalue)})" |
| 62 | +returnf"{field} like '%{self._pre_process_value(field,value)}%'" |
| 63 | + |
| 64 | +defendswith_modifier(self,field:str,value:DEFAULT_VALUE_TYPE)->str: |
| 65 | +ifisinstance(value,list): |
| 66 | +returnf"({self.or_token.join(self.endswith_modifier(field=field,value=v)forvinvalue)})" |
| 67 | +returnf"{field} like '%{self._pre_process_value(field,value)}'" |
| 68 | + |
| 69 | +defstartswith_modifier(self,field:str,value:DEFAULT_VALUE_TYPE)->str: |
| 70 | +ifisinstance(value,list): |
| 71 | +returnf"({self.or_token.join(self.startswith_modifier(field=field,value=v)forvinvalue)})" |
| 72 | +returnf"{field} like '{self._pre_process_value(field,value)}%'" |
| 73 | + |
| 74 | +defregex_modifier(self,field:str,value:DEFAULT_VALUE_TYPE)->str: |
| 75 | +ifisinstance(value,list): |
| 76 | +returnf"({self.or_token.join(self.regex_modifier(field=field,value=v)forvinvalue)})" |
| 77 | +regex_str=self._pre_process_value(field,value,value_type=ValueType.regex_value,wrap_str=True) |
| 78 | +returnf"regexp_like({field},{regex_str})" |
| 79 | + |
| 80 | +defkeywords(self,field:str,value:DEFAULT_VALUE_TYPE)->str: |
| 81 | +returnf'message contains "{self._pre_process_value(field,value)}"' |
| 82 | + |
| 83 | + |
| 84 | +@render_manager.register |
| 85 | +classAnomaliQueryRender(PlatformQueryRender): |
| 86 | +details:PlatformDetails=anomali_query_details |
| 87 | +mappings:AnomaliMappings=anomali_query_mappings |
| 88 | + |
| 89 | +or_token="OR" |
| 90 | +and_token="AND" |
| 91 | +not_token="NOT" |
| 92 | + |
| 93 | +comment_symbol="--" |
| 94 | +is_single_line_comment=True |
| 95 | + |
| 96 | +field_value_render=AnomaliFieldValueRender(or_token=or_token) |
| 97 | + |
| 98 | +@staticmethod |
| 99 | +def_finalize_search_query(query:str)->str: |
| 100 | +returnf"| where{query}"ifqueryelse"" |