X-Git-Url: http://git.annexia.org/?p=virt-mem.git;a=blobdiff_plain;f=extract%2Ffedora-koji%2Fkoji.ml;fp=extract%2Ffedora-koji%2Fkoji.ml;h=aa24483df07e065c672e954af76ac61a8eb6ab38;hp=0000000000000000000000000000000000000000;hb=9392e738c4c65284145e50644d463fc0aa05bbad;hpb=03dcc62ed37ff0160f211927196e48033a6aeb81

diff --git a/extract/fedora-koji/koji.ml b/extract/fedora-koji/koji.ml
new file mode 100644
index 0000000..aa24483
--- /dev/null
+++ b/extract/fedora-koji/koji.ml
@@ -0,0 +1,154 @@
+(* Memory info for virtual domains.
+   (C) Copyright 2008 Richard W.M. Jones, Red Hat Inc.
+   http://libvirt.org/
+
+   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., 675 Mass Ave, Cambridge, MA 02139, USA.
+*)
+
+open Printf
+
+type build = {
+  package_name : string;		(* eg. "kernel" *)
+  version : string;			(* eg. "2.6.25" *)
+  release : string;			(* eg. "1.fc8" *)
+  build_id : int;
+}
+
+let string_of_build { package_name = package_name;
+		      version = version; release = release;
+		      build_id = build_id } =
+  sprintf "%d: %s %s %s" build_id package_name version release
+
+type rpm = {
+  rpm_id : int;			        (* RPM ID (for downloading, etc.) *)
+  rpm_build : build;
+  rpm_name : string;			(* eg. "kernel" *)
+  rpm_version : string;			(* eg. "2.6.25" *)
+  rpm_release : string;			(* eg. "1.fc8" *)
+  rpm_size : int;			(* size in bytes of the RPM. *)
+  rpm_arch : string;			(* architecture *)
+}
+
+let string_of_rpm { rpm_id = id; rpm_build = { build_id = build_id };
+		    rpm_name = name;
+		    rpm_version = version; rpm_release = release;
+		    rpm_size = size; rpm_arch = arch } =
+  sprintf "%d: (build %d) %s %s %s (%d bytes) %s"
+    id build_id name version release size arch
+
+type rpc = < call : string -> XmlRpc.value list -> XmlRpc.value >
+
+let get_string_from_struct name items =
+  match List.assoc name items with
+  | `String str -> str
+  | _ -> invalid_arg (name ^ ": expected string type")
+
+let get_int_from_struct name items =
+  match List.assoc name items with
+  | `Int i -> i
+  | _ -> invalid_arg (name ^ ": expected int type")
+
+let list_builds rpc ~prefix =
+  let builds = rpc#call "listBuilds" [
+    `Struct [
+      (* __starstar is some wierd Python thing which is needed for
+       * Python optional arguments to work.
+       *)
+      "__starstar", `Int 1;
+      "prefix", `String prefix;
+    ]
+  ] in
+
+  match builds with
+  | `Array builds ->
+      List.map (
+	function
+	| `Struct items ->
+	    (try
+	       let package_name = get_string_from_struct "package_name" items in
+	       let version = get_string_from_struct "version" items in
+	       let release = get_string_from_struct "release" items in
+	       let build_id = get_int_from_struct "build_id" items in
+	       { package_name = package_name;
+		 version = version; release = release;
+		 build_id = build_id }
+	     with
+	     | Not_found ->
+		 prerr_endline "missing element in build structure from koji listBuilds() calls";
+		 exit 1
+	     | Invalid_argument err ->
+		 prerr_endline err;
+		 exit 1
+	    )
+	| t ->
+	    prerr_endline "unexpected type from koji listBuilds() call";
+	    prerr_endline (XmlRpc.dump t);
+	    exit 1
+      ) builds
+  | t ->
+      prerr_endline "unexpected type from koji listBuilds() call:";
+      prerr_endline (XmlRpc.dump t);
+      exit 1
+
+let list_build_rpms rpc ({ build_id = build_id } as build) =
+  let rpms = rpc#call "listBuildRPMs" [ `Int build_id ] in
+
+  match rpms with
+  | `Array rpms ->
+      List.map (
+	function
+	| `Struct items ->
+	    (try
+	       let name = get_string_from_struct "name" items in
+	       let version = get_string_from_struct "version" items in
+	       let release = get_string_from_struct "release" items in
+	       let build_id' = get_int_from_struct "build_id" items in
+	       let id = get_int_from_struct "id" items in
+	       let size = get_int_from_struct "size" items in
+	       let arch = get_string_from_struct "arch" items in
+	       assert (build_id = build_id');
+	       { rpm_name = name; rpm_version = version; rpm_release = release;
+		 rpm_build = build; rpm_id = id; rpm_size = size;
+		 rpm_arch = arch }
+	     with
+	     | Not_found ->
+		 prerr_endline "missing element in build structure from koji listBuildRPMs() calls";
+		 exit 1
+	     | Invalid_argument err ->
+		 prerr_endline err;
+		 exit 1
+	    )
+	| t ->
+	    prerr_endline "unexpected type from koji listBuildRPMs() call";
+	    prerr_endline (XmlRpc.dump t);
+	    exit 1
+      ) rpms
+  | t ->
+      prerr_endline "unexpected type from koji listBuildRPMs() call:";
+      prerr_endline (XmlRpc.dump t);
+      exit 1
+
+(* This gets the RPM download URL for an RPM.  I can't see a way to
+ * get this using the Koji API, but the URLs are fairly predictable
+ * anyway.
+ *)
+let rpm_download_url { rpm_build = { package_name = build_name };
+		       rpm_name = rpm_name;
+		       rpm_version = version; rpm_release = release;
+		       rpm_arch = arch } =
+  let filename = sprintf "%s-%s-%s.%s.rpm" rpm_name version release arch in
+  let uri = sprintf "http://koji.fedoraproject.org/packages/%s/%s/%s/%s/%s"
+    build_name version release arch filename in
+  uri, filename