Add table azure_tenant closes #140

azure-test/tests/azure_tenant/test-list-expected.json

@@ -0,0 +1,7 @@
+[
+  {
+    "display_name": null,
+    "name": "{{ output.current_tenant_display_name.value }}",
+    "tenant_id": "{{ output.tenant_id.value }}"
+  }
+]

azure-test/tests/azure_tenant/test-list-query.sql

@@ -0,0 +1,3 @@
+select name, display_name, tenant_id
+from azure.azure_tenant
+where tenant_id = '{{ output.tenant_id.value }}'

azure-test/tests/azure_tenant/test-notfound-expected.json

@@ -0,0 +1 @@
+null

azure-test/tests/azure_tenant/test-notfound-query.sql

@@ -0,0 +1,3 @@
+select display_name, title
+from azure.azure_tenant
+where display_name = 'dummy-{{ resourceName }}';

azure-test/tests/azure_tenant/test-turbot-expected.json

@@ -0,0 +1,6 @@
+[
+  {
+    "name": "{{ output.current_tenant_display_name.value }}",
+    "title": "{{ output.current_tenant_display_name.value }}"
+  }
+]

azure-test/tests/azure_tenant/test-turbot-query.sql

@@ -0,0 +1,3 @@
+select name, title
+from azure.azure_tenant
+where tenant_id = '{{ output.tenant_id.value }}';

azure-test/tests/azure_tenant/variable.tf

@@ -0,0 +1,46 @@
+variable "resource_name" {
+  type        = string
+  default     = "turbot-test-20200125-create-update"
+  description = "Name of the resource used throughout the test."
+}
+
+variable "azure_environment" {
+  type        = string
+  default     = "public"
+  description = "Azure environment used for the test."
+}
+
+variable "azure_subscription" {
+  type        = string
+  default     = "3510ae4d-530b-497d-8f30-53b9616fc6c1"
+  description = "Azure environment used for the test."
+}
+
+variable "azure_tenant" {
+  type        = string
+  default     = "cdffd708-7da0-4cea-abeb-0a4c334d7f64"
+  description = "Azure environment used for the test."
+}
+
+provider "azuread" {
+  # Cannot be passed as a variable
+  # version         = "=0.10.0"
+  environment     = var.azure_environment
+  subscription_id = var.azure_subscription
+  tenant_id       = var.azure_tenant
+}
+
+data "azurerm_client_config" "current" {
+}
+
+provider "azurerm" {
+  features {}
+}
+
+output "current_tenant_display_name" {
+  value = data.azurerm_client_config.current.tenant_id
+}
+
+output "tenant_id" {
+  value = data.azurerm_client_config.current.tenant_id
+}

azure/plugin.go

