skycaiji/SkycaijiApp/admin/controller/Index.php

<?php
/*
 |--------------------------------------------------------------------------
 | SkyCaiji (蓝天采集器)
 |--------------------------------------------------------------------------
 | Copyright (c) 2018 http://www.skycaiji.com All rights reserved.
 |--------------------------------------------------------------------------
 | 使用协议  http://www.skycaiji.com/licenses
 |--------------------------------------------------------------------------
 */

namespace skycaiji\admin\controller;
use skycaiji\admin\model\User;
use skycaiji\admin\model\CacheModel;
class Index extends BaseController{
	public function indexAction(){
		return $this->fetch();
	}
	/*后台运行采集，会无限运行下去*/
	public function backstageAction(){
		if(empty($GLOBALS['config']['caiji']['auto'])){
			$this->error('请先开启自动采集');
		}
		if(!input('?autorun')){
			
			$this->error('参数错误');
		}
		
		
		$runtime=input('runtime/d');
		if(empty($runtime)){
			
			$runtime=time();
			cache('caiji_auto_backstage_runtime',$runtime);
		}else{
			
			$cahce_runtime=cache('caiji_auto_backstage_runtime');
			$cahce_runtime=intval($cahce_runtime);
			if($runtime<$cahce_runtime){
				
				$this->error('终止旧进程');
			}
		}
		ignore_user_abort(true);
		set_time_limit(0);
		
		$mconfig=new \skycaiji\admin\model\Config();
		$caijiConfig=$mconfig->where('cname','caiji')->find();
		$caijiConfig=$mconfig->convertData($caijiConfig);
		
		if(empty($caijiConfig['data']['auto'])||$caijiConfig['data']['run']!='backstage'){
			
			$this->error('自动采集已停止');
		}
		$collectTime1=time();
		try{
			@get_html(url('Admin/Api/collect?backstage=1',null,false,true),null,array('timeout'=>3));
		}catch(\Exception $ex){
			
		}
		
		sleep(20);
		
		if($GLOBALS['config']['caiji']['auto']){
			
			try{
				@get_html(url('Admin/Index/backstage?autorun=1&runtime='.$runtime,null,false,true),null,array('timeout'=>3));
			}catch(\Exception $ex){
			
			}
		}
		exit();
		
	}
	/*访问执行采集*/
	public function caijiAction(){
		if(empty($GLOBALS['config']['caiji']['auto'])){
			$this->error('请先开启自动采集','Admin/Setting/caiji');
		}
		@get_html(url('Admin/Api/collect?backstage=1',null,false,true),null,array('timeout'=>3));
		$waitTime=$GLOBALS['config']['caiji']['interval']*60;
		$waitTime=$waitTime>0?$waitTime:3;
		$this->success('正在采集...','Admin/Index/caiji',null,$waitTime);
	}
	/*任务api发布*/
	public function apiTaskAction(){
		controller('admin/Api','controller')->taskAction();
	}
	
