From e555536f2413927508db91affa371560ba633c88 Mon Sep 17 00:00:00 2001
From: Kyle Spearrin <kyle.spearrin@gmail.com>
Date: Wed, 18 Jul 2018 17:07:59 -0400
Subject: [PATCH] add org 2fa apis

---
 src/abstractions/api.service.ts |  7 +++++++
 src/services/api.service.ts     | 25 +++++++++++++++++++++++++
 2 files changed, 32 insertions(+)

diff --git a/src/abstractions/api.service.ts b/src/abstractions/api.service.ts
index 576befa3174..debb78580df 100644
--- a/src/abstractions/api.service.ts
+++ b/src/abstractions/api.service.ts
@@ -181,9 +181,12 @@ export abstract class ApiService {
     putSettingsDomains: (request: UpdateDomainsRequest) => Promise<DomainsResponse>;
 
     getTwoFactorProviders: () => Promise<ListResponse<TwoFactorProviderResponse>>;
+    getTwoFactorOrganizationProviders: (organizationId: string) => Promise<ListResponse<TwoFactorProviderResponse>>;
     getTwoFactorAuthenticator: (request: PasswordVerificationRequest) => Promise<TwoFactorAuthenticatorResponse>;
     getTwoFactorEmail: (request: PasswordVerificationRequest) => Promise<TwoFactorEmailResponse>;
     getTwoFactorDuo: (request: PasswordVerificationRequest) => Promise<TwoFactorDuoResponse>;
+    getTwoFactorOrganizationDuo: (organizationId: string,
+        request: PasswordVerificationRequest) => Promise<TwoFactorDuoResponse>;
     getTwoFactorYubiKey: (request: PasswordVerificationRequest) => Promise<TwoFactorYubiKeyResponse>;
     getTwoFactorU2f: (request: PasswordVerificationRequest) => Promise<TwoFactorU2fResponse>;
     getTwoFactorRecover: (request: PasswordVerificationRequest) => Promise<TwoFactorRecoverResponse>;
@@ -191,9 +194,13 @@ export abstract class ApiService {
         request: UpdateTwoFactorAuthenticatorRequest) => Promise<TwoFactorAuthenticatorResponse>;
     putTwoFactorEmail: (request: UpdateTwoFactorEmailRequest) => Promise<TwoFactorEmailResponse>;
     putTwoFactorDuo: (request: UpdateTwoFactorDuoRequest) => Promise<TwoFactorDuoResponse>;
+    putTwoFactorOrganizationDuo: (organizationId: string,
+        request: UpdateTwoFactorDuoRequest) => Promise<TwoFactorDuoResponse>;
     putTwoFactorYubiKey: (request: UpdateTwoFactorYubioOtpRequest) => Promise<TwoFactorYubiKeyResponse>;
     putTwoFactorU2f: (request: UpdateTwoFactorU2fRequest) => Promise<TwoFactorU2fResponse>;
     putTwoFactorDisable: (request: TwoFactorProviderRequest) => Promise<TwoFactorProviderResponse>;
+    putTwoFactorOrganizationDisable: (organizationId: string,
+        request: TwoFactorProviderRequest) => Promise<TwoFactorProviderResponse>;
     postTwoFactorRecover: (request: TwoFactorRecoveryRequest) => Promise<any>;
     postTwoFactorEmailSetup: (request: TwoFactorEmailRequest) => Promise<any>;
     postTwoFactorEmail: (request: TwoFactorEmailRequest) => Promise<any>;
diff --git a/src/services/api.service.ts b/src/services/api.service.ts
index 8226d260476..39dc06a8fbb 100644
--- a/src/services/api.service.ts
+++ b/src/services/api.service.ts
@@ -560,6 +560,11 @@ export class ApiService implements ApiServiceAbstraction {
         return new ListResponse(r, TwoFactorProviderResponse);
     }
 
+    async getTwoFactorOrganizationProviders(organizationId: string): Promise<ListResponse<TwoFactorProviderResponse>> {
+        const r = await this.send('GET', '/organizations/' + organizationId + '/two-factor', null, true, true);
+        return new ListResponse(r, TwoFactorProviderResponse);
+    }
+
     async getTwoFactorAuthenticator(request: PasswordVerificationRequest): Promise<TwoFactorAuthenticatorResponse> {
         const r = await this.send('POST', '/two-factor/get-authenticator', request, true, true);
         return new TwoFactorAuthenticatorResponse(r);
@@ -575,6 +580,13 @@ export class ApiService implements ApiServiceAbstraction {
         return new TwoFactorDuoResponse(r);
     }
 
+    async getTwoFactorOrganizationDuo(organizationId: string,
+        request: PasswordVerificationRequest): Promise<TwoFactorDuoResponse> {
+        const r = await this.send('POST', '/organizations/' + organizationId + '/two-factor/get-duo',
+            request, true, true);
+        return new TwoFactorDuoResponse(r);
+    }
+
     async getTwoFactorYubiKey(request: PasswordVerificationRequest): Promise<TwoFactorYubiKeyResponse> {
         const r = await this.send('POST', '/two-factor/get-yubikey', request, true, true);
         return new TwoFactorYubiKeyResponse(r);
@@ -606,6 +618,12 @@ export class ApiService implements ApiServiceAbstraction {
         return new TwoFactorDuoResponse(r);
     }
 
+    async putTwoFactorOrganizationDuo(organizationId: string,
+        request: UpdateTwoFactorDuoRequest): Promise<TwoFactorDuoResponse> {
+        const r = await this.send('PUT', '/organizations/' + organizationId + '/two-factor/duo', request, true, true);
+        return new TwoFactorDuoResponse(r);
+    }
+
     async putTwoFactorYubiKey(request: UpdateTwoFactorYubioOtpRequest): Promise<TwoFactorYubiKeyResponse> {
         const r = await this.send('PUT', '/two-factor/yubikey', request, true, true);
         return new TwoFactorYubiKeyResponse(r);
@@ -621,6 +639,13 @@ export class ApiService implements ApiServiceAbstraction {
         return new TwoFactorProviderResponse(r);
     }
 
+    async putTwoFactorOrganizationDisable(organizationId: string,
+        request: TwoFactorProviderRequest): Promise<TwoFactorProviderResponse> {
+        const r = await this.send('PUT', '/organizations/' + organizationId + '/two-factor/disable',
+            request, true, true);
+        return new TwoFactorProviderResponse(r);
+    }
+
     postTwoFactorRecover(request: TwoFactorRecoveryRequest): Promise<any> {
         return this.send('POST', '/two-factor/recover', request, false, false);
     }