Added Label Primary Field w/ the integration of all related titles fo…

…r easier searching of topics via phonetic and common spellings.
benjamin-stern · Oct 16, 2020 · 0740b76 · 0740b76
1 parent bce695d
commit 0740b76
Show file tree

Hide file tree

Showing 3 changed files with 96 additions and 0 deletions.
diff --git a/Converter/MainWindow.xaml.cs b/Converter/MainWindow.xaml.cs
@@ -122,6 +122,9 @@ private void ConversionLogic() {
             _serviceSQLite.AddAsync(topic);
             _serviceSQLite.SaveChanges();
 
+            _serviceMongo.UpdateTopicsWithIndexData(_serviceSQLite);
+            _serviceSQLite.SaveChanges();
+
             //Processing Texts
             var totalTexts = Total = _serviceMongo.TextsCount();
             Log($"Processing: Texts #{totalTexts}");

diff --git a/Converter/Model/SQLite/Label.cs b/Converter/Model/SQLite/Label.cs
@@ -28,5 +28,7 @@ public class Label
         public Language Language { get; set; }
 
         public string Text { get; set; }
+
+        public bool Primary { get; set; }
     }
 }
diff --git a/Converter/SefariaMongoDBService.cs b/Converter/SefariaMongoDBService.cs
@@ -21,6 +21,7 @@ class SefariaMongoDBService
     {
         private IMongoCollection<BsonDocument> _texts;
         private IMongoCollection<BsonDocument> _summaries;
+        private IMongoCollection<BsonDocument> _index;
         private IMongoCollection<BsonDocument> _links;
 
         private MongoClient _client;
@@ -32,6 +33,7 @@ public SefariaMongoDBService()
 
             _texts = database.GetCollection<BsonDocument>("texts");
             _summaries = database.GetCollection<BsonDocument>("summaries");
+            _index = database.GetCollection<BsonDocument>("index");
             _links = database.GetCollection<BsonDocument>("links");
 
             //var amount = _texts.CountDocuments(new BsonDocument());
@@ -41,9 +43,98 @@ public Topic GetSummaryTopics()
         {
             //Topic root = new Topic() { Name = "toc" };
             Topic root = GetNestedTopics(_summaries.Find(_ => true).FirstOrDefault());
+
+
+            //root.
+
             return root;
         }
 
+        public void UpdateTopicsWithIndexData(SefariaSQLiteConversionContext targetContext) {
+
+            //TODO: Need to parse Index and retrieve all alternative names with primary prioritizations
+            var indexedList = _index.Find(_ => true).ToList();
+            var topics = targetContext.Topics.Where(_ => true).ToList();
+
+            for (int j = 0; j < indexedList.Count; j++)
+            {
+                var index = indexedList[j];
+                //for (int i = 0; i < index.Elements.Count(); i++)
+                //{
+                    //var element = index.GetElement(i);
+                    var schema = index.GetElement("schema").Value.AsBsonDocument;
+
+                    var titles = schema.GetValue("titles", null).AsBsonArray;
+                    var key = schema.GetValue("key", null).AsString;
+
+                    if (key != null) {
+                        var topic = topics.Where(t => t.Name == key).FirstOrDefault();
+                        if (topic != null) {
+                            for (int k = 0; k < titles.Count; k++)
+                            {
+                                var title = titles[k];
+                                var text = title.AsBsonDocument.GetValue("text", null).AsString;
+                                var lang = title.AsBsonDocument.GetValue("lang", null).AsString;
+                                var primary = title.AsBsonDocument.GetValue("primary", false).AsBoolean;
+
+                                if (text != null)
+                                {
+                                    Label label = null;
+                                    //topics.Where(t => t == topic).SelectMany(t => t.LabelGroup.Labels).FirstOrDefault(l => l.Text == text);
+                                    bool isAdded = false;
+                                    foreach (var l in topic.LabelGroup.Labels)
+                                    {
+                                        if (l.Text == text)
+                                        {
+                                            label = l;
+                                            break;
+                                        }
+                                    }
+
+                                    if (label == null)
+                                    {
+                                        isAdded = true;
+                                        label = new Label();
+                                        label.LabelGroupId = (int)topic.LabelGroupId;
+                                        label.Text = text;
+                                        label.LanguageId = GetLanguageIdByString(lang);
+                                    }
+
+                                    if (label != null)
+                                    {
+                                        label.Primary = primary;
+                                    }
+
+                                    if (isAdded)
+                                    {
+                                        targetContext.Add(targetContext.Labels, label);
+                                    }
+                                    else
+                                    {
+                                        targetContext.Labels.Update(label);
+                                    }
+                                }
+                            }
+                        }
+
+                    }
+
+                //}
+            }
+
+        }
+
+        private List<Topic> GetTopicList(Topic t) {
+            List<Topic> result = new List<Topic>();
+            result.Add(t);
+
+            foreach (var item in t.Children)
+            {
+                result.AddRange(GetTopicList(item));
+            }
+            return result;
+        }
+
         private Topic GetNestedTopics(BsonDocument document, Topic parent = null, int index = 1)
         {
             Topic instance = new Topic { Index = index };