godot/core/os/dir_access.cpp

/*************************************************************************/
/*  dir_access.cpp                                                       */
/*************************************************************************/
/*                       This file is part of:                           */
/*                           GODOT ENGINE                                */
/*                    http://www.godotengine.org                         */
/*************************************************************************/
/* Copyright (c) 2007-2015 Juan Linietsky, Ariel Manzur.                 */
/*                                                                       */
/* Permission is hereby granted, free of charge, to any person obtaining */
/* a copy of this software and associated documentation files (the       */
/* "Software"), to deal in the Software without restriction, including   */
/* without limitation the rights to use, copy, modify, merge, publish,   */
/* distribute, sublicense, and/or sell copies of the Software, and to    */
/* permit persons to whom the Software is furnished to do so, subject to */
/* the following conditions:                                             */
/*                                                                       */
/* The above copyright notice and this permission notice shall be        */
/* included in all copies or substantial portions of the Software.       */
/*                                                                       */
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,       */
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF    */
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY  */
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,  */
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE     */
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.                */
/*************************************************************************/
#include "dir_access.h"
#include "os/file_access.h"
#include "os/memory.h"
#include "os/os.h"
#include "globals.h"


String DirAccess::_get_root_path() const {

	switch(_access_type) {

		case ACCESS_RESOURCES: return Globals::get_singleton()->get_resource_path();
		case ACCESS_USERDATA: return OS::get_singleton()->get_data_dir();
		default: return "";
	}

	return "";
}
String DirAccess::_get_root_string() const {

	switch(_access_type) {

		case ACCESS_RESOURCES: return "res://";
		case ACCESS_USERDATA: return "user://";
		default: return "";
	}

	return "";
}

int DirAccess::get_current_drive() {

	String path = get_current_dir().to_lower();
	for(int i=0;i<get_drive_count();i++) {
		String d = get_drive(i).to_lower();
		if (path.begins_with(d))
			return i;
	}

	return 0;
}

static Error _erase_recursive(DirAccess *da) {

	List<String> dirs;
	List<String> files;

	da->list_dir_begin();
	String n = da->get_next();
	while(n!=String()) {

		if (n!="." && n!="..") {

			if (da->current_is_dir())
				dirs.push_back(n);
			else
				files.push_back(n);
		}

		n=da->get_next();
	}

	da->list_dir_end();

	for(List<String>::Element *E=dirs.front();E;E=E->next()) {

		Error err = da->change_dir(E->get());
		if (err==OK) {

			err = _erase_recursive(da);
			if (err) {
				print_line("err recurso "+E->get());
				return err;
			}
			err = da->change_dir("..");
			if (err) {
				print_line("no go back "+E->get());
				return err;
			}
			err = da->remove(da->get_current_dir().plus_file(E->get()));
			if (err) {
				print_line("no remove dir"+E->get());
				return err;
			}
		} else {
			print_line("no change to "+E->get());
			return err;
		}
	}

	for(List<String>::Element *E=files.front();E;E=E->next()) {

		Error err = da->remove(da->get_current_dir().plus_file(E->get()));
		if (err) {

			print_line("no remove file"+E->get());
			return err;
		}
	}

	return OK;
}


Error DirAccess::erase_contents_recursive() {

	return _erase_recursive(this);
}


Error DirAccess::make_dir_recursive(String p_dir) {

	if (p_dir.length() < 1) {
		return OK;
	};

	String full_dir;
	Globals* g = Globals::get_singleton();

	if (!p_dir.is_abs_path()) {
		//append current

		String cur = normalize_path(g->globalize_path(get_current_dir()));
		if (cur[cur.length()-1] != '/') {
			cur = cur + "/";
		};

		full_dir=(cur+"/"+p_dir).simplify_path();
	} else {
		//validate and use given
		String dir = normalize_path(g->globalize_path(p_dir));
		if (dir.length() < 1) {
			return OK;
		};
		if (dir[dir.length()-1] != '/') {
			dir = dir + "/";
		};
		full_dir=dir;
	}

	//int slices = full_dir.get_slice_count("/");

	int pos = 0;
	while (pos < full_dir.length()) {

		int n = full_dir.find("/", pos);
		if (n < 0) {
			n = full_dir.length();
		};
		pos = n + 1;

		if (pos > 1) {
			String to_create = full_dir.substr(0, pos -1);
			//print_line("MKDIR: "+to_create);
			Error err = make_dir(to_create);
			if (err != OK && err != ERR_ALREADY_EXISTS) {

				ERR_FAIL_V(err);
			};
		};
	};

	return OK;
};


