Switch to ahochorasick-rs

requirements.txt

@@ -1 +1,2 @@
 tqdm
+ahocorasick-rs

setup.py

@@ -11,79 +11,26 @@
 from sys import version_info as python_version
 
 if python_version.major not in [3]:
-    raise ValueError('Python %s is not supported' % python_version)
+    raise ValueError("Python %s is not supported" % python_version)
 
 ## Reading long_description
 this_directory = path.abspath(path.dirname(__file__))
-with open(path.join(this_directory, 'README.md'), encoding='utf-8') as f:
+with open(path.join(this_directory, "README.md"), encoding="utf-8") as f:
     long_description = f.read()
 
-## Install pyahocorasick
-pyahocorasick_dir = "./stringcheese/pyahocorasick-1.4.0/"
-
-module = setuptools.Extension(
-    'stringcheese.ahocorasick',
-    sources=[
-        pyahocorasick_dir+'pyahocorasick.c',
-    ],
-    define_macros=[], #[('AHOCORASICK_UNICODE', '')],
-    depends=[pyahocorasick_dir+d for d in [
-        'common.h',
-        'Automaton.c',
-        'Automaton.h',
-        'Automaton_pickle.c',
-        'AutomatonItemsIter.c',
-        'AutomatonItemsIter.h',
-        'AutomatonSearchIter.c',
-        'AutomatonSearchIter.h',
-        'AutomatonSearchIterLong.c',
-        'AutomatonSearchIterLong.h',
-        'trie.c',
-        'trie.h',
-        'slist.c',
-        'utils.c',
-        'trienode.c',
-        'trienode.h',
-        'msinttypes/stdint.h',
-        'src/inline_doc.h',
-        'src/pickle/pickle.h',
-        'src/pickle/pickle_data.h',
-        'src/pickle/pickle_data.c',
-        'src/custompickle/custompickle.h',
-        'src/custompickle/custompickle.c',
-        'src/custompickle/pyhelpers.h',
-        'src/custompickle/pyhelpers.c',
-        'src/custompickle/save/automaton_save.h',
-        'src/custompickle/save/automaton_save.c',
-        'src/custompickle/save/savebuffer.h',
-        'src/custompickle/save/savebuffer.c',
-        'src/custompickle/load/module_automaton_load.h',
-        'src/custompickle/load/module_automaton_load.c',
-        'src/custompickle/load/loadbuffer.h',
-        'src/custompickle/load/loadbuffer.c',
-        'src/pycallfault/pycallfault.h',
-        'src/pycallfault/pycallfault.c',
-    ]],
-)
-
 setuptools.setup(
-    name         = 'stringcheese',
-    version      = '2.0',
-    description  = 'StringCheese is a tool to get easy CTF flags automatically.',
-    long_description = long_description,
-    long_description_content_type = 'text/markdown',
-    url          = 'https://github.com/MathisHammel/stringcheese',
-    author       = 'MathisHammel',
-    author_email = 'mathis@h25.io',
-    license      = 'GPL2',
-    ext_modules  = [module],
-    packages     = ['stringcheese'],
+    name="stringcheese",
+    version="2.1",
+    description="StringCheese is a tool to get easy CTF flags automatically.",
+    long_description=long_description,
+    long_description_content_type="text/markdown",
+    url="https://github.com/MathisHammel/stringcheese",
+    author="MathisHammel",
+    author_email="mathis@h25.io",
+    license="GPL2",
+    packages=["stringcheese"],
     include_package_data=True,
-    entry_points = {
-        'console_scripts' : ['stringcheese=stringcheese.stringcheese:main']
-    },
-    install_requires = [
-        'tqdm'
-    ],
-    zip_safe     = False
+    entry_points={"console_scripts": ["stringcheese=stringcheese.stringcheese:main"]},
+    install_requires=["tqdm", "ahocorasick-rs"],
+    zip_safe=False,
 )