using System;
using System.Text;
using System.IO;
using System.Collections.Generic;
using libsecondlife;

namespace libsecondlife.TestClient
{
    public class DumpOutfitCommand : Command
    {
        List<LLUUID> OutfitAssets = new List<LLUUID>();
        AssetManager.ImageReceivedCallback ImageReceivedHandler;

        public DumpOutfitCommand(TestClient testClient)
        {
            Name = "dumpoutfit";
            Description = "Dumps all of the textures from an avatars outfit to the hard drive. Usage: dumpoutfit [avatar-uuid]";
            ImageReceivedHandler = new AssetManager.ImageReceivedCallback(Assets_OnImageReceived);
        }

        public override string Execute(string[] args, LLUUID fromAgentID)
        {
            if (args.Length != 1)
                return "Usage: dumpoutfit [avatar-uuid]";

            LLUUID target;

            if (!LLUUID.TryParse(args[0], out target))
                return "Usage: dumpoutfit [avatar-uuid]";

            lock (Client.AvatarList)
            {
                foreach (Avatar avatar in Client.AvatarList.Values)
                {
                    if (avatar.ID == target)
                    {
                        StringBuilder output = new StringBuilder("Downloading ");

                        lock (OutfitAssets) OutfitAssets.Clear();
                        Client.NewAssetManager.OnImageReceived += ImageReceivedHandler;

                        foreach (KeyValuePair<uint, LLObject.TextureEntryFace> face in avatar.Textures.FaceTextures)
                        {
                            ImageType type = ImageType.Normal;

                            switch ((AppearanceManager.TextureIndex)face.Key)
                            {
                                case AppearanceManager.TextureIndex.HeadBaked:
                                case AppearanceManager.TextureIndex.EyesBaked:
                                case AppearanceManager.TextureIndex.UpperBaked:
                                case AppearanceManager.TextureIndex.LowerBaked:
                                case AppearanceManager.TextureIndex.SkirtBaked:
                                    type = ImageType.Baked;
                                    break;
                            }

                            OutfitAssets.Add(face.Value.TextureID);
                            Client.NewAssetManager.RequestImage(face.Value.TextureID, type, 100000.0f, 0);

                            output.Append(((AppearanceManager.TextureIndex)face.Key).ToString());
                            output.Append(" ");
                        }

                        return output.ToString();
                    }
                }
            }

            return "Couldn't find avatar " + target.ToStringHyphenated();
        }

        private void Assets_OnImageReceived(ImageDownload image)
        {
            lock (OutfitAssets)
            {
                if (OutfitAssets.Contains(image.ID))
                {
                    if (image.Success)
                    {
                        try
                        {
                            File.WriteAllBytes(image.ID.ToStringHyphenated() + ".jp2", image.AssetData);
                            Console.WriteLine("Wrote JPEG2000 image " + image.ID.ToStringHyphenated() + ".jp2");

                            byte[] tgaFile = OpenJPEGNet.OpenJPEG.DecodeToTGA(image.AssetData);
                            File.WriteAllBytes(image.ID.ToStringHyphenated() + ".tga", tgaFile);
                            Console.WriteLine("Wrote TGA image " + image.ID.ToStringHyphenated() + ".tga");
                        }
                        catch (Exception e)
                        {
                            Console.WriteLine(e.ToString());
                        }
                    }
                    else
                    {
                        Console.WriteLine("Failed to download image " + image.ID.ToStringHyphenated());
                    }

                    OutfitAssets.Remove(image.ID);

                    if (OutfitAssets.Count == 0)
                        Client.NewAssetManager.OnImageReceived -= ImageReceivedHandler;
                }
            }
        }
    }
}