From: Stefan Schantl <stefan.schantl@ipfire.org>
Date: Thu, 20 Aug 2020 17:28:54 +0000 (+0200)
Subject: perl: Add database_countries() function.
X-Git-Tag: 0.9.4~20
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=c1590ddd718fb3659ea7a7ac07549cb5b08dba93;p=location%2Flibloc.git

perl: Add database_countries() function.

This function is used to the stored countries of a database, which
easily can be assigned to an array.

Signed-off-by: Stefan Schantl <stefan.schantl@ipfire.org>
Signed-off-by: Michael Tremer <michael.tremer@ipfire.org>
---

diff --git a/src/perl/Location.xs b/src/perl/Location.xs
index 5693744..3c347db 100644
--- a/src/perl/Location.xs
+++ b/src/perl/Location.xs
@@ -6,11 +6,10 @@
 #include <stdio.h>
 #include <string.h>
 
-
 #include <loc/libloc.h>
 #include <loc/database.h>
 #include <loc/network.h>
-
+#include <loc/country.h>
 
 MODULE = Location		PACKAGE = Location
 
@@ -119,6 +118,45 @@ get_license(db)
 	OUTPUT:
 		RETVAL
 
+void
+database_countries(db)
+	struct loc_database* db;
+
+	PPCODE:
+		// Create Database enumerator
+		struct loc_database_enumerator* enumerator;
+		int err = loc_database_enumerator_new(&enumerator, db, LOC_DB_ENUMERATE_COUNTRIES);
+
+		if (err) {
+			croak("Could not create a database enumerator\n");
+		}
+
+		// Init and enumerate first country.
+		struct loc_country* country;
+		err = loc_database_enumerator_next_country(enumerator, &country);
+		if (err) {
+			croak("Could not enumerate next country\n");
+		}
+
+		while (country) {
+			// Extract the country code.
+			const char* ccode = loc_country_get_code(country);
+
+			// Push country code.
+			XPUSHs(sv_2mortal(newSVpv(ccode, 2)));
+
+			// Unref country pointer.
+			loc_country_unref(country);
+
+			// Enumerate next item.
+			err = loc_database_enumerator_next_country(enumerator, &country);
+			if (err) {
+				croak("Could not enumerate next country\n");
+			}
+		}
+
+		loc_database_enumerator_unref(enumerator);
+
 #
 # Lookup functions
 #