brouter/brouter-server/src/main/java/btools/server/request/ProfileUploadHandler.java

package btools.server.request;

import btools.router.RoutingContext;
import btools.router.RoutingEngine;
import btools.server.ServiceContext;
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.util.HashMap;
import java.util.Map;

/**
 * Custom profile uploads
 */
public class ProfileUploadHandler
{
  // maximum number of characters (file size limit for custom profiles)
  private static final int MAX_LENGTH = 100000;

  // prefix for custom profile id to distinguish from default profiles
  public static final String CUSTOM_PREFIX = "custom_";
  public static final String SHARED_PREFIX = "shared_";

	private ServiceContext serviceContext;
  
	public ProfileUploadHandler( ServiceContext serviceContext)
	{
		this.serviceContext = serviceContext;
	}

  public void handlePostRequest(String profileId, BufferedReader br, BufferedWriter response) throws IOException
  {
    BufferedWriter fileWriter = null;

    try
    {
      String id;
      if ( profileId != null )
      {
        // update existing file when id appended
        id = profileId.substring( ProfileUploadHandler.CUSTOM_PREFIX.length() );
      } else {
        id = "" + System.currentTimeMillis();
      }
      File file = new File( getOrCreateCustomProfileDir(), id + ".brf" );
      fileWriter = new BufferedWriter(new OutputStreamWriter(new FileOutputStream( file ) ) );
      //StringWriter sw = new StringWriter(); bw = new BufferedWriter(sw);

      // only profile text as content
      readPostData(br, fileWriter, id);

      fileWriter.flush();
      //System.out.println("data: |" + sw.toString() + "|");

      Map<String, String> responseData = new HashMap<String, String>();
      responseData.put( "profileid", CUSTOM_PREFIX + id );

      validateProfile( id, responseData );

      response.write( toJSON( responseData ) );
    }
    finally
    {
      if ( fileWriter != null ) try { fileWriter.close(); } catch( Exception e ) {}
    }
  }

  private File getOrCreateCustomProfileDir()
  {
    // workaround: customProfileDir relative to profileDir, because RoutingEngine doesn't know custom profiles
    File customProfileDir = new File(serviceContext.profileDir, serviceContext.customProfileDir);
    if (!customProfileDir.exists())
    {
      customProfileDir.mkdir();
    }
    return customProfileDir;
  }

  // reads HTTP POST content from input into output stream/writer
  private static void readPostData( BufferedReader ir, BufferedWriter bw, String id ) throws IOException
  {
    // Content-Type: text/plain;charset=UTF-8

    int numChars = 0;

    // Content-Length header is in bytes (!= characters for UTF8),
    // but Reader reads characters, so don't know number of characters to read
    for(;;)
    {
      // read will block when false, occurs at end of stream rather than -1
      if (!ir.ready()) {
        try { Thread.sleep(1000); } catch( Exception e ) {}
        if ( !ir.ready() ) break;
      }
      int c = ir.read();
      if ( c == -1) break;
      bw.write( c );
          
      numChars++;
      if (numChars > MAX_LENGTH)
        throw new IOException("Maximum number of characters exceeded (" + MAX_LENGTH + ", " + id + ")");
    }
  }

  private String toJSON( Map<String, String> data )
  {
    boolean first = true;
    StringBuilder sb = new StringBuilder();
    sb.append("{");
    for (Map.Entry<String, String> entry : data.entrySet()) {
      sb.append(first ? "\n" : ",\n");
      sb.append("  \"");
      sb.append(entry.getKey());
      sb.append("\": \"");
      sb.append(entry.getValue());
      sb.append("\"");
      first = false;
    }
    sb.append("\n}\n");
    return sb.toString();
  }

  public void validateProfile(String id, Map<String, String> responseData)
  {
    // validate by initializing RoutingEngine, where parsing is done, and catching exceptions
    // see https://github.com/abrensch/brouter/issues/14
    try
    {
      RoutingContext rc = new RoutingContext();
      rc.localFunction =  serviceContext.customProfileDir + "/" + id;
      new RoutingEngine( null, null, null, null, rc );
    }
    catch ( Exception e )
    {
      String msg = e.getMessage();
      if ( msg == null )
      {
        msg = "";
      }
      else if ( msg.indexOf( "does not contain expressions for context" ) >= 0 )
      {
        // remove absolute path in this specific exception, useful for server, but don't disclose to client
        msg = msg.substring( msg.indexOf( "does not contain expressions for context" ) );
      }
      responseData.put( "error", "Profile error: " + msg );
    }
  }
}