Deduplicate code and make some small improvements

Signed-off-by: Mattias Andrée <maandree@kth.se>

1 files changed, 6 insertions, 85 deletions

diff --git a/get-contact-sites.c b/get-contact-sites.c
index 4bd53e6..ae1356a 100644
--- a/get-contact-sites.c
+++ b/get-contact-sites.c
@@ -1,88 +1,9 @@
 /* See LICENSE file for copyright and license details. */
-#include "common.h"
-
-USAGE("[-a address] [-c context] [-AC] contact-id ...");
-
-
-int
-main(int argc, char *argv[])
-{
-	int display_ctx = 0, display_addr = 0;
-	const char *lookup_ctx = NULL, *lookup_addr = NULL;
-	struct passwd *user;
-	struct libcontacts_contact contact;
-	struct libcontacts_site **sites, *site;
-	int ret = 0;
-	size_t i;
-
-	ARGBEGIN {
-	case 'a':
-		if (lookup_addr)
-			usage();
-		lookup_addr = ARG();
-		break;
-	case 'c':
-		if (lookup_ctx)
-			usage();
-		lookup_ctx = ARG();
-		break;
-	case 'A':
-		display_addr = 1;
-		break;
-	case 'C':
-		display_ctx = 1;
-		break;
-	default:
-		usage();
-	} ARGEND;
+#define CATEGORY sites
 
-	if (!argc)
-		usage();
+#define LIST_PARAMS(X)\
+	X('C', "C", 'c', "c", context, "context")\
+	X('A', "A", 'a', "a", address, "address")
 
-	if (lookup_ctx && !lookup_addr && !display_ctx && !display_addr)
-		display_addr = 1;
-	if (lookup_addr && !lookup_ctx && !display_ctx && !display_addr)
-		display_ctx = 1;
-
-	for (i = 0; argv[i]; i++)
-		if (!*argv[i] || strchr(argv[i], '/'))
-			usage();
-
-	errno = 0;
-	user = getpwuid(getuid());
-	if (!user)
-		eprintf("getpwuid: %s\n", errno ? strerror(errno) : "user does not exist");
-
-	for (; *argv; argv++) {
-		if (libcontacts_load_contact(*argv, &contact, user)) {
-			weprintf("libcontacts_load_contact %s: %s\n", *argv, errno ? strerror(errno) : "contact file is malformatted");
-			ret = 1;
-			continue;
-		}
-		if ((sites = contact.sites)) {
-			for (; (site = *sites); sites++) {
-				if (lookup_ctx && strcmpnul(site->context, lookup_ctx))
-					continue;
-				if (lookup_addr && strcmpnul(site->address, lookup_addr))
-					continue;
-				if (lookup_ctx && lookup_addr && !display_ctx && !display_addr) {
-					printf("%s\n", *argv);
-					continue;
-				}
-				if (argc > 1)
-					printf("%s: ", *argv);
-				if (!display_ctx == !display_addr)
-					printf("%s: %s\n", site->context, site->address);
-				else if (display_addr)
-					printf("%s\n", site->address);
-				else
-					printf("%s\n", site->context);
-			}
-		}
-		libcontacts_contact_destroy(&contact);
-	}
-
-	if (fflush(stdout) || ferror(stdout) || fclose(stdout))
-		eprintf("printf:");
-	return ret;
-}
+#include "common.h"
+IMPLEMENT_GET_ON_LIST(site)