#ifdef USE_PCH
 #include "pch.h"
#else
 #include <cassert>
 #include "sitecing/util.h"
#endif

namespace sitecing {

    static const char *unsafeChars = "<>& \n\"";

    string html_escape(const string& str,int flags) {
	string rv = str;
	string::size_type screwed = 0;
	for(;;) {
	    screwed = rv.find_first_of(unsafeChars,screwed);
	    if(screwed == string::npos)
		break;
	    while(screwed<rv.length() && strchr(unsafeChars,rv.at(screwed))) {
		char danger = rv.at(screwed);
		switch(danger) {
		    case '<':
			rv.replace(screwed,1,"&lt;"); screwed+=4;
			break;
		    case '>':
			rv.replace(screwed,1,"&gt;"); screwed+=4;
			break;
		    case '&':
			rv.replace(screwed,1,"&amp;"); screwed+=5;
			break;
		    case ' ':
			if(flags&html_escape_nbsp) {
			    rv.replace(screwed,1,"&nbsp;"); screwed+=6;
			}else
			    screwed++;
			break;
		    case '\n':
			if(flags&html_escape_br) {
			    if(flags&html_escape_br_noslash) {
				rv.replace(screwed,1,"<br>\n"); screwed += 5;
			    }else{
				rv.replace(screwed,1,"<br/>\n"); screwed += 6;
			    }
			}else
			    screwed++;
			break;
		    case '\"':
			if(flags&html_escape_quot) {
			    rv.replace(screwed,1,"&quot;"); screwed+=6;
			}else
			    screwed++;
			break;
		    default:
			assert(false);
			break;
		}
	    }
	}
	return rv;
    }

    void checkpoint::set() {
	point = stream->tellp();
	if(last_will==will_intestate)
	    last_will = will_rollback;
    }

    void checkpoint::make_will(will_t lw) {
	last_will = lw;
    }

    void checkpoint::rollback() {
	stream->seekp(point);
	if(last_will == will_rollback)
	    last_will = will_intestate;
    }
    void checkpoint::commit() {
	point = stream->tellp();
	if(last_will == will_rollback)
	    last_will = will_intestate;
    }

}