This should make HG asset transfers work much better. It now uses HGUuidGatherer, which is a subclass of UuidGatherer. Hence, on-line HG asset transfers use exactly the same UUID collection code as save oar/xml. If it doesn't work, it's Justin's fault :D

5 files changed, 73 insertions, 116 deletions

diff --git a/OpenSim/Region/CoreModules/ServiceConnectors/Asset/HGAssetBroker.cs b/OpenSim/Region/CoreModules/ServiceConnectors/Asset/HGAssetBroker.cs
index aef04bf..42acde7 100644
--- a/OpenSim/Region/CoreModules/ServiceConnectors/Asset/HGAssetBroker.cs
+++ b/OpenSim/Region/CoreModules/ServiceConnectors/Asset/HGAssetBroker.cs
@@ -169,7 +169,7 @@ namespace OpenSim.Region.CoreModules.ServiceConnectors.Asset
 
         public AssetBase Get(string id)
         {
-            m_log.DebugFormat("[HG ASSET CONNECTOR]: Get {0}", id);
+            //m_log.DebugFormat("[HG ASSET CONNECTOR]: Get {0}", id);
             AssetBase asset = null;
             
             if (m_Cache != null)
diff --git a/OpenSim/Region/Framework/Scenes/Hypergrid/HGAssetMapper.cs b/OpenSim/Region/Framework/Scenes/Hypergrid/HGAssetMapper.cs
index 720a09b..1d4bc9c 100644
--- a/OpenSim/Region/Framework/Scenes/Hypergrid/HGAssetMapper.cs
+++ b/OpenSim/Region/Framework/Scenes/Hypergrid/HGAssetMapper.cs
@@ -97,7 +97,7 @@ namespace OpenSim.Region.Framework.Scenes.Hypergrid
             return false;
         }
 
-        private AssetBase FetchAsset(string url, UUID assetID, bool isTexture)
+        public AssetBase FetchAsset(string url, UUID assetID)
         {
             AssetBase asset = m_scene.AssetService.Get(url + "/" + assetID.ToString());
 
@@ -109,7 +109,7 @@ namespace OpenSim.Region.Framework.Scenes.Hypergrid
             return null;
         }
 
-        private bool PostAsset(string url, AssetBase asset)
+        public bool PostAsset(string url, AssetBase asset)
         {
             if (asset != null)
             {
@@ -129,7 +129,7 @@ namespace OpenSim.Region.Framework.Scenes.Hypergrid
                     }
                     catch
                     {
-                        m_log.Warn("[HGScene]: This won't work until Melanie kills a few more dragons");
+                        m_log.Warn("[HGScene]: Oops.");
                     }
 
                     m_scene.AssetService.Store(asset1);
@@ -155,86 +155,6 @@ namespace OpenSim.Region.Framework.Scenes.Hypergrid
 
         }
 
