Umbraco Media Icons

AppBase.cs

using System;
using System.IO;
using umbraco.BusinessLogic;
using umbraco.cms.businesslogic.media;
using umbraco.cms.presentation.Trees;
using umbraco.IO;

namespace TheOutfield.UmbExt.MediaIcons.Logic
{
    public class AppBase : ApplicationBase
    {
        public AppBase()
        {
            BaseTree.BeforeNodeRender += BaseTree_BeforeNodeRender;
        }

        protected void BaseTree_BeforeNodeRender(ref XmlTree sender, ref XmlTreeNode node, EventArgs e)
        {
            if (node.NodeType != "media") return;

            var ext = "";

            var mediaItem = new Media(Convert.ToInt32(node.NodeID));
            if (mediaItem.ContentType.Alias != "Folder")
            {
                var mediaFileProp = mediaItem.getProperty("umbracoFile");
                if (mediaFileProp != null)
                {
                    var mediaFile = mediaFileProp.Value.ToString();
                    if (!string.IsNullOrEmpty(mediaFile) && mediaFile.LastIndexOf(".") < mediaFile.Length)
                    {
                        ext = mediaFile.Substring(mediaFile.ToLower().LastIndexOf(".") + 1);
                    }
                }
            }

            if (!string.IsNullOrEmpty(ext))
            {
                var iconUrl = string.Format("/umbraco/images/umbraco/{0}.gif", ext);
                if (File.Exists(IOHelper.MapPath(iconUrl)))
                {
                    node.Icon = ext + ".gif";
                }
            }
        }
    }
}