o
    HiI                     @   sF   d dl Z d dlZd dlmZ d dlmZ dd ZdddZd	d
 ZdS )    N)custom_util)oc_utilc                   C   s   t jddkS )u<   是否处于批处理模式（服务端调用，无交互）OC_BATCH_MODE1)osenvironget r	   r	   H/data/storage/obfuscation/tool/ObjectiveC/oc_custom/custom_get_params.py_batch_mode   s   r    c                 C   s<   t  rtj| | }td|  d|  |S t| S )uD   批处理模式下从环境变量读取，交互模式下用 input()u   [批处理] =)r   r   r   r   stripprintinput)env_nameprompt	validatordefaultvaluer	   r	   r
   _batch_input   s
   r   c               	      s  	 t dddd} | dkrdt_d} n| dv r| t_ntd t r(td	|  qtd
tj  tjdkr8d S t rDtj	
dd nd}|rK|}n!d}ttddD ]}|ttj  }|ttj  }qU|t_td|  	 t dddd}|dkrdt_nt|dkr|t_ntd t rtd| qwd}| dkrg d d}n
| dkrg d d}d}t D ]\}}	|| d|	 d7 }q|d7 }t d|dd}|dkr|}td|  ndd |d D }
|
r<td!d" |
D r<z+d#d |
D } fd$d|D }|s|}td%|  nd |}td&|  W nT ttfy;   |}td'|  Y n@w g }|
D ]!} D ]}	|	d(d) }||	ksW||kr^||	  nqDq@|rrd |}td&|  n	|}td%|  |t_d*|v rdt_d S d+t_d S ),NTOC_SDK_REGIONu5   请输入需要国内还是海外
1.国内
2.海外

2)r   r   )r   r   u   输入无效，请输入1或2u   OC_SDK_REGION 无效值: u   当前工程名: 
XXGPlayKitOC_NEW_PROJECT_NAME   u   新工程名: 
OC_STARTIDu    请输入初始化id(可为空)
    u4   输入无效，请输入32位的初始化id或留空u   OC_STARTID 无效值: r   )u   ShanYanSDK(一键登录)u   BDASignalManager(巨量归因)u   CSJ(激励广告)u   ShareSDK(分享)zShanYanSDK,BDASignalManager)	AppsFlyerFacebookFirebaseAdjustu   VK(俄罗斯VK登录)u   AppLovin(Max广告)u   Poopo(渠道)u   ADP(IAA项目)zAppsFlyer,Facebook,FirebaseuF   请选择需要包含的SDK（多选请用逗号分隔，如：0,1）
.
OC_SDK_OPTIONSu   默认选择: c                 S   s   g | ]
}|  r|  qS r	   )r   .0tr	   r	   r
   
<listcomp>j   s    zget_params.<locals>.<listcomp>,c                 s   s    | ]}|  V  qd S )N)isdigitr%   r	   r	   r
   	<genexpr>k   s    zget_params.<locals>.<genexpr>c                 S   s   g | ]}t |qS r	   )intr&   idxr	   r	   r
   r(   n   s    c                    s0   g | ]}d |  krt  k rn n | qS )r   )lenr-   sdk_listr	   r
   r(   o   s    u    无有效选择，默认选择: u   已选择: u#   输入格式有误，默认选择: (r   AppLovinF)r   r   sdk_cn_or_osr   r   
ValueErrorr   name_current_projectr   r   r   r   rangerandomrandintchoicelist_random_words
capitalizenew_project_namestartidr/   	enumeratesplitalljoin
IndexErrorappendsdk_options	need_pods)r4   override_namename_r>   rE   default_sdksdk_options_promptisdktokensselected_indicesselected_sdksvalid_namesr'   sdk_baser	   r0   r
   
get_params   s   







rS   )Nr   )	r8   r   ObjectiveC.oc_customr   
ObjectiveCr   r   r   rS   r	   r	   r	   r
   <module>   s    
	