@@ -78,6 +78,7 @@ func Plugin(ctx context.Context) *plugin.Plugin {
 			"azure_storage_table_service":                tableAzureStorageTableService(ctx),
 			"azure_subnet":                               tableAzureSubnet(ctx),
 			"azure_subscription":                         tableAzureSubscription(ctx),
+			"azure_tenant":                               tableAzureTenant(ctx),
 			"azure_virtual_network":                      tableAzureVirtualNetwork(ctx),
 			// "azure_storage_blob":                tableAzureStorageBlob(ctx),
 			// "azure_storage_table":               tableAzureStorageTable(ctx),

azure/table_azure_tenant.go

@@ -0,0 +1,115 @@
+package azure
+
+import (
+	"context"
+
+	"github.com/Azure/azure-sdk-for-go/services/resources/mgmt/2019-06-01/subscriptions"
+	"github.com/turbot/steampipe-plugin-sdk/grpc/proto"
+	"github.com/turbot/steampipe-plugin-sdk/plugin"
+	"github.com/turbot/steampipe-plugin-sdk/plugin/transform"
+)
+
+//// TABLE DEFINITION
+
+func tableAzureTenant(_ context.Context) *plugin.Table {
+	return &plugin.Table{
+		Name:        "azure_tenant",
+		Description: "Azure Tenant",
+		List: &plugin.ListConfig{
+			Hydrate: listTenants,
+		},
+		Columns: []*plugin.Column{
+			{
+				Name:        "name",
+				Type:        proto.ColumnType_STRING,
+				Description: "The display name of the tenant.",
+				Transform:   transform.From(getNameOrID),
+			},
+			{
+				Name:        "id",
+				Type:        proto.ColumnType_STRING,
+				Description: "The fully qualified ID of the tenant. For example, /tenants/00000000-0000-0000-0000-000000000000.",
+				Transform:   transform.FromGo(),
+			},
+			{
+				Name:        "tenant_id",
+				Type:        proto.ColumnType_STRING,
+				Description: "The tenant ID. For example, 00000000-0000-0000-0000-000000000000.",
+				Transform:   transform.FromField("TenantID"),
+			},
+			{
+				Name:        "tenant_category",
+				Type:        proto.ColumnType_STRING,
+				Description: "The tenant category. Possible values include: 'Home', 'ProjectedBy', 'ManagedBy'",
+				Transform:   transform.FromField("TenantCategory").Transform(transform.ToString),
+			},
+			{
+				Name:        "country",
+				Type:        proto.ColumnType_STRING,
+				Description: "Country/region name of the address for the tenant.",
+			},
+			{
+				Name:        "country_code",
+				Type:        proto.ColumnType_STRING,
+				Description: "Country/region abbreviation for the tenant.",
+			},
+			{
+				Name:        "display_name",
+				Type:        proto.ColumnType_STRING,
+				Description: "The list of domains for the tenant.",
+			},
+			{
+				Name:        "domains",
+				Type:        proto.ColumnType_JSON,
+				Description: "The list of domains for the tenant.",
+			},
+
+			// Standard columns
+			{
+				Name:        "title",
+				Description: ColumnDescriptionTitle,
+				Type:        proto.ColumnType_STRING,
+				Transform:   transform.From(getNameOrID),
+			},
+			{
+				Name:        "akas",
+				Description: ColumnDescriptionAkas,
+				Type:        proto.ColumnType_JSON,
+				Transform:   transform.FromField("ID").Transform(idToAkas),
+			},
+		},
+	}
+}
+
+//// LIST FUNCTION
+
+func listTenants(ctx context.Context, d *plugin.QueryData, _ *plugin.HydrateData) (interface{}, error) {
+	session, err := GetNewSession(ctx, d, "MANAGEMENT")
+	if err != nil {
+		return nil, err
+	}
+
+	client := subscriptions.NewTenantsClient()
+	client.Authorizer = session.Authorizer
+
+	op, err := client.List(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	for _, resp := range op.Values() {
+		d.StreamListItem(ctx, resp)
+	}
+
+	return nil, nil
+}
+
+//// TRANSFORM FUNCTION
+
+func getNameOrID(ctx context.Context, d *transform.TransformData) (interface{}, error) {
+	data := d.HydrateItem.(subscriptions.TenantIDDescription)
+	if data.DisplayName != nil {
+		return data.DisplayName, nil
+	} 
+	return data.TenantID, nil
+}

azure/utils.go

@@ -24,6 +24,15 @@ func idToSubscriptionID(ctx context.Context, d *transform.TransformData) (interf
 func idToAkas(ctx context.Context, d *transform.TransformData) (interface{}, error) {
 	id := types.SafeString(d.Value)
 	akas := []string{"azure://" + id, "azure://" + strings.ToLower(id)}
+	occured := map[string]bool{}
+	result := []string{}
+	for i := range akas {
+		if !occured[akas[i]] {
+			occured[akas[i]] = true
+			result = append(result, akas[i])
+		}
+	}
+	akas = result
 	return akas, nil
 }
 

docs/tables/azure_tenant.md

@@ -0,0 +1,21 @@
+# Table: azure_tenant
+
+A dedicated and trusted instance of Azure AD that's automatically created when your organization signs up for a Microsoft cloud service subscription, such as Microsoft Azure, Microsoft Intune, or Microsoft 365. An Azure tenant represents a single organization.
+
+## Examples
+
+### Basic info
+
+```sql
+select
+  name,
+  id,
+  tenant_id,
+  tenant_category,
+  country,
+  country_code,
+  display_name,
+  domains
+from
+  azure_tenant;
+```