-        private void _guardedAdd(Dictionary<UUID, bool> lst, UUID obj, bool val)
-        {
-            if (!lst.ContainsKey(obj))
-                lst.Add(obj, val);
-        }
-
-        private void SniffTextureUUIDs(Dictionary<UUID, bool> uuids, SceneObjectGroup sog)
-        {
-            try
-            {
-                _guardedAdd(uuids, sog.RootPart.Shape.Textures.DefaultTexture.TextureID, true);
-            }
-            catch (Exception) { }
-
-            foreach (Primitive.TextureEntryFace tface in sog.RootPart.Shape.Textures.FaceTextures)
-            {
-                try
-                {
-                    _guardedAdd(uuids, tface.TextureID, true);
-                }
-                catch (Exception) { }
-            }
-
-            foreach (SceneObjectPart sop in sog.Children.Values)
-            {
-                try
-                {
-                    _guardedAdd(uuids, sop.Shape.Textures.DefaultTexture.TextureID, true);
-                }
-                catch (Exception) { }
-                foreach (Primitive.TextureEntryFace tface in sop.Shape.Textures.FaceTextures)
-                {
-                    try
-                    {
-                        _guardedAdd(uuids, tface.TextureID, true);
-                    }
-                    catch (Exception) { }
-                }
-            }
-        }
-
-        private void SniffTaskInventoryUUIDs(Dictionary<UUID, bool> uuids, SceneObjectGroup sog)
-        {
-            TaskInventoryDictionary tinv = sog.RootPart.TaskInventory;
-
-            lock (tinv)
-            {
-                foreach (TaskInventoryItem titem in tinv.Values)
-                {
-                    uuids.Add(titem.AssetID, (InventoryType)titem.Type == InventoryType.Texture);
-                }
-            }
-        }
-
-        private Dictionary<UUID, bool> SniffUUIDs(AssetBase asset)
-        {
-            Dictionary<UUID, bool> uuids = new Dictionary<UUID, bool>();
-            if ((asset != null) && ((AssetType)asset.Type == AssetType.Object))
-            {
-                string ass_str = Utils.BytesToString(asset.Data);
-                SceneObjectGroup sog = SceneObjectSerializer.FromOriginalXmlFormat(ass_str);
-
-                SniffTextureUUIDs(uuids, sog);
-
-                // We need to sniff further...
-                SniffTaskInventoryUUIDs(uuids, sog);
-            }
-
-            return uuids;
-        }
-
-        private Dictionary<UUID, bool> SniffUUIDs(UUID assetID)
-        {
-            //Dictionary<UUID, bool> uuids = new Dictionary<UUID, bool>();
-
-            AssetBase asset = m_scene.AssetService.Get(assetID.ToString());
-
-            return SniffUUIDs(asset);
-        }
-
         private void Dump(Dictionary<UUID, bool> lst)
         {
             m_log.Debug("XXX -------- UUID DUMP ------- XXX");
@@ -259,16 +179,18 @@ namespace OpenSim.Region.Framework.Scenes.Hypergrid
                 if (userAssetURL != null)
                 {
                     m_log.Debug("[HGScene]: Fetching object " + assetID + " to asset server " + userAssetURL);
-                    AssetBase asset = FetchAsset(userAssetURL, assetID, false); 
+                    AssetBase asset = FetchAsset(userAssetURL, assetID); 
 
                     if (asset != null)
                     {
                         m_log.Debug("[HGScene]: Successfully fetched item from remote asset server " + userAssetURL);
+
                         // OK, now fetch the inside.
-                        Dictionary<UUID, bool> ids = SniffUUIDs(asset);
-                        Dump(ids);
-                        foreach (KeyValuePair<UUID, bool> kvp in ids)
-                            FetchAsset(userAssetURL, kvp.Key, kvp.Value);
+                        Dictionary<UUID, int> ids = new Dictionary<UUID, int>();
+                        HGUuidGatherer uuidGatherer = new HGUuidGatherer(this, m_scene.AssetService, userAssetURL);
+                        uuidGatherer.GatherAssetUuids(asset.FullID, (AssetType)asset.Type, ids);
+                        foreach (UUID uuid in ids.Keys)
+                            FetchAsset(userAssetURL, uuid);
                     }
                     else
                         m_log.Warn("[HGScene]: Could not fetch asset from remote asset server " + userAssetURL);
@@ -315,21 +237,23 @@ namespace OpenSim.Region.Framework.Scenes.Hypergrid
                 if (userAssetURL != null)
                 {
                     m_log.Debug("[HGScene]: Posting object " + assetID + " to asset server " + userAssetURL);
-                    AssetBase ass1 = m_scene.AssetService.Get(assetID.ToString());
-                    if (ass1 != null)
+                    AssetBase asset = m_scene.AssetService.Get(assetID.ToString());
+                    if (asset != null)
                     {
-                        bool success = PostAsset(userAssetURL, ass1);
-
-                        // Now the inside
-                        Dictionary<UUID, bool> ids = SniffUUIDs(assetID);
-                        Dump(ids);
-                        foreach (KeyValuePair<UUID, bool> kvp in ids)
+                        Dictionary<UUID, int> ids = new Dictionary<UUID, int>();
+                        HGUuidGatherer uuidGatherer = new HGUuidGatherer(this, m_scene.AssetService, string.Empty);
+                        uuidGatherer.GatherAssetUuids(asset.FullID, (AssetType)asset.Type, ids);
+                        foreach (UUID uuid in ids.Keys)
                         {
-                            ass1 = m_scene.AssetService.Get(kvp.Key.ToString());
-                            PostAsset(userAssetURL, ass1);
+                            asset = m_scene.AssetService.Get(uuid.ToString());
+                            if (asset != null)
+                                m_log.DebugFormat("[HGScene]: Posting {0} {1}", asset.Type.ToString(), asset.Name);
+                            else
+                                m_log.DebugFormat("[HGScene]: Could not find asset {0}", uuid);
+                            PostAsset(userAssetURL, asset);
                         }
 
-                        if (success)
+                        if (ids.Count > 0) // maybe it succeeded...
                             m_log.DebugFormat("[HGScene]: Successfully posted item {0} to remote asset server {1}", assetID, userAssetURL);
                         else
                             m_log.WarnFormat("[HGScene]: Could not post asset {0} to remote asset server {1}", assetID, userAssetURL);
diff --git a/OpenSim/Region/Framework/Scenes/Hypergrid/HGScene.Inventory.cs b/OpenSim/Region/Framework/Scenes/Hypergrid/HGScene.Inventory.cs
index 4fde9bb..da9b5ec 100644
--- a/OpenSim/Region/Framework/Scenes/Hypergrid/HGScene.Inventory.cs
+++ b/OpenSim/Region/Framework/Scenes/Hypergrid/HGScene.Inventory.cs
@@ -113,24 +113,28 @@ namespace OpenSim.Region.Framework.Scenes.Hypergrid
                                                    UUID RayTargetID, byte BypassRayCast, bool RayEndIsIntersection,
                                                    bool RezSelected, bool RemoveItem, UUID fromTaskID, bool attachment)
         {
-            CachedUserInfo userInfo = CommsManager.UserProfileCacheService.GetUserDetails(remoteClient.AgentId);
-            if (userInfo != null)
+            //m_log.DebugFormat("[HGScene] RezObject itemID={0} fromTaskID={1}", itemID, fromTaskID);
+
+            if (fromTaskID.Equals(UUID.Zero))
             {
-                if (userInfo.RootFolder != null)
+                CachedUserInfo userInfo = CommsManager.UserProfileCacheService.GetUserDetails(remoteClient.AgentId);
+                if (userInfo != null)
                 {
-                    InventoryItemBase item = userInfo.RootFolder.FindItem(itemID);
-
-                    if (item == null)
-                    { // Fetch the item
-                        item = new InventoryItemBase();
-                        item.Owner = remoteClient.AgentId;
-                        item.ID = itemID;
-                        item = m_assMapper.Get(item, userInfo.RootFolder.ID, userInfo);
-                    }
-                    if (item != null)
+                    if (userInfo.RootFolder != null)
                     {
-                        m_assMapper.Get(item.AssetID, remoteClient.AgentId);
-                        
+                        InventoryItemBase item = userInfo.RootFolder.FindItem(itemID);
+                        if (item == null)
+                        { // Fetch the item
+                            item = new InventoryItemBase();
+                            item.Owner = remoteClient.AgentId;
+                            item.ID = itemID;
+                            item = m_assMapper.Get(item, userInfo.RootFolder.ID, userInfo);
+                        }
+                        if (item != null)
+                        {
+                            m_assMapper.Get(item.AssetID, remoteClient.AgentId);
+
+                        }
                     }
                 }
             }
diff --git a/OpenSim/Region/Framework/Scenes/Hypergrid/HGUuidGatherer.cs b/OpenSim/Region/Framework/Scenes/Hypergrid/HGUuidGatherer.cs
new file mode 100644
index 0000000..e83478d
--- /dev/null
+++ b/OpenSim/Region/Framework/Scenes/Hypergrid/HGUuidGatherer.cs
@@ -0,0 +1,29 @@
+using System;
+using System.Collections.Generic;
+
+using OpenSim.Framework;
+using OpenSim.Services.Interfaces;
+using OpenMetaverse;
+
+namespace OpenSim.Region.Framework.Scenes.Hypergrid
+{
+    public class HGUuidGatherer : UuidGatherer
+    {
+        protected string m_assetServerURL;
+        protected HGAssetMapper m_assetMapper;
+
+        public HGUuidGatherer(HGAssetMapper assMap, IAssetService assetCache, string assetServerURL) : base(assetCache)
+        {
+            m_assetMapper = assMap;
+            m_assetServerURL = assetServerURL;
+        }
+
+        protected override AssetBase GetAsset(UUID uuid)
+        {
+            if (string.Empty == m_assetServerURL)
+                return m_assetCache.Get(uuid.ToString());
+            else
+                return m_assetMapper.FetchAsset(m_assetServerURL, uuid);
+        }
+    }
+}
diff --git a/OpenSim/Region/Framework/Scenes/UuidGatherer.cs b/OpenSim/Region/Framework/Scenes/UuidGatherer.cs
index 5eb42f7..1324978 100644
--- a/OpenSim/Region/Framework/Scenes/UuidGatherer.cs
+++ b/OpenSim/Region/Framework/Scenes/UuidGatherer.cs
@@ -187,7 +187,7 @@ namespace OpenSim.Region.Framework.Scenes
         /// </summary>
         /// <param name="uuid"></param>
         /// <returns></returns>
-        protected AssetBase GetAsset(UUID uuid)
+        protected virtual AssetBase GetAsset(UUID uuid)
         {
             m_waitingForObjectAsset = true;
             m_assetCache.Get(uuid.ToString(), this, AssetReceived);