godot/core/io/translation_loader_po.cpp

/*************************************************************************/
/*  translation_loader_po.cpp                                            */
/*************************************************************************/
/*                       This file is part of:                           */
/*                           GODOT ENGINE                                */
/*                    http://www.godotengine.org                         */
/*************************************************************************/
/* Copyright (c) 2007-2016 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 "translation_loader_po.h"
#include "os/file_access.h"
#include "translation.h"

RES TranslationLoaderPO::load(const String &p_path, const String& p_original_path, Error *r_error) {

	if (r_error)
		*r_error=ERR_CANT_OPEN;

	FileAccess *f=FileAccess::open(p_path,FileAccess::READ);
	ERR_FAIL_COND_V(!f,RES());

	String l = f->get_line();

	enum Status {

		STATUS_NONE,
		STATUS_READING_ID,
		STATUS_READING_STRING,
	};

	Status status=STATUS_NONE;

	String msg_id;
	String msg_str;
	String config;
	if (r_error)
		*r_error=ERR_FILE_CORRUPT;

	Ref<Translation> translation = Ref<Translation>( memnew( Translation ));
	int line = 1;

	while(true) {

		String l = f->get_line();

		if (f->eof_reached()) {

			if ( status == STATUS_READING_STRING) {

				if (msg_id!="")
					translation->add_message(msg_id,msg_str);
				else if (config=="")
					config=msg_str;
				break;

			} else if ( status==STATUS_NONE)
				break;

			memdelete(f);
			ERR_EXPLAIN(p_path+":"+itos(line)+" Unexpected EOF while reading 'msgid' at file: ");
			ERR_FAIL_V(RES());
		}

		l=l.strip_edges();

		if (l.begins_with("msgid")) {

			if (status==STATUS_READING_ID) {

				memdelete(f);
				ERR_EXPLAIN(p_path+":"+itos(line)+" nexpected 'msgid', was expecting 'msgstr' while parsing: ");
				ERR_FAIL_V(RES());
			}

			if (msg_id!="")
				translation->add_message(msg_id,msg_str);
			else if (config=="")
				config=msg_str;

			l=l.substr(5,l.length()).strip_edges();
			status=STATUS_READING_ID;
			msg_id="";
			msg_str="";
		}

		if (l.begins_with("msgstr")) {

			if (status!=STATUS_READING_ID) {

				memdelete(f);
				ERR_EXPLAIN(p_path+":"+itos(line)+" Unexpected 'msgstr', was expecting 'msgid' while parsing: ");
				ERR_FAIL_V(RES());
			}

			l=l.substr(6,l.length()).strip_edges();
			status=STATUS_READING_STRING;
		}

		if (l=="" || l.begins_with("#")) {
			line++;
			continue; //nothing to read or comment
		}

		if (!l.begins_with("\"") || status==STATUS_NONE) {
			//not a string? failure!
			ERR_EXPLAIN(p_path+":"+itos(line)+" Invalid line '"+l+"' while parsing: ");
			ERR_FAIL_V(RES());

		}

		l=l.substr(1,l.length());
		//find final quote
		int end_pos=-1;
		for(int i=0;i<l.length();i++) {

			if (l[i]=='"' && (i==0 || l[i-1]!='\\')) {
				end_pos=i;
				break;
			}
		}

		if (end_pos==-1) {
			ERR_EXPLAIN(p_path+":"+itos(line)+" Expected '\"' at end of message while parsing file: ");
			ERR_FAIL_V(RES());
		}

		l=l.substr(0,end_pos);
		l=l.c_unescape();


		if (status==STATUS_READING_ID)
			msg_id+=l;
		else
			msg_str+=l;

		line++;
	}


	f->close();
	memdelete(f);

	if (config=="") {
		ERR_EXPLAIN("No config found in file: "+p_path);
		ERR_FAIL_V(RES());
	}

	Vector<String> configs = config.split("\n");
	for(int i=0;i<configs.size();i++) {

		String c = configs[i].strip_edges();
		int p = c.find(":");
		if (p==-1)
			continue;
		String prop = c.substr(0,p).strip_edges();
		String value = c.substr(p+1,c.length()).strip_edges();

		if (prop=="X-Language") {
			translation->set_locale(value);
		}
	}

	if (r_error)
		*r_error=OK;

	return translation;

}

void TranslationLoaderPO::get_recognized_extensions(List<String> *p_extensions) const{

	p_extensions->push_back("po");
	//p_extensions->push_back("mo"); //mo in the future...
}
bool TranslationLoaderPO::handles_type(const String& p_type) const{

	return (p_type=="Translation");
}

String TranslationLoaderPO::get_resource_type(const String &p_path) const {

	if (p_path.extension().to_lower()=="po")
		return "Translation";
	return "";
}

TranslationLoaderPO::TranslationLoaderPO()
{
}