feat: added some extra functions. Did not mean to edit on master, but this is a class that is not used by anyone else.

Author: edkoch

include/vb_util_lib/file_spec.h

@@ -23,11 +23,17 @@ class FileSpec
 	void setType(FileType type);
 	Directory getDir() const;
 	void setDir(Directory dir);
+	void setPath(std::string& base, std::string param, std::string env_name);
+	void getPath(std::string& full_name);
+	std::string getPrefix() {
+		return prefix_;
+	}
 
 private:
 	Directory dir_;
 	FileType type_;
 	std::string name_;
+	std::string prefix_ {""};
 	std::string md5_;
 	boost::posix_time::ptime date_;
 	int size_;

src/vb_util_lib/file_spec.cpp

@@ -1,4 +1,5 @@
 #include <vb_util_lib/file_spec.h>
+#include <cstdlib>
 
 const boost::posix_time::ptime& FileSpec::getDate() const {
 	return date_;
@@ -47,3 +48,61 @@ FileSpec::Directory FileSpec::getDir() const {
 void FileSpec::setDir(FileSpec::Directory dir) {
 	dir_ = dir;
 }
+
+void FileSpec::setPath(std::string& base, std::string param, std::string env_name) {
+	char sep = '/';
+	std::string base_dir = "";
+	prefix_ = "";
+	name_ = "";
+
+	size_t i = base.rfind(sep, base.length());
+	if (i != std::string::npos)
+	{
+		name_ = base.substr(i+1, base.length() - i);
+		base_dir = base.substr(0, i+1);
+	}
+	else
+	{
+		name_= base;
+		base_dir = "";
+	}
+
+	if (base_dir.length() != 0 && base_dir.at(0) == sep)
+	{
+		// Absolute path name
+		prefix_ = base_dir;
+		return;
+	}
+
+	// Append the separator to the base_dir
+	if (base_dir.length() > 0 && base_dir.at(base_dir.length()-1) != sep)
+	{
+		base_dir += "/";
+	}
+
+	if (param.length() > 0)
+	{
+		// use the parameter as the directory name.
+		prefix_ = param;
+	} else if (env_name.length() > 0)
+	{
+		// use the environment variable
+	    if(const char* env_p = std::getenv(env_name.c_str()))
+	    {
+	    	prefix_ = env_p;
+	    }
+	}
+
+	// Append the separator to the prefix
+	if (prefix_.length() > 0 && prefix_.at(prefix_.length()-1) != sep)
+	{
+		prefix_ += "/";
+	}
+	prefix_ += base_dir;
+
+	return;
+}
+
+void FileSpec::getPath(std::string& full_name) {
+	full_name = prefix_ + name_;
+}