set proxy header

Author: github0null

.gitignore

@@ -1,7 +1,6 @@
 /node_modules
 /out
 /out.min
-src/Telemetry/
 vsc-extension-quickstart.md
 *.vsix
 .EIDE
@@ -12,11 +11,18 @@ vsc-extension-quickstart.md
 pack.js
 test.js
 *.js.map
-**/data/config.json
-**/data/sln.record
 *.cpuprofile
 package-lock.json
 
+# private
+/src/Telemetry
+/src/Private
+
+# eide vars
+/res/data/config.json
+/res/data/sln.record
+/res/data/data.yaml
+
 # internal template projects
 res/template/projects
 

src/ResManager.ts

@@ -140,6 +140,10 @@ export class ResManager extends events.EventEmitter {
         return resManager;
     }
 
+    static instance(): ResManager {
+        return ResManager.GetInstance();
+    }
+
     static getLocalCodePage(): string | undefined {
         return codePage;
     }
@@ -223,6 +227,29 @@ export class ResManager extends events.EventEmitter {
         return this.appConfig;
     }
 
+    getAppUsrData(): { [key: string]: any } | undefined {
+
+        const dataFile = File.fromArray([this.GetAppDataDir().path, 'data.yaml']);
+
+        if (dataFile.IsFile()) {
+            try {
+                return yaml.parse(dataFile.Read());
+            } catch (error) {
+                // nothing todo
+            }
+        }
+    }
+
+    setAppUsrData(key: string, val: any): void {
+
+        let data = this.getAppUsrData() || {};
+
+        data[key] = val;
+
+        File.fromArray([this.GetAppDataDir().path, 'data.yaml'])
+            .Write(yaml.stringify(data));
+    }
+
     //=====================
 
     getCache(name: string): FileCacheInfo | undefined {

src/extension.ts

@@ -259,14 +259,12 @@ export async function activate(context: vscode.ExtensionContext) {
     // load project in this workspace
     projectExplorer.loadWorkspace();
 
+    // hook
+    postLaunchHook(context);
+
     // launch done
     GlobalEvent.emit('extension_launch_done');
     GlobalEvent.emit('globalLog', newMessage('Info', 'Embedded IDE launch done'));
-
-    // refresh external tools now
-    ResInstaller.instance().refreshExternalToolsIndex().catch(err => {
-        GlobalEvent.emit('globalLog', ExceptionToMessage(err, 'Warning'));
-    });
 }
 
 // this method is called when your extension is deactivated
@@ -275,6 +273,32 @@ export function deactivate() {
     StatusBarManager.getInstance().disposeAll();
 }
 
+function postLaunchHook(extensionCtx: vscode.ExtensionContext) {
+
+    const resManager = ResManager.instance();
+    const appUsrData = resManager.getAppUsrData() || {};
+    const isFirstLaunch = appUsrData['InstallTime'] == undefined;
+
+    // is first launch ?
+    if (isFirstLaunch) {
+
+        // setup install time
+        resManager.setAppUsrData('InstallTime', Date.now().toString());
+
+        // only enable github proxy for GMT+8:00 by default
+        const timeZone = Math.floor((new Date().getTimezoneOffset() / 60) * -1);
+        if (timeZone != 8) {
+            // disable settings: 'EIDE.Repository.UseProxy'
+            SettingManager.GetInstance().setConfigValue('Repository.UseProxy', false);
+        }
+    }
+
+    // refresh external tools now
+    ResInstaller.instance().refreshExternalToolsIndex().catch(err => {
+        GlobalEvent.emit('globalLog', ExceptionToMessage(err, 'Warning'));
+    });
+}
+
 //////////////////////////////////////////////////
 // internal vsc-commands funcs
 //////////////////////////////////////////////////
@@ -704,14 +728,16 @@ async function tryInstallBinaries(binFolder: File, binVersion: string): Promise<
 
 let isEnvSetuped: boolean = false;
 
-function exportEnvToSysPath() {
+let isExport2ExtensionCtx = false;
+
+function exportEnvToSysPath(context?: vscode.ExtensionContext) {
 
     const settingManager = SettingManager.GetInstance();
     const resManager = ResManager.GetInstance();
     const builderFolder = resManager.getBuilderDir();
 
     // export some eide binaries path to system env path
-    const defEnvPath: string[] = [
+    const systemEnvPaths: string[] = [
         File.normalize(`${builderFolder.path}/bin`), // builder bin folder
         File.normalize(`${builderFolder.path}/utils`), // utils tool folder
         File.normalize(`${builderFolder.dir}/scripts`),
@@ -802,22 +828,20 @@ function exportEnvToSysPath() {
         }
     });
 
+    // append all tools to system env paths
+    pathList
+        .filter((env) => File.IsDir(env.path))
+        .forEach(envInfo => {
+            systemEnvPaths.push(envInfo.path);
+            if (envInfo.extraPath) {
+                envInfo.extraPath.forEach(p => systemEnvPaths.push(p));
+            }
+        });
+
     /* append to System Path if we not */
     if (isEnvSetuped == false) {
-
-        // append all tools env paths
-        pathList
-            .filter((env) => File.IsDir(env.path))
-            .forEach(envInfo => {
-                defEnvPath.push(envInfo.path);
-                if (envInfo.extraPath) {
-                    envInfo.extraPath.forEach(p => defEnvPath.push(p));
-                }
-            });
-
-        // apply to system env path
-        platform.prependToSysEnv(process.env, defEnvPath);
         isEnvSetuped = true;
+        platform.prependToSysEnv(process.env, systemEnvPaths);
     }
 
     /* update env key value */
@@ -833,6 +857,24 @@ function exportEnvToSysPath() {
     // 你可通过使用喜欢的 shell 将 DOTNET_CLI_TELEMETRY_OPTOUT 环境变量设置为 "1" 或 "true" 来选择退出遥测。
     // 阅读有关 .NET CLI 工具遥测的更多信息: https://aka.ms/dotnet-cli-telemetry
     process.env['DOTNET_CLI_TELEMETRY_OPTOUT'] = '1'; // disable telemetry
+
+    //
+    // export to vscode extension envs
+    //
+    if (context && !isExport2ExtensionCtx) {
+
+        isExport2ExtensionCtx = true;
+
+        context.environmentVariableCollection.persistent = false;
+
+        context.environmentVariableCollection.prepend('DOTNET_CLI_TELEMETRY_OPTOUT', '1');
+
+        for (const env of pathList) {
+            if (File.IsDir(env.path)) {
+                context.environmentVariableCollection.append(env.key, env.path);
+            }
+        }
+    }
 }
 
 async function checkAndInstallRuntime() {
@@ -1033,15 +1075,15 @@ async function InitComponents(context: vscode.ExtensionContext): Promise<boolean
     // register telemetry hook if user enabled
     try {
         if (settingManager.isEnableTelemetry()) {
-            const TelemetryTask = require('./Telemetry/TelemetryTask');
+            const TelemetryTask = require('./Private/TelemetryTask');
             TelemetryTask.registerTelemetryHook();
         }
     } catch (error) {
         // ignore error
     }
 
     // set some toolpath to env
-    exportEnvToSysPath();
+    exportEnvToSysPath(context);
 
     // init status bar
     {

src/utility.ts

@@ -288,21 +288,21 @@ export function wrapCommand(cmds: string[]): string {
     }).join(' ');
 }
 
-export function md5(str: string): string {
+export function md5(str_or_buff: string | Buffer): string {
     const md5 = crypto.createHash('md5');
-    md5.update(str);
+    md5.update(str_or_buff);
     return md5.digest('hex');
 }
 
-export function sha256(str: string): string {
+export function sha256(str_or_buff: string | Buffer): string {
     const md5 = crypto.createHash('sha256');
-    md5.update(str);
+    md5.update(str_or_buff);
     return md5.digest('hex');
 }
 
-export function sha1(str: string): string {
+export function sha1(str_or_buff: string | Buffer): string {
     const md5 = crypto.createHash('sha1');
-    md5.update(str);
+    md5.update(str_or_buff);
     return md5.digest('hex');
 }
 
@@ -372,8 +372,15 @@ export function redirectHost(url: string) {
     return url;
 }
 
-export function setProxyHeader(headers: { [key: string]: string }): { [key: string]: string } {
-    // TODO
+export function setProxyHeader(headers: { [key: string]: string | undefined }): { [key: string]: string | undefined } {
+
+    try {
+        const proxyUtils = require('./Private/GithubProxy');
+        proxyUtils.setProxyHeader(headers);
+    } catch (error) {
+        // ignore error
+    }
+
     return headers;
 }