Synnefo » Pithos MS Client

using System;

using System.Diagnostics;

using System.Diagnostics.Contracts;

using System.IO;

using System.ComponentModel.Composition;

using Pithos.Interfaces;

namespace Pithos.ShellExtensions.Overlays

{

    public class IconOverlayBase : IShellIconOverlayIdentifier

    {

        protected static string PithosPrefix = "0Pithos";

        public string OverlayName { get; private set; }

        public string IconPath { get; private set; }

        [Import(typeof (IStatusChecker))] 

        public IStatusChecker StatusChecker;

        [Import(typeof (IPithosSettings))] 

        public IPithosSettings Settings;

        public IconOverlayBase(string iconName)

        {

            Trace.Write("Icon Overlay Instance created");

            IoC.Current.Compose(this);

            string overlayName=PithosPrefix + iconName;

            string iconFile = iconName + "Icon.ico";

            var iconsFolder = Settings.IconsPath;

            var fullPath = Path.Combine(iconsFolder, iconFile);

            OverlayName = overlayName;

            IconPath = fullPath;

        }

        public int IsMemberOf(string path, uint attributes)

        {

            if (String.IsNullOrWhiteSpace(path))

                throw new ArgumentNullException("Empty path");

            Trace.Write(String.Format("ICON Status check for {0} - {1}",path,GetType().Name));

            if (!path.StartsWith(Settings.PithosPath,true,null))

                return WinError.S_FALSE;

            var status=StatusChecker.GetFileOverlayStatus(path);

            var isMember = (PithosPrefix + status == OverlayName);

            Trace.Write(String.Format("Status check for {0} is {1}",path,isMember));

            return isMember ? WinError.S_OK : WinError.S_FALSE;

        }

        public int GetOverlayInfo(

            IntPtr iconFileBuffer,

            int iconFileBufferSize,

            out int iconIndex,

            out uint flags)

        {

            Trace.Write("Looking for icons");

            Trace.WriteLine(string.Format("ICON file {0}", IconPath));

            int bytesCount = System.Text.Encoding.Unicode.GetByteCount(IconPath);

            Trace.WriteLine(string.Format(" GetOverlayInfo::{0}",bytesCount));

            MarshalHelpers.CopyToBuffer(IconPath, iconFileBuffer, iconFileBufferSize);

            flags = (uint)(ISIOI.ISIOI_ICONFILE);// | ISIOI.ISIOI_ICONINDEX);

            iconIndex = 0;

            return WinError.S_OK;

        }

        public int  GetPriority(out int priority)

        {

            Trace.Write("Checking for priority");

            priority = 0;

            return WinError.S_OK;

        }

        public static void RegisterOverlay(Type t,string iconName)

        {

            try

            {

                ShellExtReg.RegisterIconOverlayIdentifier(t.GUID, iconName);

                NativeMethods.SHChangeNotify(HChangeNotifyEventID.SHCNE_ASSOCCHANGED, HChangeNotifyFlags.SHCNF_IDLIST,

                                             IntPtr.Zero, IntPtr.Zero);

                Trace.Write("Registered icon handler");

            }

            catch (Exception ex)

            {

                Trace.WriteLine(ex.Message); // Log the error

                throw;  // Re-throw the exception

            }

        }

        public static void UnregisterOverlay(Type t,string iconName)

        {

            try

            {                

                ShellExtReg.UnregisterIconOverlayIdentifier(t.GUID, iconName);

                Trace.Write("UnRegistered icon handler");

            }

            catch (Exception ex)

            {

                Trace.WriteLine(ex.Message); // Log the error

                throw;  // Re-throw the exception

            }

        }

    }

}