package org.opensolaris.opengrok.search;

import java.io.BufferedReader;

import java.io.FileInputStream;

import java.io.FileReader;

import java.io.IOException;

import java.io.InputStreamReader;

import java.io.Reader;

import java.io.Writer;

import java.util.ArrayList;

import java.util.LinkedHashMap;

import java.util.Map;

import java.util.logging.Level;

import java.util.zip.GZIPInputStream;

import org.apache.lucene.document.Document;

import org.apache.lucene.document.Fieldable;

import org.opensolaris.opengrok.OpenGrokLogger;

import org.opensolaris.opengrok.analysis.Definitions;

import org.opensolaris.opengrok.analysis.TagFilter;

import org.opensolaris.opengrok.configuration.RuntimeEnvironment;

import org.opensolaris.opengrok.history.HistoryException;

import org.opensolaris.opengrok.search.context.Context;

import org.opensolaris.opengrok.search.context.HistoryContext;

import org.opensolaris.opengrok.web.EftarFileReader;

import org.opensolaris.opengrok.web.Util;

public final class Results {

private Results() {

// Util class, should not be constructed

}

public static void prettyPrintHTML(Document[] docs, int start, int end, Writer out,

Context sourceContext, HistoryContext historyContext,

Summarizer summer, String urlPrefix,

String morePrefix,

String srcRoot,

String dataRoot,

EftarFileReader desc)

throws HistoryException, IOException, ClassNotFoundException

{

char[] content = new char[1024*8];

LinkedHashMap<String, ArrayList<Document>> dirHash = new LinkedHashMap<String, ArrayList<Document>>();

for (int i = start; i < end; i++) {

Document doc = docs[i];

String rpath = doc.get("path");

String parent = rpath.substring(0,rpath.lastIndexOf('/'));

ArrayList<Document> dirDocs = dirHash.get(parent);

if(dirDocs == null) {

dirDocs = new ArrayList<Document>();

dirHash.put(parent, dirDocs);

}

dirDocs.add(doc);

}

for (Map.Entry<String, ArrayList<Document>> entry: dirHash.entrySet()) {

String parent = entry.getKey();

String tag = (desc == null) ? "" : " - <i>" + desc.get(parent) + "</i>";

out.write("<tr class=\"dir\"><td colspan=\"2\">&nbsp;&nbsp;<a href=\"");

out.write(Util.URIEncodePath(urlPrefix + parent));

out.write("/\">" + parent + "/</a>" + tag + "</td></tr>");

boolean alt = false;

for (Document doc: entry.getValue()) {

String rpath = doc.get("path");

String self = rpath.substring(rpath.lastIndexOf('/')+1, rpath.length());

String selfUrl = Util.URIEncodePath(urlPrefix + rpath);

out.write("<tr ");

if(alt) {

out.write(" class=\"alt\"");

}

alt ^= true;

out.write("><td class=\"f\"><a href=\"" +

selfUrl + "\">"+self+"</a>&nbsp;</td><td><tt class=\"con\">");

if (sourceContext != null) {

String genre = doc.get("t");

Definitions tags = null;

Fieldable tagsField = doc.getFieldable("tags");

if (tagsField != null) {

tags = Definitions.deserialize(tagsField.binaryValue());

}

try {

if ("p".equals(genre) && srcRoot != null) {

sourceContext.getContext(new FileReader(srcRoot + rpath), out, urlPrefix, morePrefix, rpath,

tags, true, null);

} else if("x".equals(genre) && dataRoot != null && summer != null){

Reader r = null;

if ( RuntimeEnvironment.getInstance().isCompressXref() ) {

r = new TagFilter(new BufferedReader(new InputStreamReader(new GZIPInputStream(new FileInputStream(dataRoot + "/xref" + rpath+".gz"))))); }

else {

r = new TagFilter(new BufferedReader(new FileReader(dataRoot + "/xref" + rpath))); }

int len = r.read(content);

out.write(summer.getSummary(new String(content, 0, len)).toString());

r.close();

} else if("h".equals(genre) && srcRoot != null && summer != null){

Reader r = new TagFilter(new BufferedReader(new FileReader(srcRoot + rpath)));

int len = r.read(content);

out.write(summer.getSummary(new String(content, 0, len)).toString());

r.close();

} else {

sourceContext.getContext(null, out, urlPrefix, morePrefix, rpath, tags, true, null);

}

} catch (IOException e) {

OpenGrokLogger.getLogger().log(Level.WARNING, "An error occured while creating summary of "+rpath, e);

}

//out.write("Genre = " + genre);

}

if(historyContext != null) {

historyContext.getContext(srcRoot + parent, self, rpath, out);

}

out.write("</tt></td></tr>\n");

}

}

}

}