using System;
using System.IO;
using System.Text;

namespace NewLife.IP
{
    class Zip : IDisposable
    {
        uint Index_Set;
        uint Index_End;
        uint Index_Count;
        uint Search_Index_Set;
        uint Search_Index_End;
        IndexInfo Search_Set;
        IndexInfo Search_Mid;
        IndexInfo Search_End;

        #region 属性
        private Stream _Stream;
        /// <summary>数据流</summary>
        public Stream Stream { get { return _Stream; } set { _Stream = value; } }
        #endregion

        #region 构造
        /// <summary>析构</summary>
        ~Zip() { OnDispose(false); }

        /// <summary>销毁</summary>
        public void Dispose() { OnDispose(true); }

        void OnDispose(Boolean disposing)
        {
            if (_Stream != null) _Stream.Dispose();

            if (disposing) GC.SuppressFinalize(this);
        }
        #endregion

        #region 数据源
        public Zip SetStream(Stream stream)
        {
            var ms = new MemoryStream();

            var buf = new Byte[3];
            stream.Read(buf, 0, buf.Length);
            stream.Position -= 3;

            // 仅支持Gzip压缩，可用7z软件先压缩为gz格式
            if (buf[0] == 0x1F & buf[1] == 0x8B && buf[2] == 0x08)
                IOHelper.DecompressGZip(stream, ms);
            else
                IOHelper.CopyTo(stream, ms);

            ms.Position = 0;
            _Stream = ms;

            Index_Set = GetUInt32();
            Index_End = GetUInt32();
            Index_Count = (Index_End - Index_Set) / 7u + 1u;

            return this;
        }
        #endregion

        #region 方法
        public String GetAddress(UInt32 ip)
        {
            if (_Stream == null) return "";

            Search_Index_Set = 0u;
            Search_Index_End = Index_Count - 1u;

            while (true)
            {
                Search_Set = IndexInfoAtPos(Search_Index_Set);
                Search_End = IndexInfoAtPos(Search_Index_End);
                if (ip >= Search_Set.IpSet && ip <= Search_Set.IpEnd) break;

                if (ip >= Search_End.IpSet && ip <= Search_End.IpEnd) return ReadAddressInfoAtOffset(Search_End.Offset);

                Search_Mid = IndexInfoAtPos((Search_Index_End + Search_Index_Set) / 2u);
                if (ip >= Search_Mid.IpSet && ip <= Search_Mid.IpEnd) return ReadAddressInfoAtOffset(Search_Mid.Offset);

                if (ip < Search_Mid.IpSet)
                    Search_Index_End = (Search_Index_End + Search_Index_Set) / 2u;
                else
                    Search_Index_Set = (Search_Index_End + Search_Index_Set) / 2u;
            }
            return ReadAddressInfoAtOffset(Search_Set.Offset);
        }

        String ReadAddressInfoAtOffset(uint Offset)
        {
            _Stream.Position = Offset + 4;
            Byte tag = GetTag();
            String addr;
            String area;
            if (tag == 1)
            {
                _Stream.Position = GetOffset();
                tag = GetTag();
                if (tag == 2)
                {
                    uint offset = GetOffset();
                    area = ReadArea();
                    _Stream.Position = offset;
                    addr = ReadString();
                }
                else
                {
                    _Stream.Position -= 1;
                    addr = ReadString();
                    area = ReadArea();
                }
            }
            else
            {
                if (tag == 2)
                {
                    uint offset = GetOffset();
                    area = ReadArea();
                    _Stream.Position = offset;
                    addr = ReadString();
                }
                else
                {
                    _Stream.Position -= 1;
                    addr = ReadString();
                    area = ReadArea();
                }
            }
            return (addr + " " + area).Trim();
        }

        uint GetOffset()
        {
            return BitConverter.ToUInt32(new Byte[]
                {
                    (Byte)_Stream.ReadByte(),
                    (Byte)_Stream.ReadByte(),
                    (Byte)_Stream.ReadByte(),
                    0
                }, 0);
        }

        String ReadArea()
        {
            Byte tag = GetTag();
            if (tag == 1 || tag == 2)
                _Stream.Position = GetOffset();
            else
                _Stream.Position -= 1;

            return ReadString();
        }

        String ReadString()
        {
            var k = 0;
            var buf = new Byte[256];
            buf[k] = (Byte)_Stream.ReadByte();
            while (buf[k] != 0)
            {
                k += 1;
                buf[k] = (Byte)_Stream.ReadByte();
            }
            var str = Encoding.GetEncoding("GB2312").GetString(buf).Trim().Trim('\0').Trim();
            if (str == "CZ88.NET") return null;
            return str;
        }

        Byte GetTag()
        {
            return (Byte)_Stream.ReadByte();
        }

        IndexInfo IndexInfoAtPos(uint Index_Pos)
        {
            var inf = new IndexInfo();
            _Stream.Position = Index_Set + 7u * Index_Pos;
            inf.IpSet = GetUInt32();
            inf.Offset = GetOffset();
            _Stream.Position = inf.Offset;
            inf.IpEnd = GetUInt32();
            return inf;
        }

        uint GetUInt32()
        {
            Byte[] array = new Byte[4];
            _Stream.Read(array, 0, 4);
            return BitConverter.ToUInt32(array, 0);
        }
        #endregion

        /// <summary>索引结构</summary>
        class IndexInfo
        {
            public uint IpSet;
            public uint IpEnd;
            public uint Offset;
        }
    }
}