wxwork oauth2

Author: zhyale

proxy.config.json

@@ -1,16 +1,16 @@
 {
     "/janusec-admin/": {
-        "target": "http://192.168.100.107",
+        "target": "http://192.168.100.107:9080",
         "secure": false
     },
     "/janusec-admin/webssh": {
-        "target": "ws://192.168.100.107",
+        "target": "ws://192.168.100.107:9080",
         "secure": false,
         "ws": true,
         "changeOrigin": true
     },
     "/favicon.ico": {
-        "target": "http://192.168.100.107/janusec-admin/",
+        "target": "http://192.168.100.107:9080/janusec-admin/",
         "secure": false
     }
 }

src/app/application.service.ts

@@ -2,7 +2,7 @@ import { Injectable }    from '@angular/core';
 import { HttpClient, HttpHeaders } from '@angular/common/http';
 import { Observable, of } from 'rxjs';
 import { catchError, map, tap } from 'rxjs/operators';
-import { AuthUser, Application, Certificate, Domain, AppAdmin, VulnType, APIResponse, NodesKey, Node, LastRegexLogs, LastCCLogs } from './models';
+import { AuthUser, Application, Certificate, Domain, AppAdmin, VulnType, APIResponse, NodesKey, Node, LastRegexLogs, LastCCLogs, OAuthInfo } from './models';
 import { MessageService } from './message.service';
 
 const httpOptions = {
@@ -23,6 +23,7 @@ export class ApplicationService {
   vulntypemap: object = new(Object);
   lastRegexLogs: LastRegexLogs = new(LastRegexLogs);
   lastCCLogs: LastCCLogs = new (LastCCLogs);
+  oauth: OAuthInfo = new (OAuthInfo);
 
   constructor(private http: HttpClient,
     private messageService: MessageService) { }
@@ -54,6 +55,19 @@ export class ApplicationService {
     });
   }
 
+  getResponseByURL(url:string, callback:(obj: object)=>any, id?:number, obj?:object) {
+    // Get Request
+    this.http.get<APIResponse>(url, httpOptions).pipe(
+      tap( _ => {}),
+      catchError(this.handleError<APIResponse>('Get response'))
+    ).subscribe((response: APIResponse) => {  
+      if(response.err==null)  {
+        callback(response.object);
+      }
+      else this.messageService.add('Error:' + response.err);
+    });
+  }
+
   getApplications() {
     var self = this;
     this.getResponse('getapps', function(obj: Application[]){

src/app/login-form/login-form.component.html

@@ -9,7 +9,9 @@ <h1>Login</h1>
             <input matInput type="password" placeholder="Password" [(ngModel)]="password" required>
         </mat-form-field>
 
-        <button mat-button (click)="onSubmit($event)">Submit</button>
-        
+        <button mat-button (click)="onSubmit($event)">Submit</button>        
+    </div>
+    <div *ngIf="oauth.use_oauth==true" class="container">
+        <a href="{{ oauth.entrance_url }}">{{ oauth.display_name }}</a>
     </div>
 </div>

src/app/login-form/login-form.component.ts

@@ -3,7 +3,7 @@ import { HttpClient } from '@angular/common/http';
 import { Router } from '@angular/router';
 import * as bcrypt from 'bcryptjs';
 import * as cryptojs from 'crypto-js';
-import { AuthUser } from '../models';
+import { AuthUser, OAuthInfo } from '../models';
 import { MessageService } from '../message.service';
 import { ApplicationService } from '../application.service';
 declare var require: any;
@@ -17,6 +17,7 @@ export class LoginFormComponent implements OnInit {
   username: string;
   password: string;
   submitted = false; 
+  oauth: OAuthInfo = new (OAuthInfo);
 
   onSubmit(event) { 
     if(!this.username || !this.password){
@@ -42,7 +43,14 @@ export class LoginFormComponent implements OnInit {
     private http: HttpClient,
     private router: Router) { }
 
-  ngOnInit() {    
+  ngOnInit() { 
+    let self=this;   
+    this.applicationService.getResponseByURL('/janusec-admin/oauth/get',
+      function(obj: OAuthInfo){
+        self.oauth=obj;
+        console.log(self.oauth);
+      })
   }
+  
 
 }

src/app/models.ts

@@ -272,4 +272,10 @@ export class Server {
         this.username =username;
         this.password = password;
     }  
+}
+
+export class OAuthInfo {
+    use_oauth: boolean;
+    display_name: string;
+    entrance_url: string;
 }