obsidian-douban/src/douban/data/handler/DoubanAbstractLoadHandler.ts

import { DoubanPluginSettings, PersonNameMode } from "src/douban/Douban";

import DoubanPlugin from "main";
import DoubanSubject from '../model/DoubanSubject';
import DoubanSubjectLoadHandler from "./DoubanSubjectLoadHandler";
import {Editor, moment, request, requestUrl, RequestUrlParam, sanitizeHTMLToDom} from "obsidian";
import { i18nHelper } from 'src/lang/helper';
import { log } from "src/utils/Logutil";
import {CheerioAPI, load} from "cheerio";

export default abstract class DoubanAbstractLoadHandler<T extends DoubanSubject> implements DoubanSubjectLoadHandler<T> {
    
    
    public doubanPlugin:DoubanPlugin;

    constructor(doubanPlugin:DoubanPlugin) {
        this.doubanPlugin = doubanPlugin;
    }

	parse(extract: T, settings:DoubanPluginSettings): string {
		let template:string = this.getTemplate(settings);
		let resultContent = template ? template
			.replaceAll("{{id}}", extract.id)
			.replaceAll("{{type}}", extract.type ? extract.type : "")
			.replaceAll("{{title}}", extract.title ? extract.title : "")
			.replaceAll("{{desc}}", extract.desc ? extract.desc : "")
			.replaceAll("{{image}}", extract.image  ? extract.image : "")
			.replaceAll("{{url}}", extract.url  ? extract.url : "")
			.replaceAll("{{score}}", extract.score  ? extract.score + "": "")
			.replaceAll("{{publisher}}", extract.publisher  ? extract.publisher : "")
			.replaceAll("{{datePublished}}", extract.datePublished ? moment(extract.datePublished).format(settings.dateFormat) : "")
			.replaceAll("{{timePublished}}", extract.datePublished ? moment(extract.datePublished).format(settings.timeFormat) : "")
			.replaceAll("{{genre}}", extract.genre  ? extract.genre.join(settings.arraySpilt) : "")
			: ""
		;
		return this.parseText(resultContent, extract, settings);
	}

	abstract getTemplate(settings:DoubanPluginSettings):string;

    abstract parseText(beforeContent:string, extract: T, settings:DoubanPluginSettings): string;

    abstract support(extract: DoubanSubject): boolean;
    
    handle(url:string, editor:Editor):void {
		let requestUrlParam:RequestUrlParam = {
			url: url,
			method: "GET",
			headers:  JSON.parse(this.doubanPlugin.settings.searchHeaders),
			throw: true
		};
		request(requestUrlParam)
			.then(e => {log.trace(e.toString());return e;})
            .then(load)
            .then(this.parseSubjectFromHtml)
            .then(content => this.toEditor(editor, content))
            // .then(content => content ? editor.replaceSelection(content) : content)
            .catch(e => log.error(i18nHelper.getMessage('130101')))
        ;

    }

    abstract parseSubjectFromHtml(data:CheerioAPI):T | undefined;
    
    toEditor(editor:Editor, extract: T):T {
        this.doubanPlugin.putToEditor(editor, extract);
        return extract;
    }

    getPersonName(name:string, settings:DoubanPluginSettings):string {
        if(!name || !settings || !settings.personNameMode) {
            return "";
        }
        let resultName:string = "";
		let regValue:RegExpExecArray;
        switch(settings.personNameMode) {
            case PersonNameMode.CH_NAME:
                regValue = /[\u4e00-\u9fa5]{2,20}/g.exec(name);
                resultName = regValue?regValue[0]:name;
                break;
            case PersonNameMode.EN_NAME:
                regValue = /[a-zA-Z.\s\-]{2,50}/g.exec(name);
                resultName = regValue?regValue[0]:name;
                break;
            default:
                resultName = name;
        }
        return resultName;
    }

     html_encode(str:string):string 
    { 
        let s = "";
        if (str.length == 0) return ""; 
        s = str.replace(/&/g, "&amp;"); 
        s = s.replace(/</g, "&lt;"); 
        s = s.replace(/>/g, "&gt;"); 
        s = s.replace(/ /g, "&nbsp;"); 
        s = s.replace(/\'/g, "&#39;"); 
        s = s.replace(/\"/g, "&quot;"); 
            s = s.replace(/\n/g, "<br/>"); 
        return s; 
    } 

     html_decode(str:string):string 
    { 
        let s = "";
        if (str.length == 0) return ""; 
        s = str.replace(/&amp;/g, "&"); 
        s = s.replace(/&lt;/g, "<"); 
        s = s.replace(/&gt;/g, ">"); 
        s = s.replace(/&nbsp;/g, " "); 
        s = s.replace(/&#39;/g, "\'"); 
        s = s.replace(/&quot;/g, "\""); 
        s = s.replace(/<br\/>/g, "\n"); 
        return s; 
    } 

}