Flow.Launcher/Flow.Launcher.Infrastructure/PinyinAlphabet.cs

using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using JetBrains.Annotations;
using Flow.Launcher.Infrastructure.Logger;
using Flow.Launcher.Infrastructure.Storage;
using Flow.Launcher.Infrastructure.UserSettings;
using ToolGood.Words.Pinyin;
using System.Threading.Tasks;

namespace Flow.Launcher.Infrastructure
{
    public interface IAlphabet
    {
        string Translate(string stringToTranslate);
    }

    public class PinyinAlphabet : IAlphabet
    {
        private ConcurrentDictionary<string, string> _pinyinCache = new ConcurrentDictionary<string, string>();
        private Settings _settings;

        public void Initialize([NotNull] Settings settings)
        {
            _settings = settings ?? throw new ArgumentNullException(nameof(settings));
        }


        public string Translate(string content)
        {
            if (_settings.ShouldUsePinyin)
            {
                if (!_pinyinCache.ContainsKey(content))
                {
                    if (WordsHelper.HasChinese(content))
                    {
                        var result = WordsHelper.GetPinyin(content, ";");
                        result = GetFirstPinyinChar(result) + result.Replace(";", "");
                        _pinyinCache[content] = result;
                        return result;
                    }
                    else
                    {
                        return content;
                    }
                }
                else
                {
                    return _pinyinCache[content];
                }
            }
            else
            {
                return content;
            }
        }

        private string GetFirstPinyinChar(string content)
        {
            return string.Concat(content.Split(';').Select(x => x.First()));
        }
    }
}