FLocal/Common/BBCodes/helpers/UrlProcessor.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Web;

namespace FLocal.Common.BBCodes {
	class UrlProcessor {

		public struct UrlInfo {
			public readonly bool isLocal;
			private readonly string rawRelativeUrl;
			public string relativeUrl {
				get {
					return this.rawRelativeUrl;
					//return HttpUtility.UrlEncode(this.rawRelativeUrl);
				}
			}
			public UrlInfo(bool isLocal, string relativeUrl) {
				this.isLocal = isLocal;
				this.rawRelativeUrl = relativeUrl;
			}
		}

		private static string Safe(string str) {
			return System.Web.HttpUtility.HtmlEncode(str);
		}

		public static HashSet<string> KnownAliases = new HashSet<string> {
			"forum.local",
			"forum.b.gz.ru",
			"www.snto-msu.net",
			"hq.sectorb.msk.ru",
			"petaflop.b.gz.ru",
			"194.88.210.5",
			"forumlocal.ru",
			"forumbgz.ru",
			"forum.hn",
		};

		public const string DOMAIN = ".forum.hn";

		public static UrlInfo Process(string url) {
			if (url.StartsWith("/")) {
				return new UrlInfo(true, url);
			}
			Uri uri;
			try {
				uri = new Uri(url);
			} catch(UriFormatException) {
				throw new Core.FLocalException("wrong url: " + url);
			}
			if (KnownAliases.Contains(uri.Host.ToLower()) || uri.Host.ToLower().EndsWith(DOMAIN)) {
				return new UrlInfo(true, uri.PathAndQuery);
			} else {
				return new UrlInfo(false, uri.ToString());
			}
		}

		public static string ProcessLink(string link, string title, bool shortenRelative) {
			bool isExternal = true;
			string url;
			if(shortenRelative) {
				var urlInfo = UrlProcessor.Process(link);
				url = urlInfo.relativeUrl;
				isExternal = !urlInfo.isLocal;
			} else {
				var urlInfo = new Uri(link);
				url = urlInfo.ToString();
			}
			if(title == null) {
				if(!isExternal) {
					var parts = url.Split(new char[] {'/'}, StringSplitOptions.RemoveEmptyEntries);
					var combinedToLower = string.Join("/", parts).ToLower();
					if(combinedToLower.StartsWith("upload/item/")) {
						title = Safe(dataobjects.Upload.LoadById(int.Parse(parts[2])).filename);
					} else if(combinedToLower.StartsWith("post/")) {
						title = Safe(dataobjects.Post.LoadById(int.Parse(parts[1])).title);
					} else if(combinedToLower.StartsWith("thread/")) {
						title = Safe(dataobjects.Thread.LoadById(int.Parse(parts[1])).title);
					} else if(combinedToLower.StartsWith("board/") || combinedToLower.StartsWith("boardasthread/")) {
						title = Safe(dataobjects.Board.LoadById(int.Parse(parts[1])).name);
					} else if(combinedToLower.StartsWith("poll/")) {
						title = Safe(dataobjects.Poll.LoadById(int.Parse(parts[1])).title);
					} else {
						title = link;
					}
				} else {
					title = link;
				}
			}
			string result = "<a href=\"" + url + "\">" + title + "</a>";
			if(isExternal) {
				result += "<img src=\"/static/images/external.png\" border=\"0\"/>";
			}
			return result;
		}

	}
}