X-Git-Url: http://git.argeo.org/?a=blobdiff_plain;f=org.argeo.slc.rpmfactory%2Fsrc%2Forg%2Fargeo%2Fslc%2Frpmfactory%2Fcore%2FYumListParser.java;fp=org.argeo.slc.rpmfactory%2Fsrc%2Forg%2Fargeo%2Fslc%2Frpmfactory%2Fcore%2FYumListParser.java;h=9327fedb75275a6636e3f5b6ecdc5c6d7f2aac89;hb=825d60c5348dbe3f5be25b0bccf7bdebfe694219;hp=0000000000000000000000000000000000000000;hpb=5e991fff5cba01858dcc5747a27e637325bc5c8e;p=gpl%2Fargeo-jcr.git

diff --git a/org.argeo.slc.rpmfactory/src/org/argeo/slc/rpmfactory/core/YumListParser.java b/org.argeo.slc.rpmfactory/src/org/argeo/slc/rpmfactory/core/YumListParser.java
new file mode 100644
index 0000000..9327fed
--- /dev/null
+++ b/org.argeo.slc.rpmfactory/src/org/argeo/slc/rpmfactory/core/YumListParser.java
@@ -0,0 +1,93 @@
+package org.argeo.slc.rpmfactory.core;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.util.Set;
+import java.util.StringTokenizer;
+import java.util.TreeSet;
+
+import org.apache.commons.io.FilenameUtils;
+import org.apache.commons.io.IOUtils;
+import org.apache.commons.io.LineIterator;
+import org.argeo.api.cms.CmsLog;
+import org.argeo.slc.SlcException;
+
+/**
+ * Reads the output of a 'yum list all' command and interpret the list of
+ * packages.
+ */
+public class YumListParser implements RpmPackageSet {
+	private final static CmsLog log = CmsLog.getLog(YumListParser.class);
+
+	private Set<String> installed = new TreeSet<String>();
+	/** Not installed but available */
+	private Set<String> installable = new TreeSet<String>();
+
+	private Path yumListOutput;
+
+	public void init() {
+		if (yumListOutput != null) {
+			try (InputStream in = Files.newInputStream(yumListOutput)) {
+				load(in);
+				if (log.isDebugEnabled())
+					log.debug(installed.size() + " installed, " + installable.size() + " installable, from "
+							+ yumListOutput);
+			} catch (IOException e) {
+				throw new SlcException("Cannot initialize yum list parser", e);
+			}
+		}
+	}
+
+	public Boolean contains(String packageName) {
+		if (installed.contains(packageName))
+			return true;
+		else
+			return installable.contains(packageName);
+	}
+
+	protected void load(InputStream in) throws IOException {
+		Boolean readingInstalled = false;
+		Boolean readingAvailable = false;
+		LineIterator it = IOUtils.lineIterator(in, "UTF-8");
+		while (it.hasNext()) {
+			String line = it.nextLine();
+			if (line.trim().equals("Installed Packages")) {
+				readingInstalled = true;
+			} else if (line.trim().equals("Available Packages")) {
+				readingAvailable = true;
+				readingInstalled = false;
+			} else if (readingAvailable) {
+				if (Character.isLetterOrDigit(line.charAt(0))) {
+					installable.add(extractRpmName(line));
+				}
+			} else if (readingInstalled) {
+				if (Character.isLetterOrDigit(line.charAt(0))) {
+					installed.add(extractRpmName(line));
+				}
+			}
+		}
+	}
+
+	protected String extractRpmName(String line) {
+		StringTokenizer st = new StringTokenizer(line, " \t");
+		String packageName = st.nextToken();
+		// consider the arch as an extension
+		return FilenameUtils.getBaseName(packageName);
+		// return packageName.split("\\.")[0];
+	}
+
+	public Set<String> getInstalled() {
+		return installed;
+	}
+
+	public Set<String> getInstallable() {
+		return installable;
+	}
+
+	public void setYumListOutput(Path yumListOutput) {
+		this.yumListOutput = yumListOutput;
+	}
+
+}