Merge pull request #15 from Mostey/tbm_treasures

Retrieving Treasures bound to the user account

Author: Mostey

src/epvpapi/Connection/ProfileSession.cs

@@ -1,4 +1,5 @@
-using HtmlAgilityPack;
+using epvpapi.TBM;
+using HtmlAgilityPack;
 using System;
 using System.Collections.Generic;
 using System.Linq;
@@ -181,6 +182,95 @@ public List<PrivateMessage> GetPrivateMessages(uint firstPage, uint pageCount, P
                 return fetchedMessages;
             }
 
+
+            /// <summary>
+            /// Retrieves all <c>Treasure</c>s that have been bought and/or sold using the logged-in user account
+            /// </summary>
+            /// <param name="queryStatus">
+            /// Type of <c>Treasure</c> to query. Either <c>Treasure.Query.SoldListed</c> 
+            /// for querying treasures that have been sold/listed or <c>Treasure.Query.Bought</c> 
+            /// for treasure that have been bought </param>
+            /// <param name="pageCount"> Amount of pages to retrieve, one page may contain up to 15 treasures </param>
+            /// <param name="startIndex"> Index indcating the page to start from </param>
+            /// <returns> List of all <c>Treasure</c>s that could be retrieved </returns>
+            public List<Treasure> GetTreasures(Treasure.Query queryStatus = Treasure.Query.SoldListed, uint pageCount = 1, uint startIndex = 1)
+            {
+                Session.ThrowIfInvalid();
+
+                var listedTreasures = new List<Treasure>();
+                for (var i = startIndex; i < (startIndex + pageCount); ++i)
+                {
+                    var res = Session.Get("http://www.elitepvpers.com/theblackmarket/treasures/" +
+                                         ((queryStatus == Treasure.Query.Bought) ?  "bought" : "soldunsold") 
+                                         + "/" + i);
+                    var htmlDocument = new HtmlDocument();
+                    htmlDocument.LoadHtml(res.ToString());
+
+                    var rootFormNode = htmlDocument.GetElementbyId("contentbg");
+                    if (rootFormNode == null) continue;
+
+                    var tableNode = rootFormNode.SelectSingleNode("table[1]/tr[1]/td[1]/table[1]/tr[2]/td[1]/div[1]/div[3]/table[1]");
+                    if (tableNode == null) continue;
+
+                    // skip the first <tr> element since that is the table header
+                    foreach (var treasureListingNode in tableNode.GetElementsByTagName("tr").Skip(1))
+                    {
+                        var idNode = treasureListingNode.SelectSingleNode("td[1]");
+                        var titleNode = treasureListingNode.SelectSingleNode("td[2]");
+                        var costNode = treasureListingNode.SelectSingleNode("td[3]");
+                        var opponentNode = treasureListingNode.SelectSingleNode("td[4]");
+                        var listedTreasure = new Treasure
+                        {
+                            // first column is the id with a trailing #
+                            ID = (idNode != null) ? Convert.ToUInt32(idNode.InnerText.TrimStart('#')) : 0,
+
+                            // second column is the treasure title
+                            Title = (titleNode != null) ? titleNode.InnerText : "",
+                        };
+
+                        // since this function is only available for logged-in users, the seller (or buyer, depends on the request) is automatically the logged-in user
+                        if (queryStatus == Treasure.Query.Bought)
+                            listedTreasure.Buyer = User;
+                        else
+                            listedTreasure.Seller = User;
+
+                        // third column is the cost
+                        var match = new Regex(@"([0-9]+) eg").Match(costNode.InnerText);
+                        if (match.Groups.Count > 1)
+                            listedTreasure.Cost = Convert.ToUInt32(match.Groups[1].Value);
+
+                        // the last column is the treasure buyer or seller
+                        if (opponentNode != null)
+                        {
+                            opponentNode = opponentNode.SelectSingleNode("a[1]");
+                            if (opponentNode != null)
+                            {
+                                var opponent = opponentNode.Attributes.Contains("href")
+                                                ? new User(opponentNode.InnerText,
+                                                    epvpapi.User.FromURL(opponentNode.Attributes["href"].Value))
+                                                : new User();
+
+                                if (queryStatus == Treasure.Query.Bought)
+                                {
+                                    listedTreasure.Seller = opponent;
+                                    listedTreasure.Available = false;
+                                }
+                                else
+                                {
+                                    listedTreasure.Buyer = opponent;
+                                    listedTreasure.Available = false;
+                                }
+                            }
+                        }
+
+                        listedTreasures.Add(listedTreasure);
+                    }
+                }
+
+                return listedTreasures;
+            }
+
+
             /// <summary>
             /// Removes/disables the current Avatar of the <c>User</c>
             /// </summary>

src/epvpapi/TBM/Treasure.cs

@@ -15,6 +15,19 @@ namespace epvpapi.TBM
     /// </summary>
     public class Treasure : UniqueObject, IUniqueWebObject, IDefaultUpdatable, IDeletable
     {
+        public enum Query
+        {
+            /// <summary>
+            /// If the <c>Treasure</c> was bought, basically the same as <c>SoldListed</c>
+            /// </summary>
+            Bought,
+
+            /// <summary>
+            /// If the <c>Treasure</c> was sold and/or listed
+            /// </summary>
+            SoldListed
+        }
+
         /// <summary>
         /// Title of the treasure, visible for everyone
         /// </summary>
@@ -61,17 +74,16 @@ public class Treasure : UniqueObject, IUniqueWebObject, IDefaultUpdatable, IDele
         public DateTime PurchaseDate { get; set; }
 
         /// <summary>
-        /// If the <c>Treasure</c> was already sold to someone, this value is set to true
+        /// If the <c>Treasure</c> was already sold to someone, this value is set to fakse
         /// </summary>
-        public bool Sold { get; set; }
-
+        public bool Available { get; set; }
 
         public Treasure(string title, string content, uint cost):
             this(0, title, content, cost)
         { }
 
 
-        public Treasure(uint id, string title = null, string content = null, uint cost = 0):
+        public Treasure(uint id = 0, string title = null, string content = null, uint cost = 0):
             base(id)
         {
             Title = title;
@@ -81,6 +93,7 @@ public Treasure(uint id, string title = null, string content = null, uint cost =
             Buyer = new User();
             CreationDate = new DateTime();
             PurchaseDate = new DateTime();
+            Available = true;
         }
 
         /// <summary>
@@ -185,7 +198,7 @@ public void Update(Session session)
                     }
                     else if (key == "Purchase date:")
                     {
-                        Sold = true;
+                        Available = false;
                         var countdownNode = valueNode.SelectSingleNode("div[1]");
                         PurchaseDate = (countdownNode != null) ? countdownNode.InnerText.ToElitepvpersDateTime() : new DateTime();
                     }