#include "RunWebEngineViewManager.h"
#include "RunWebEngineView.h"
#include "Utils/models.h"
#include "Utils/database.h"
#include "constant.h"
#include "RunDialog.h"
#include "RunHelper.h"
#include "Task/TaskWebEngineProfile.h"
#include "TaskFlow/mFcModels.h"
#include <QWebEngineProfile>
#include <QWebEngineScriptCollection>
#include <QWebEngineScript>
#include <QWebEngineSettings>
#include <QWebEngineCookieStore>
#include <QWebEngineHttpRequest>
#include <QNetworkAccessManager>
#include <QWebEngineCookieStore>
#include <QLineEdit>
#include <QsLog.h>

RunWebEngineViewManager::RunWebEngineViewManager(RunDialog *runDialog) : QObject(runDialog){
 QLOG_INFO()<<"RunWebEngineViewManager::RunWebEngineViewManager()";
 this->mHelper = runDialog->getHelper();


 if(mHelper->getTask()->runParams.isTraceless){//开启无痕浏览器模式
 mProfile = new QWebEngineProfile(this);
 mProfile->setHttpCacheType(QWebEngineProfile::HttpCacheType::NoCache);
 mProfile->setHttpCacheMaximumSize(0);
 mProfile->setPersistentCookiesPolicy(QWebEngineProfile::PersistentCookiesPolicy::NoPersistentCookies);
 }else{
 mProfile = TaskWebEngineProfile::getInstance()->copyProfile();
 }


 mProfile->setHttpUserAgent(mHelper->getTask()->defaultUserAgent);

 QString sourceCode= QString(
 "let ej = document.createElement('script');ej.type = 'text/javascript';"
 "ej.appendChild(document.createTextNode('%1'));"
 "document.body.appendChild(ej);").arg(Database::getInstance()->read_file(JS_EXECUTE_FILE).replace("\r"," ").replace("\n"," ").replace("\t"," "));
 QWebEngineScript script;
 mProfile->scripts()->remove(script);
 script.setSourceCode(sourceCode);
 mProfile->scripts()->insert(script);
 QWebEngineSettings *settings = mProfile->settings();

// settings->setAttribute(QWebEngineSettings::LocalStorageEnabled, false); //关闭缓存
 settings->setAttribute(QWebEngineSettings::AllowRunningInsecureContent,mHelper->getTask()->isBrowserAllowRunningInsecureContent);
 settings->setAttribute(QWebEngineSettings::AutoLoadImages,mHelper->getTask()->isBrowserAutoLoadImages);
 settings->setAttribute(QWebEngineSettings::AutoLoadIconsForPage,mHelper->getTask()->isBrowserAutoLoadIconsForPage);
 settings->setAttribute(QWebEngineSettings::PluginsEnabled,mHelper->getTask()->isBrowserPluginsEnabled);


 connect(this,&RunWebEngineViewManager::cleanWebViews,this,&RunWebEngineViewManager::onCleanWebViews);
 connect(this,&RunWebEngineViewManager::webViewNotifyUrlChanged,runDialog,&RunDialog::onWebViewUrlChanged);
 connect(this,&RunWebEngineViewManager::cleanWebViews,runDialog,&RunDialog::onCleanWebViews);
 this->createWebView();

}
RunWebEngineViewManager::~RunWebEngineViewManager(){
 QLOG_INFO()<<"RunWebEngineViewManager::~RunWebEngineViewManager()";

 if(m_reusedWebView!=nullptr){
 delete m_reusedWebView;
 m_reusedWebView = nullptr;
 }
 for (int i = 0; i < mWebViews.length(); ++i) {
 delete mWebViews[i];
 mWebViews[i] = nullptr;
 }
 mWebViews.clear();

 if(mProfile){
 delete mProfile;
 mProfile = nullptr;
 }
}
QWebEngineProfile* RunWebEngineViewManager::profile(){
 return mProfile;
}
void RunWebEngineViewManager::setCurrentWebView(RunWebEngineView *webView){

 m_currentWebView = webView;
}
RunWebEngineView * RunWebEngineViewManager::getCurrentWebView(){

 return m_currentWebView;
}

RunWebEngineView * RunWebEngineViewManager::createWebView(){
 if(m_reusedWebView==nullptr){
 RunWebEngineView *webView = new RunWebEngineView(this,(QWidget*)this->parent());
 mWebViews.append(webView);

 emit cleanWebViews(webView);
 return webView;
 }else {
 RunWebEngineView *webView = m_reusedWebView;
 m_reusedWebView = nullptr;
 mWebViews.append(webView);

 emit cleanWebViews(webView);
 return webView;
 }
}

void RunWebEngineViewManager::onCleanWebViews(RunWebEngineView *webView){
 setCurrentWebView(webView);
 // 清理webView
 QVector<RunWebEngineView *> inUseWebViews;

 inUseWebViews.append(m_currentWebView);
 for (int i = 0; i < mHelper->loops.length(); ++i) {
 RunWebEngineView *webView = mHelper->loops[i]->carryCurrentWebView;
 if(webView!=nullptr){
 inUseWebViews.append(webView);
 }
 }
 for (int i = 0; i < mWebViews.length(); ++i) {
 RunWebEngineView *webView = mWebViews[i];

 bool isDelete = true;
 for (int j = 0; j < inUseWebViews.length(); ++j) {
 if(inUseWebViews.at(j)==webView){
 isDelete = false;
 break;
 }
 }
 if(isDelete){
 webView->page()->profile()->clearHttpCache();
 webView->page()->profile()->clearAllVisitedLinks();

 mWebViews.remove(i);
 if(m_reusedWebView==nullptr){
 m_reusedWebView = webView;
 }else {
 webView->disconnect();
 delete webView;
 webView = nullptr;
 }
 }
 }
}