@@ -739,6 +739,8 @@ export default (fastify, options, done) => {
739739 const query = request . query ; // 获取 query 参数
740740 const pwd = query . pwd || '' ;
741741 const sub_code = query . sub || '' ;
742+ const cat_sub_code = ENV . get ( 'cat_sub_code' , 'all' ) ;
743+ const must_sub_code = Number ( ENV . get ( 'must_sub_code' , '0' ) ) || 0 ;
742744 const cfg_path = request . params [ '*' ] ; // 捕获整个路径
743745 try {
744746 // 获取主机名,协议及端口
@@ -783,7 +785,7 @@ export default (fastify, options, done) => {
783785 // }
784786 const getFilePath = ( cfgPath , rootDir , fileName ) => path . join ( rootDir , `data/cat/${ fileName } ` ) ;
785787 const processContent = ( content , cfgPath , requestUrl , requestHost ) => {
786- const $config_url = requestUrl . replace ( cfgPath , `/1?sub=all &pwd=${ process . env . API_PWD || '' } ` ) ;
788+ const $config_url = requestUrl . replace ( cfgPath , `/1?sub=${ cat_sub_code } &pwd=${ process . env . API_PWD || '' } ` ) ;
787789 return content . replaceAll ( '$config_url' , $config_url ) . replaceAll ( '$host' , requestHost ) ;
788790 }
789791
@@ -832,10 +834,14 @@ export default (fastify, options, done) => {
832834 if ( sub_code ) {
833835 let subs = getSubs ( options . subFilePath ) ;
834836 sub = subs . find ( it => it . code === sub_code ) ;
835- // console.log('sub:', sub);
837+ console . log ( 'sub:' , sub ) ;
836838 if ( sub && sub . status === 0 ) {
837839 return reply . status ( 500 ) . send ( { error : `此订阅码:【${ sub_code } 】已禁用` } ) ;
840+ } else if ( ! sub && must_sub_code ) {
841+ return reply . status ( 500 ) . send ( { error : `此订阅码:【${ sub_code } 】不存在` } ) ;
838842 }
843+ } else if ( ! sub_code && must_sub_code ) {
844+ return reply . status ( 500 ) . send ( { error : `缺少订阅码参数` } ) ;
839845 }
840846
841847 const siteJSON = await generateSiteJSON ( options , requestHost , sub , pwd ) ;
0 commit comments