[Libguestfs] [PATCH] sysprep: remove the cache of package manager

[Wanlong Gao]

Remove the cache of package manager like yum and apt.

Signed-off-by: Wanlong Gao <gaowanlong at cn.fujitsu.com>
---
 sysprep/Makefile.am                                |    2 +
 sysprep/sysprep_operation_package_manager_cache.ml |   49 ++++++++++++++++++++
 2 files changed, 51 insertions(+)
 create mode 100644 sysprep/sysprep_operation_package_manager_cache.ml

diff --git a/sysprep/Makefile.am b/sysprep/Makefile.am
index e4b7f7b..78b1fa1 100644
--- a/sysprep/Makefile.am
+++ b/sysprep/Makefile.am
@@ -42,6 +42,7 @@ SOURCES = \
 	sysprep_operation_logfiles.ml \
 	sysprep_operation_mail_spool.ml \
 	sysprep_operation_net_hwaddr.ml \
+	sysprep_operation_package_manager_cache.ml \
 	sysprep_operation_random_seed.ml \
 	sysprep_operation_rhn_systemid.ml \
 	sysprep_operation_script.ml \
@@ -69,6 +70,7 @@ OBJECTS = \
 	sysprep_operation_logfiles.cmx \
 	sysprep_operation_mail_spool.cmx \
 	sysprep_operation_net_hwaddr.cmx \
+	sysprep_operation_package_manager_cache.ml \
 	sysprep_operation_random_seed.cmx \
 	sysprep_operation_rhn_systemid.cmx \
 	sysprep_operation_script.cmx \
diff --git a/sysprep/sysprep_operation_package_manager_cache.ml b/sysprep/sysprep_operation_package_manager_cache.ml
new file mode 100644
index 0000000..f70f310
--- /dev/null
+++ b/sysprep/sysprep_operation_package_manager_cache.ml
@@ -0,0 +1,49 @@
+(* virt-sysprep
+ * Copyright (C) 2012 FUJITSU LIMITED
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * 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 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 Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ *)
+
+open Sysprep_operation
+
+module G = Guestfs
+
+let package_manager_cache_perform g root =
+  let packager = g#inspect_get_package_management root in
+    match packager with
+    | "yum" ->
+      let cache = g#glob_expand "/var/cache/yum/*" in
+      Array.iter (
+        fun dir ->
+          g#rm_rf dir
+      ) cache; []
+    | "apt" ->
+      let cache = g#glob_expand "/var/cache/apt/archives/*" in
+      Array.iter (
+        fun dir ->
+          g#rm_rf dir
+      ) cache; []
+    | _ -> (); []
+
+let package_manager_cache_op = {
+  name = "package-manager-cache";
+  enabled_by_default = true;
+  heading = "Remove the cache of package manager";
+  pod_description = None;
+  extra_args = [];
+  perform = package_manager_cache_perform;
+}
+
+let () = register_operation package_manager_cache_op
-- 
1.7.10