MLFullTextSearch.m

/*
 
 Copyright (c) 2017 Max Lungarella <cybrmx@gmail.com>
 
 Created on 28/04/2017.
 
 This file is part of AmiKo for OSX.
 
 AmiKo for OSX is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.
 
 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 GNU General Public License for more details.
 
 You should have received a copy of the GNU General Public License
 along with this program. If not, see <http://www.gnu.org/licenses/>.
 
 ------------------------------------------------------------------------ */

#import "MLFullTextSearch.h"
#import "MLMedication.h"

@implementation MLFullTextSearch
{
    // Instance variable declarations go here
    NSArray *mListOfArticles;
    NSDictionary *mDict;
}

/** Properties
 */
#pragma mark properties

@synthesize listOfSectionIds;
@synthesize listOfSectionTitles;

/** Instance functions
 */
#pragma mark public methods

- (NSString *) tableWithArticles:(NSArray *)listOfArticles
              andRegChaptersDict:(NSDictionary *)dict
                       andFilter:(NSString *)filter
{
    int rows = 0;
    NSMutableDictionary *chaptersCountDict = [[NSMutableDictionary alloc] init];
    NSString *htmlStr = @"<ul>";

    // Assign list and dictionaries only if != nil
    if (listOfArticles!=nil) {
        mListOfArticles = listOfArticles;
        // Sort alphabetically (this is pretty neat!)
        NSSortDescriptor *sort = [NSSortDescriptor sortDescriptorWithKey:@"title" ascending:YES];
        mListOfArticles = [mListOfArticles sortedArrayUsingDescriptors:@[sort]];
    }
    if (dict!=nil)
        mDict = dict;
    
    // Loop through all articles
    for (MLMedication *m in mListOfArticles) {
        BOOL filtered = true;
        NSString *contentStyle;
        NSString *firstLetter = [[m.title substringToIndex: 1] uppercaseString];
        if (rows % 2 == 0)
            contentStyle = [NSString stringWithFormat:@"<li style=\"background-color:var(--background-color-gray);\" id=\"%@\">", firstLetter];
        else
            contentStyle = [NSString stringWithFormat:@"<li style=\"background-color:transparent;\" id=\"%@\">", firstLetter];

        NSString *contentChapters = @"";
        NSArray *regnrs = [m.regnrs componentsSeparatedByString:@","];
        NSDictionary *indexToTitlesDict = [m indexToTitlesDict];    // id -> chapter title

        NSString *anchor;
        // List of chapters
        if ([regnrs count]>0) {
            NSString *r = [regnrs objectAtIndex:0];
            if ([mDict objectForKey:r]) {
                NSSet *chapters = mDict[r];
                for (NSString *c in chapters) {
                    if ([indexToTitlesDict objectForKey:c]) {
                        NSString *cStr = indexToTitlesDict[c];
                        NSString *sectionNamePrefix;
                        if ([c intValue] > 100)
                            sectionNamePrefix = @"Section";
                        else
                            sectionNamePrefix = @"section";

                        anchor = [NSString stringWithFormat:@"%@%@", sectionNamePrefix, c];

                        int count = 0;
                        if ([chaptersCountDict objectForKey:cStr])
                            count = [chaptersCountDict[cStr] intValue];
                        chaptersCountDict[cStr] = [NSNumber numberWithInt:count+1];
                        if ([filter length]==0 || [filter isEqualToString:cStr]) {
                            contentChapters = [contentChapters stringByAppendingFormat:@"<span style=\"font-size:0.75em; color:#0088BB\"> <a onclick=\"displayFachinfo('%@','%@')\">%@</a></span><br>", m.regnrs, anchor, cStr];                            
                            filtered = false;
                        }
                    }
                }
            }
        }
        NSString *contentTitle = [NSString stringWithFormat:@"<a onclick=\"displayFachinfo('%@','%@')\"><span style=\"font-size:0.8em\"><b>%@</b></span></a> <span style=\"font-size:0.7em\"> | %@</span><br>", m.regnrs, anchor, m.title, m.auth];
        if (!filtered) {
            htmlStr = [htmlStr stringByAppendingFormat:@"%@%@%@</li>", contentStyle, contentTitle, contentChapters];
            rows++;
        }
    }
    
    htmlStr = [htmlStr stringByAppendingFormat:@"</ul>"];
    
    NSMutableArray *listOfIds = [[NSMutableArray alloc] init];
    NSMutableArray *listOfTitles = [[NSMutableArray alloc] init];
    for (NSString *cStr in chaptersCountDict) {
        [listOfIds addObject:cStr];
        [listOfTitles addObject:[NSString stringWithFormat:@"%@ (%@)", cStr, chaptersCountDict[cStr]]];
    }
    // Update section ids (anchors)
    listOfSectionIds = [NSArray arrayWithArray:listOfIds];
    // Update section titles
    listOfSectionTitles = [NSArray arrayWithArray:listOfTitles];
    
    return htmlStr;
}

@end