	public function loginAction(){
		if(request()->isPost()){
			if(!check_usertoken()){
				$this->error(lang('usertoken_error'));
			}
			
			$mcacheLogin=CacheModel::getInstance('login');
			$config_login=$GLOBALS['config']['site']['login'];
			$clientIpMd5=md5(request()->ip());
			if(!empty($config_login['limit'])){
				
				$ipLoginData=$mcacheLogin->getCache($clientIpMd5,'data');
				if((NOW_TIME-$ipLoginData['lastdate'])<($config_login['time']*3600)&&$ipLoginData['failed']>=$config_login['failed']){
					
					$this->error("您已登录失败{$ipLoginData['failed']}次，被锁定{$config_login['time']}小时");
				}
			}
			if(input('post.sublogin')){
				$username=strtolower(trim(input('post.username')));
				$pwd=trim(input('post.password'));
				if($GLOBALS['config']['site']['verifycode']){
					
					$verifycode=trim(input('post.verifycode'));
					$check=check_verify($verifycode);
					if(!$check['success']){
						$this->error($check['msg']);
					}
				}
				
				
				$check=User::right_username($username);
				if(!$check['success']){
					$this->error($check['msg']);
				}
				$check=User::right_pwd($pwd);
				if(!$check['success']){
					$this->error($check['msg']);
				}
		
				$muser=new \skycaiji\admin\model\User();
				$userData=$muser->where('username',$username)->find();
				if(empty($userData)||$userData['password']!=pwd_encrypt($pwd)){
					
					if(!empty($config_login['limit'])){
						
						$ipLoginData=$mcacheLogin->getCache($clientIpMd5,'data');
						if(!empty($ipLoginData)){
							
							if((NOW_TIME-$ipLoginData['lastdate'])<($config_login['time']*3600)){
								
								$ipLoginData['failed']++;
							}else{
								
								$ipLoginData['lastdate']=NOW_TIME;
								$ipLoginData['failed']=1;
							}
						}else{
							
							$ipLoginData['lastdate']=NOW_TIME;
							$ipLoginData['failed']=1;
						}
						$ipLoginData['ip']=request()->ip();
						$mcacheLogin->setCache($clientIpMd5, $ipLoginData);
						$this->error(lang('user_error_login')."失败{$config_login['failed']}次将被锁定{$config_login['time']}小时，已失败{$ipLoginData['failed']}次");
					}
					$this->error(lang('user_error_login'));
				}
		
				if(input('post.auto')){
					
					cookie('login_history',$username.'|'.md5($username.$userData['password']),array('expire'=>3600*24*15));
				}
				session('user_id',$userData['uid']);
				
				$serverinfo=input('serverinfo');
				if(empty($serverinfo)){
					$this->success(lang('user_login_in'),'Admin/Backstage/index');
				}else{
					
					$this->success(lang('user_login_in'),null,array('js'=>'window.parent.postMessage("login_success","http://www.skycaiji.com");'));
				}
			}else{
				$this->error(lang('user_error_sublogin'));
			}
		}else{
    		return $this->fetch('index');
		}
	}
	/*退出*/
	public function logoutAction(){
		\think\Cookie::delete('login_history');
		unset($GLOBALS['user']);
		session('user_id',null);
		$this->success(lang('op_success'),'Admin/Index/index');
	}
	/*验证码*/
	public function verifyAction(){
		$config=array(
			'fontSize'=>30,	
			'length'=>3,	
			'fontttf'=>'5.ttf',
			'useCurve'=>true,
			'useNoise'=>true 
		);
		ob_clean();

		$captcha = new \think\captcha\Captcha($config);
		return $captcha->entry();
	}
	/*找回密码*/
	public function find_passwordAction(){
		$username=trim(input('post.username'));
		if(empty($username)){
			$username=trim(input('username'));
			$username=base64_decode($username);
		}
		
		$step=max(1,input('step/d',1));
		$stepSname='find_password_step.'.md5($username);

		$stepSession=session($stepSname);
		$muser=model('User');
		if($step>1){
			
			if(strcasecmp(('step'.$step),$stepSession['step'])!==0){
				$this->error(lang('find_pwd_error_step'),'Index/find_password');
			}
			
			if(empty($stepSession['user'])){
				$this->error(lang('find_pwd_error_none_user'));
			}
		}
		if(request()->isPost()){
			if(input('post.subForPwd')){
				if(empty($username)){
					$this->error(lang('find_pwd_error_username'));
				}
				if($step===1){
					
					if(!check_usertoken()){
						$this->error(lang('usertoken_error'));
					}
					if($GLOBALS['config']['site']['verifycode']){
						
						$verifycode=trim(input('verifycode'));
						$check=check_verify($verifycode);
						if(!$check['success']){
							$this->error($check['msg']);
						}
					}

					/*获取用户信息*/
					
					$username_is_email=false;
					$check=\skycaiji\admin\model\User::right_email($username);
					if($check['success']){
						$username_is_email=true;
					}
					if($username_is_email){
						
						$emailCount=$muser->where(array('email'=>$username))->count();
						if($emailCount<=0){
							$this->error(lang('find_pwd_error_none_email'));
						}elseif($emailCount>1){
							$this->error(lang('find_pwd_error_multiple_emails'));
						}else{
							$userData=$muser->where(array('email'=>$username))->find();
						}
					}else{
						
						$userData=$muser->where(array('username'=>$username))->find();
					}
					if(empty($userData)){
						$this->error(lang('find_pwd_error_none_user'));
					}
					$userData=$userData->toArray();
					
					session($stepSname,array('step'=>'step2','user'=>$userData));
					
					$this->success(lang('redirecting'),'Index/find_password?step=2&username='.rawurlencode(base64_encode($username)));
				}elseif($step===2){
					
					$yzm=trim(input('yzm'));
					
					$check=\skycaiji\admin\model\User::right_yzm($username, $yzm);
					if(!$check['success']){
						$this->error($check['msg']);
					}
					$stepSession['step']='step3';
					session($stepSname,$stepSession);

					$this->success(lang('redirecting'),'Index/find_password?step=3&username='.rawurlencode(base64_encode($username)));
				}elseif($step===3){
					$pwd=trim(input('password'));
					$repwd=trim(input('repassword'));
					$check=\skycaiji\admin\model\User::right_pwd($pwd);
					if(!$check['success']){
						$this->error($check['msg']);
					}
					$check=\skycaiji\admin\model\User::right_repwd($pwd,$repwd);
					if(!$check['success']){
						$this->error($check['msg']);
					}
					
					$muser->strict(false)->where(array('username'=>$stepSession['user']['username']))->update(array('password'=>pwd_encrypt($pwd)));
					session($stepSname,null);
					$this->success(lang('find_pwd_success'),'Admin/Index/index');
				}else{
					$this->error(lang('find_pwd_error_step'),'Index/find_password');
				}
			}else{
				$this->error(lang('find_pwd_error_post'));
			}
		}else{
			if($step===2){
				$emailStatus=array('success'=>false,'msg'=>'');
				if(empty($GLOBALS['config']['email'])){
					$emailStatus['msg']=lang('config_error_none_email');
				}else{
					$waitTime=60;
					$waitSname='send_yzm_wait';
					
					$passTime=abs(NOW_TIME-session($waitSname));
					if($passTime<=$waitTime){
						$emailStatus['msg']=lang('find_pwd_email_wait',array('seconds'=>$waitTime-$passTime));
					}else{
						$expire=config('yzm_expire');
						$minutes=floor($expire/60);
						$yzm=mt_rand(100000,999999);
						session($waitSname,NOW_TIME);
						$mailReturn=send_mail($GLOBALS['config']['email'], $stepSession['user']['email'], $stepSession['user']['username'],lang('find_pwd_email_subject'),lang('find_pwd_email_body',array('yzm'=>$yzm,'minutes'=>$minutes)));
						if($mailReturn===true){
							$yzmSname='send_yzm.'.md5($username);
							session(array('name'=>$yzmSname,'expire'=>$expire));
							session($yzmSname,array('yzm'=>$yzm,'time'=>NOW_TIME));
							$emailStatus['success']=true;
							$emailStatus['msg']=lang('find_pwd_sended',array('email'=>preg_replace('/.{2}\@/', '**@', $stepSession['user']['email'])));
						}else{
							$emailStatus['msg']=lang('find_pwd_email_failed').'<br>'.$mailReturn;
						}
					}
				}
				
				$this->assign('emailStatus',$emailStatus);
			}

			$this->assign('userData',$stepSession['user']);
			$this->assign('username',$username);
			$this->assign('step',$step);
			return $this->fetch();
		}
	}
	/*验证站点*/
	public function site_certificationAction(){
		$keyFile=cache('site_certification');
		$key=$keyFile['key'];
		if(abs(NOW_TIME-$keyFile['time'])>60){
			
			$key='';
		}
		exit($key);
	}
	/*客户端信息*/
	public function clientinfoAction(){
		return json(clientinfo());
	}
}