azuredatastudio/src/sql/platform/accounts/common/firewallRuleViewModel.ts

/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *  Licensed under the Source EULA. See License.txt in the project root for license information.
 *--------------------------------------------------------------------------------------------*/

import * as azdata from 'azdata';

/**
 * View model for firewall rule dialog
 */
export class FirewallRuleViewModel {
	public isIPAddressSelected: boolean;
	public selectedAccount: azdata.Account | undefined;
	public selectedTenantId: string | undefined;

	private _defaultIPAddress?: string;
	private _defaultFromSubnetIPRange?: string;
	private _defaultToSubnetIPRange?: string;
	private _fromSubnetIPRange?: string;
	private _toSubnetIPRange?: string;

	constructor() {
		this.isIPAddressSelected = true;
	}

	public get defaultIPAddress(): string | undefined {
		return this._defaultIPAddress;
	}

	public get defaultFromSubnetIPRange(): string | undefined {
		return this._defaultFromSubnetIPRange;
	}

	public get defaultToSubnetIPRange(): string | undefined {
		return this._defaultToSubnetIPRange;
	}

	public set fromSubnetIPRange(IPAddress: string | undefined) {
		this._fromSubnetIPRange = IPAddress;
	}

	public get fromSubnetIPRange(): string | undefined {
		if (this._fromSubnetIPRange) {
			return this._fromSubnetIPRange;
		} else {
			return this._defaultFromSubnetIPRange;
		}
	}

	public set toSubnetIPRange(IPAddress: string | undefined) {
		this._toSubnetIPRange = IPAddress;
	}

	public get toSubnetIPRange(): string | undefined {
		if (this._toSubnetIPRange) {
			return this._toSubnetIPRange;
		} else {
			return this._defaultToSubnetIPRange;
		}
	}

	public updateDefaultValues(ipAddress: string): void {
		this._defaultIPAddress = ipAddress;
		this._defaultFromSubnetIPRange = ipAddress.replace(/\.[0-9]+$/g, '.0');
		this._defaultToSubnetIPRange = ipAddress.replace(/\.[0-9]+$/g, '.255');
	}
}