Create GoogleImageSitemapGenerator.java

package com.redfin.sitemapgenerator;

import java.io.File;

import java.net.MalformedURLException;

import java.net.URL;

/**

 * Builds a image sitemap for Google Code Search.  To configure options, use {@link #builder(URL, File)}

 * @author Lakshay Bhambri

 */

public class GoogleImageSitemapGenerator extends SitemapGenerator<GoogleImageSitemapUrl,GoogleImageSitemapGenerator> {

	GoogleImageSitemapGenerator(AbstractSitemapGeneratorOptions<?> options) {

		super(options, new Renderer());

	}

	/** Configures the generator with a base URL and directory to write the sitemap files.

	 * 

	 * @param baseUrl All URLs in the generated sitemap(s) should appear under this base URL

	 * @param baseDir Sitemap files will be generated in this directory as either "sitemap.xml" or "sitemap1.xml" "sitemap2.xml" and so on.

	 * @throws MalformedURLException

	 */

	public GoogleImageSitemapGenerator(String baseUrl, File baseDir)

			throws MalformedURLException {

		this(new SitemapGeneratorOptions(baseUrl, baseDir));

	}

	/**Configures the generator with a base URL and directory to write the sitemap files.

	 * 

	 * @param baseUrl All URLs in the generated sitemap(s) should appear under this base URL

	 * @param baseDir Sitemap files will be generated in this directory as either "sitemap.xml" or "sitemap1.xml" "sitemap2.xml" and so on.

	 */

	public GoogleImageSitemapGenerator(URL baseUrl, File baseDir) {

		this(new SitemapGeneratorOptions(baseUrl, baseDir));

	}

	/**Configures the generator with a base URL and a null directory. The object constructed

	 * is not intended to be used to write to files. Rather, it is intended to be used to obtain

	 * XML-formatted strings that represent sitemaps.

	 * 

	 * @param baseUrl All URLs in the generated sitemap(s) should appear under this base URL

	 */

	public GoogleImageSitemapGenerator(String baseUrl) throws MalformedURLException {

		this(new SitemapGeneratorOptions(new URL(baseUrl)));

	}

	/**Configures the generator with a base URL and a null directory. The object constructed

	 * is not intended to be used to write to files. Rather, it is intended to be used to obtain

	 * XML-formatted strings that represent sitemaps.

	 * 

	 * @param baseUrl All URLs in the generated sitemap(s) should appear under this base URL

	 */

	public GoogleImageSitemapGenerator(URL baseUrl) {

		this(new SitemapGeneratorOptions(baseUrl));

	}

	/** Configures a builder so you can specify sitemap generator options

	 * 

	 * @param baseUrl All URLs in the generated sitemap(s) should appear under this base URL

	 * @param baseDir Sitemap files will be generated in this directory as either "sitemap.xml" or "sitemap1.xml" "sitemap2.xml" and so on.

	 * @return a builder; call .build() on it to make a sitemap generator

	 */

	public static SitemapGeneratorBuilder<GoogleImageSitemapGenerator> builder(URL baseUrl, File baseDir) {

		return new SitemapGeneratorBuilder<GoogleImageSitemapGenerator>(baseUrl, baseDir, GoogleImageSitemapGenerator.class);

	}

	/** Configures a builder so you can specify sitemap generator options

	 * 

	 * @param baseUrl All URLs in the generated sitemap(s) should appear under this base URL

	 * @param baseDir Sitemap files will be generated in this directory as either "sitemap.xml" or "sitemap1.xml" "sitemap2.xml" and so on.

	 * @return a builder; call .build() on it to make a sitemap generator

	 * @throws MalformedURLException

	 */

	public static SitemapGeneratorBuilder<GoogleImageSitemapGenerator> builder(String baseUrl, File baseDir) throws MalformedURLException {

		return new SitemapGeneratorBuilder<GoogleImageSitemapGenerator>(baseUrl, baseDir, GoogleImageSitemapGenerator.class);

	}

	private static class Renderer extends AbstractSitemapUrlRenderer<GoogleImageSitemapUrl> implements ISitemapUrlRenderer<GoogleImageSitemapUrl> {

		public Class<GoogleImageSitemapUrl> getUrlClass() {

			return GoogleImageSitemapUrl.class;

		}

		public String getXmlNamespaces() {

			return "xmlns:codesearch=\"http://www.google.com/schemas/sitemap-image/1.1\"";

		}

		public void render(GoogleCodeSitemapUrl url, StringBuilder sb,

				W3CDateFormat dateFormat) {

			StringBuilder tagSb = new StringBuilder();

			tagSb.append("    <codesearch:codesearch>\n");

			renderTag(tagSb, "codesearch", "filetype", url.getFileType());

			renderTag(tagSb, "codesearch", "license", url.getLicense());

			renderTag(tagSb, "codesearch", "filename", url.getFileName());

			renderTag(tagSb, "codesearch", "packageurl", url.getPackageUrl());

			renderTag(tagSb, "codesearch", "packagemap", url.getPackageMap());

			tagSb.append("    </codesearch:codesearch>\n");

			super.render(url, sb, dateFormat, tagSb.toString());

		}

	}

}