acl/lib_acl_cpp/samples/scan_rmdir/main.cpp

#include <unistd.h>
#include "acl_cpp/lib_acl.hpp"

class myscan_dir : public acl::scan_dir
{
public:
	myscan_dir(void) {}
	~myscan_dir(void) {}

	bool rmdir_callback(const char* path) {
		if (::rmdir(path) == 0) {
			printf(">> %s: rmdir ok, path=%s\r\n",
				__FUNCTION__, path);
			return true;
		} else {
			printf(">> %s: rmdir error=%s, path=%s\r\n",
				__FUNCTION__, acl::last_serror(), path);
			return false;
		}
	}
};

static void scan_rmdir(acl::scan_dir& scan)
{
	int ndirs = 0, nfiles = 0;
	const char* name;
	bool is_file;

	while ((name = scan.next(true, &is_file)) != NULL) {
		printf("[%s: %s, path: %s]\r\n",
			is_file ? "filename" : "dirname",
			name, scan.curr_path());
		if (is_file) {
			nfiles++;
		} else {
			ndirs++;
		}

		if (nfiles >= 50 || ndirs >= 50) {
			break;
		}
	}
}

static bool rmdir_on_empty(const char* path, bool recursive, bool rmdir_on)
{
	myscan_dir scan; 

	if (scan.open(path, recursive, rmdir_on) == false) {
		printf("open path: %s error: %s\r\n", path, acl::last_serror());
		return false;
	}


	scan_rmdir(scan);
	return true;
}

static int rmdir_fn(ACL_SCAN_DIR* scan, const char* path, void* ctx)
{
	acl::scan_dir* me = (acl::scan_dir*) ctx;
	assert(scan == me->get_scan_dir());

	if (::rmdir(path) == -1) {
		printf(">> %s: rmdir error=%s, path=%s\r\n",
			__FUNCTION__, acl::last_serror(), path);
		return -1;
	} else {
		printf(">> %s: rmdir ok, path=%s\r\n", __FUNCTION__, path);
		return 0;
	}
}

static bool rmdir_on_empty2(const char* path, bool recursive, bool rmdir_on)
{
	acl::scan_dir scan;

	if (scan.open(path, recursive, rmdir_on) == false) {
		printf("open path: %s error: %s\r\n", path, acl::last_serror());
		return false;
	}

	if (rmdir_on) {
		scan.set_rmdir_callback(rmdir_fn, &scan);
	}

	scan_rmdir(scan);
	return true;
}

static void usage(const char* procname)
{
	printf("usage: %s -h [help]\r\n"
		" -d path\r\n"
		" -R [rmdir_on]\r\n"
		" -r [if recursive, default: false]\r\n"
		" -C [using myscan_dir C++ class]\r\n"
		" -a [if get fullpath, default: false]\r\n", procname);
}

int main(int argc, char* argv[])
{
	int   ch;
	bool  rmdir_on = false, recursive = false, use_class = false;
	acl::string path;
	acl::log::stdout_open(true);

	while ((ch = getopt(argc, argv, "hd:raRC")) > 0) {
		switch (ch) {
		case 'h':
			usage(argv[0]);
			return 0;
		case 'd':
			path = optarg;
			break;
		case 'r':
			recursive = true;
			break;
		case 'R':
			rmdir_on = true;
			break;
		case 'C':
			use_class = true;
			break;
		default:
			break;
		}
	}

	if (path.empty()) {
		usage(argv[0]);
		return 1;
	}

	if (!path.end_with("/")) {
		path += "/";
	}

	if (use_class) {
		rmdir_on_empty(path, recursive, rmdir_on);
	} else {
		rmdir_on_empty2(path, recursive, rmdir_on);
	}

#ifdef	WIN32
	printf("========================================================\r\n");
	printf("enter any key to exit!\r\n");
	getchar();
#endif

	return (0);
}