policy Commands

azure-devops/azext_devops/dev/common/const.py

@@ -24,3 +24,29 @@
 DEVOPS_TEAM_PROJECT_DEFAULT = 'project'
 PAT_ENV_VARIABLE_NAME = CLI_ENV_VARIABLE_PREFIX + 'PAT'
 AUTH_TOKEN_ENV_VARIABLE_NAME = CLI_ENV_VARIABLE_PREFIX + 'AUTH_TOKEN'
+
+# policy constants
+
+APPROVER_COUNT_POLICY = 'ApproverCountPolicy'
+APPROVER_COUNT_POLICY_ID = 'fa4e907d-c16b-4a4c-9dfa-4906e5d171dd'
+
+BUILD_POLICY = 'Buildpolicy'
+BUILD_POLICY_ID = '0609b952-1397-4640-95ec-e00a01b2c241'
+
+COMMENT_REQUIREMENTS_POLICY = 'CommentRequirementsPolicy'
+COMMENT_REQUIREMENTS_POLICY_ID = 'c6a1889d-b943-4856-b76f-9e46bb6b0df2'
+
+MERGE_STRATEGY_POLICY = 'MergeStrategyPolicy'
+MERGE_STRATEGY_POLICY_ID = 'fa4e907d-c16b-4a4c-9dfa-4916e5d171ab'
+
+FILE_SIZE_POLICY = 'FileSizePolicy'
+FILE_SIZE_POLICY_ID = '2e26e725-8201-4edd-8bf5-978563c34a80'
+
+WORKITEM_LINKING_POLICY = 'WorkItemLinkingPolicy'
+WORKITEM_LINKING_POLICY_ID = '40e92b44-2fe1-4dd6-b3d8-74a9c21d0c6e'
+
+REQUIRED_REVIEWER_POLICY = 'RequiredReviewersPolicy'
+REQUIRED_REVIEWER_POLICY_ID = 'fd2167ab-b0be-447a-8ec8-39368250530e'
+
+REPO_POLICY_TYPE = [APPROVER_COUNT_POLICY, BUILD_POLICY, COMMENT_REQUIREMENTS_POLICY, MERGE_STRATEGY_POLICY,
+                    FILE_SIZE_POLICY, WORKITEM_LINKING_POLICY, REQUIRED_REVIEWER_POLICY]

azure-devops/azext_devops/dev/repos/_format.py

@@ -12,6 +12,37 @@
 _WORK_ITEM_TITLE_TRUNCATION_LENGTH = 70
 
 
+def transform_repo_policies_table_output(result):
+    table_output = []
+    for item in result:
+        table_output.append(_transform_repo_policy_request_row(item))
+    return table_output
+
+
+def transform_repo_policy_table_output(result):
+    table_output = [_transform_repo_policy_request_row(result)]
+    return table_output
+
+
+def _transform_repo_policy_request_row(row):
+    table_row = OrderedDict()
+    table_row['ID'] = row['id']
+    table_row['Name'] = _get_policy_display_name(row)
+    table_row['Is Blocking'] = row['isBlocking']
+    table_row['Is Enabled'] = row['isEnabled']
+    # this will break if policy is applied across repo but that is not possible via UI at least now
+    table_row['Repository Id'] = row['settings']['scope'][0]['repositoryId']
+    table_row['Branch'] = row['settings']['scope'][0]['refName']
+    return table_row
+
+
+def _get_policy_display_name(row):
+    if 'displayName' in row['settings']:
+        return row['settings']['displayName']
+
+    return row['type']['displayName']
+
+
 def transform_pull_requests_table_output(result):
     table_output = []
     for item in result:

azure-devops/azext_devops/dev/repos/arguments.py

@@ -4,6 +4,7 @@
 # --------------------------------------------------------------------------------------------
 
 from knack.arguments import enum_choice_list
+from azext_devops.dev.common.const import REPO_POLICY_TYPE
 
 # CUSTOM CHOICE LISTS
 _ON_OFF_SWITCH_VALUES = ['on', 'off']
@@ -18,6 +19,9 @@ def load_code_arguments(self, _):
         context.argument('reviewers', nargs='*')
         context.argument('detect', **enum_choice_list(_ON_OFF_SWITCH_VALUES))
 
+    with self.argument_context('repos policies create') as context:
+        context.argument('policy_type', **enum_choice_list(REPO_POLICY_TYPE))
+
     with self.argument_context('repos pr') as context:
         context.argument('description', type=str, options_list=('--description', '-d'), nargs='*')
         context.argument('repository', options_list=('--repository', '-r'))

azure-devops/azext_devops/dev/repos/commands.py

@@ -13,7 +13,9 @@
                       transform_policies_table_output,
                       transform_policy_table_output,
                       transform_work_items_table_output,
-                      transform_repo_import_table_output)
+                      transform_repo_import_table_output,
+                      transform_repo_policy_table_output,
+                      transform_repo_policies_table_output)
 
 
 reposPullRequestOps = CliCommandType(
@@ -28,6 +30,10 @@
     operations_tmpl='azext_devops.dev.repos.import_request#{}'
 )
 
+policyOps = CliCommandType(
+    operations_tmpl='azext_devops.dev.repos.policy#{}'
+)
+
 
 def load_code_commands(self, _):
     with self.command_group('repos', command_type=reposRepositoryOps) as g:
@@ -37,6 +43,14 @@ def load_code_commands(self, _):
         g.command('list', 'list_repos', table_transformer=transform_repos_table_output)
         g.command('show', 'show_repo', table_transformer=transform_repo_table_output)
 
+    with self.command_group('repos policies', command_type=policyOps) as g:
+        # repository/ branch policies
+        g.command('create', 'create_policy', table_transformer=transform_repo_policy_table_output)
+        g.command('list', 'list_policy', table_transformer=transform_repo_policies_table_output)
+        g.command('show', 'get_policy', table_transformer=transform_repo_policy_table_output)
+        g.command('update', 'update_policy', table_transformer=transform_repo_policy_table_output)
+        g.command('delete', 'delete_policy', confirmation='Are you sure you want to delete this policy?')
+
     with self.command_group('repos pr', command_type=reposPullRequestOps) as g:
         # basic pr commands
         g.command('create', 'create_pull_request', table_transformer=transform_pull_request_table_output)