aboutsummaryrefslogtreecommitdiff
path: root/MediaBrowser.Controller/Localization/RatingsDefinition.cs
blob: 1162d2389f8a7aa38d0d5eb4e479930ef5839125 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
using MediaBrowser.Controller.Configuration;
using MediaBrowser.Model.Logging;
using System;
using System.Collections.Generic;
using System.IO;

namespace MediaBrowser.Controller.Localization
{
    /// <summary>
    /// Class RatingsDefinition
    /// </summary>
    public class RatingsDefinition
    {
        /// <summary>
        /// Initializes a new instance of the <see cref="RatingsDefinition" /> class.
        /// </summary>
        /// <param name="file">The file.</param>
        /// <param name="logger">The logger.</param>
        /// <param name="configurationManager">The configuration manager.</param>
        public RatingsDefinition(string file, IServerConfigurationManager configurationManager)
        {
            this.file = file;
            if (!Load())
            {
                Init(configurationManager.Configuration.MetadataCountryCode.ToUpper());
            }
        }

        /// <summary>
        /// Inits the specified country.
        /// </summary>
        /// <param name="country">The country.</param>
        protected void Init(string country)
        {
            //intitialze based on country
            switch (country)
            {
                case "US":
                    RatingsDict = new USRatingsDictionary();
                    break;
                case "GB":
                    RatingsDict = new GBRatingsDictionary();
                    break;
                case "NL":
                    RatingsDict = new NLRatingsDictionary();
                    break;
                case "AU":
                    RatingsDict = new AURatingsDictionary();
                    break;
                default:
                    RatingsDict = new USRatingsDictionary();
                    break;
            }
            Save();
        }

        /// <summary>
        /// The file
        /// </summary>
        readonly string file;

        /// <summary>
        /// Save to file
        /// </summary>
        public void Save()
        {
            // Use simple text serialization - no need for xml
            using (var fs = new StreamWriter(file))
            {
                foreach (var pair in RatingsDict)
                {
                    fs.WriteLine(pair.Key + "," + pair.Value);
                }
            }
        }

        /// <summary>
        /// Load from file
        /// </summary>
        /// <returns><c>true</c> if XXXX, <c>false</c> otherwise</returns>
        protected bool Load()
        {
            // Read back in our simple serialized format
            RatingsDict = new Dictionary<string, int>(StringComparer.OrdinalIgnoreCase);
            try
            {
                using (var fs = new StreamReader(file))
                {
                    while (!fs.EndOfStream)
                    {
                        var line = fs.ReadLine() ?? "";
                        var values = line.Split(',');
                        if (values.Length == 2)
                        {

                            int value;

                            if (int.TryParse(values[1], out value))
                            {
                                RatingsDict[values[0].Trim()] = value;
                            }
                            else
                            {
                                //Logger.Error("Invalid line in ratings file " + file + "(" + line + ")");
                            }
                        }
                    }
                }
            }
            catch
            {
                // Couldn't load - probably just not there yet
                return false;
            }
            return true;
        }

        /// <summary>
        /// The ratings dict
        /// </summary>
        public Dictionary<string, int> RatingsDict = new Dictionary<string, int>(StringComparer.OrdinalIgnoreCase);

    }
}