diff options
Diffstat (limited to 'core/os/dir_access.cpp')
| -rw-r--r-- | core/os/dir_access.cpp | 401 |
1 files changed, 401 insertions, 0 deletions
diff --git a/core/os/dir_access.cpp b/core/os/dir_access.cpp new file mode 100644 index 0000000000..496ebdfa27 --- /dev/null +++ b/core/os/dir_access.cpp @@ -0,0 +1,401 @@ +/*************************************************************************/ +/* dir_access.cpp */ +/*************************************************************************/ +/* This file is part of: */ +/* GODOT ENGINE */ +/* http://www.godotengine.org */ +/*************************************************************************/ +/* Copyright (c) 2007-2014 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 ""; +} + + +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; + }; + Globals* g = Globals::get_singleton(); + String cur = normalize_path(g->globalize_path(get_current_dir())); + if (cur[cur.length()-1] != '/') { + cur = cur + "/"; + }; + + String dir = normalize_path(g->globalize_path(p_dir)); + if (dir.length() < 1) { + return OK; + }; + if (dir[dir.length()-1] != '/') { + dir = dir + "/"; + }; + + ERR_FAIL_COND_V(dir.find(cur) != 0, FAILED); + + String rel = dir.substr(cur.length(), (dir.length() - cur.length())); + + int pos = 0; + while (pos < rel.length()) { + + int n = rel.find("/", pos); + if (n < 0) { + n = rel.length(); + }; + pos = n + 1; + + if (pos > 1) { + Error err = make_dir(rel.substr(0, pos -1)); + 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; +} + +DirAccess::DirAccess(){ + + _access_type=ACCESS_FILESYSTEM; +} + + +DirAccess::~DirAccess() { + +} + + |