Merge pull request conan-io#5 from uilianries/feature/size-limit

Check recipe folder size

Author: lasote

hooks/conan-center.py

@@ -139,6 +139,20 @@ def test(out):
                 if all([char in line for char in ("@", "[", "]")]):
                     out.error("Possible use of version ranges, line %s:\n %s" % (num, line))
 
+    @run_test("RECIPE FOLDER SIZE", output)
+    def test(out):
+        max_folder_size = int(os.getenv("CONAN_MAX_RECIPE_FOLDER_SIZE_KB", 256))
+        dir_path = os.path.dirname(conanfile_path)
+        total_size = 0
+        for path, dirs, files in os.walk(dir_path):
+            for files_it in files:
+                file_path = os.path.join(path, files_it)
+                total_size += os.path.getsize(file_path)
+        total_size_kb = total_size / 1024
+        out.info("total_size_kb: %sKB" % total_size_kb)
+        if total_size_kb > max_folder_size:
+            out.error("The size of your recipe folder ({}KB) is larger than the maximum allowed"
+                      " size ({}KB).".format(total_size_kb, max_folder_size))
 
 @raise_if_error_output
 def pre_source(output, conanfile, conanfile_path, **kwargs):

tests/test_hooks/test_conan-center.py

@@ -25,7 +25,7 @@ class AConan(ConanFile):
     def _get_environ(self, **kwargs):
         kwargs = super(ConanCenterTests, self)._get_environ(**kwargs)
         kwargs.update({'CONAN_HOOKS': os.path.join(os.path.dirname(__file__), '..', '..', 'hooks',
-                                                   'conan-center_reviewer')})
+                                                   'conan-center')})
         return kwargs
 
     def test_conanfile(self):
@@ -40,7 +40,7 @@ def test_conanfile(self):
         self.assertIn("[LIBCXX] OK", output)
         self.assertIn("[MATCHING CONFIGURATION] OK", output)
         self.assertIn("[SHARED ARTIFACTS] OK", output)
-        self.assertIn("ERROR: [PACKAGE LICENSE] No package licenses found", output)
+        self.assertIn("ERROR: [PACKAGE LICENSE] No 'licenses' folder found in package", output)
         self.assertIn("[DEFAULT PACKAGE LAYOUT] OK", output)
         self.assertIn("[SHARED ARTIFACTS] OK", output)
         print(output)
@@ -57,7 +57,7 @@ def test_conanfile_header_only(self):
         self.assertIn("[LIBCXX] OK", output)
         self.assertIn("[MATCHING CONFIGURATION] OK", output)
         self.assertIn("[SHARED ARTIFACTS] OK", output)
-        self.assertIn("ERROR: [PACKAGE LICENSE] No package licenses found", output)
+        self.assertIn("ERROR: [PACKAGE LICENSE] No 'licenses' folder found in package", output)
         self.assertIn("[DEFAULT PACKAGE LAYOUT] OK", output)
         self.assertIn("[SHARED ARTIFACTS] OK", output)
 
@@ -74,6 +74,26 @@ def test_conanfile_installer(self):
         self.assertIn("ERROR: [MATCHING CONFIGURATION] Built artifacts does not match the settings",
                       output)
         self.assertIn("[SHARED ARTIFACTS] OK", output)
-        self.assertIn("ERROR: [PACKAGE LICENSE] No package licenses found", output)
+        self.assertIn("ERROR: [PACKAGE LICENSE] No 'licenses' folder found in package", output)
         self.assertIn("[DEFAULT PACKAGE LAYOUT] OK", output)
         self.assertIn("[SHARED ARTIFACTS] OK", output)
+
+    def test_regular_folder_size(self):
+        tools.save('conanfile.py', content=self.conanfile_installer)
+        output = self.conan(['export', '.', 'name/version@user/channel'])
+        self.assertIn("[RECIPE FOLDER SIZE] OK", output)
+        self.assertNotIn("ERROR: [RECIPE FOLDER SIZE]", output)
+
+    def test_larger_folder_size(self):
+        content = " ".join(["test_recipe_folder_larger_size" for it in range(1048576)])
+        tools.save('conanfile.py', content=self.conanfile_installer)
+        tools.save('big_file', content=content)
+        output = self.conan(['export', '.', 'name/version@user/channel'])
+        self.assertIn("ERROR: [RECIPE FOLDER SIZE] The size of your recipe folder", output)
+
+    def test_custom_folder_size(self):
+        with tools.environment_append({"CONAN_MAX_RECIPE_FOLDER_SIZE_KB": "0"}):
+            content = " ".join(["test_recipe_folder_larger_size" for it in range(1048576)])
+            tools.save('conanfile.py', content=self.conanfile_installer)
+            output = self.conan(['export', '.', 'name/version@user/channel'])
+            self.assertIn("ERROR: [RECIPE FOLDER SIZE] The size of your recipe folder", output)