String DirAccess::normalize_path(const String &p_path) {

	static const int max_depth = 64;
	int pos_stack[max_depth];
	int curr = 0;

	int pos = 0;
	String cur_dir;

	do {

		if (curr >= max_depth) {

			ERR_PRINT("Directory depth too deep.");
			return "";
		};

		int start = pos;

		int next = p_path.find("/", pos);
		if (next < 0) {
			next = p_path.length() - 1;
		};

		pos = next + 1;

		cur_dir = p_path.substr(start, next - start);

		if (cur_dir == "" || cur_dir == ".") {
			continue;
		};
		if (cur_dir == "..") {

			if (curr > 0) { // pop a dir
				curr -= 2;
			};
			continue;
		};

		pos_stack[curr++] = start;
		pos_stack[curr++] = next;

	} while (pos < p_path.length());

	String path;
	if (p_path[0] == '/') {
		path = "/";
	};

	int i=0;
	while (i < curr) {

		int start = pos_stack[i++];

		while ( ((i+1)<curr) && (pos_stack[i] == pos_stack[i+1]) ) {

			++i;
		};
		path = path + p_path.substr(start, (pos_stack[i++] - start) + 1);
	};

	return path;
};

String DirAccess::get_next(bool* p_is_dir) {

	String next=get_next();
	if (p_is_dir)
		*p_is_dir=current_is_dir();
	return next;
}

String DirAccess::fix_path(String p_path) const {

	switch(_access_type) {

		case ACCESS_RESOURCES: {

			if (Globals::get_singleton()) {
				if (p_path.begins_with("res://")) {

					String resource_path = Globals::get_singleton()->get_resource_path();
					if (resource_path != "") {

						return p_path.replace("res:/",resource_path);
					};
					return p_path.replace("res://", "");
				}
			}


		} break;
		case ACCESS_USERDATA: {


			if (p_path.begins_with("user://")) {

				String data_dir=OS::get_singleton()->get_data_dir();
				if (data_dir != "") {

					return p_path.replace("user:/",data_dir);
				};
				return p_path.replace("user://", "");
			}

		} break;
		case ACCESS_FILESYSTEM: {

			return p_path;
		} break;
	}

	return p_path;
}


DirAccess::CreateFunc DirAccess::create_func[ACCESS_MAX]={0,0,0};

DirAccess *DirAccess::create_for_path(const String& p_path) {

	DirAccess *da=NULL;
	if (p_path.begins_with("res://")) {

		da=create(ACCESS_RESOURCES);
	} else if (p_path.begins_with("user://")) {

		da=create(ACCESS_USERDATA);
	} else {

		da=create(ACCESS_FILESYSTEM);
	}

	return da;
}

DirAccess *DirAccess::open(const String& p_path,Error *r_error) {

	DirAccess *da=create_for_path(p_path);

	ERR_FAIL_COND_V(!da,NULL);
	Error err = da->change_dir(p_path);
	if (r_error)
		*r_error=err;
	if (err!=OK) {
		memdelete(da);
		return NULL;
	}

	return da;
}

DirAccess *DirAccess::create(AccessType p_access) {
	
	DirAccess * da = create_func[p_access]?create_func[p_access]():NULL;
	if (da) {
		da->_access_type=p_access;
	}

	return da;
};


String DirAccess::get_full_path(const String& p_path,AccessType p_access) {

	DirAccess *d=DirAccess::create(p_access);
	if (!d)
		return p_path;
		
	d->change_dir(p_path);
	String full=d->get_current_dir();
	memdelete(d);
	return full;
}

Error DirAccess::copy(String p_from,String p_to) {
	
	//printf("copy %s -> %s\n",p_from.ascii().get_data(),p_to.ascii().get_data());
	Error err;
	FileAccess *fsrc = FileAccess::open(p_from, FileAccess::READ,&err);
	
	if (err) {

		ERR_FAIL_COND_V( err, err );
	}

	
	FileAccess *fdst = FileAccess::open(p_to, FileAccess::WRITE,&err );
	if (err) {
						
		fsrc->close();
		memdelete( fsrc );
		ERR_FAIL_COND_V( err, err );
	}
	
	fsrc->seek_end(0);
	int size = fsrc->get_pos();
	fsrc->seek(0);
	err = OK;
	while(size--) {
		
		if (fsrc->get_error()!=OK) {
			err= fsrc->get_error();
			break;
		}
		if (fdst->get_error()!=OK) {
			err= fdst->get_error();
			break;
		}
		
		fdst->store_8( fsrc->get_8() );
	}
	
	memdelete(fsrc);
	memdelete(fdst);
	
	return err;
}

bool DirAccess::exists(String p_dir) {

	DirAccess* da = DirAccess::create_for_path(p_dir);
	bool valid = da->change_dir(p_dir)==OK;
	memdelete(da);
	return valid;

}

DirAccess::DirAccess(){

	_access_type=ACCESS_FILESYSTEM;
}


DirAccess::~DirAccess() {

}