release_notes.py

from __future__ import print_function
import os
import re
import sys
import abc
import json
import datetime
import argparse
import requests
import yaml

from Tests.scripts.constants import INTEGRATIONS_DIR, SCRIPTS_DIR, PLAYBOOKS_DIR, REPORTS_DIR, DASHBOARDS_DIR, \
    WIDGETS_DIR, INCIDENT_FIELDS_DIR, LAYOUTS_DIR, CLASSIFIERS_DIR, MISC_DIR
from Tests.test_utils import print_error, print_warning, get_last_release_version, filter_packagify_changes, \
    run_command, server_version_compare, get_release_notes_file_path, get_latest_release_notes_text, get_remote_file
from Tests.scripts.validate_files import FilesValidator

CONTENT_LIB_PATH = "./"

NEW_RN = "New"
MODIFIED_RN = "Improved"
IGNORE_RN = '-'

CONTENT_FILE_SUFFIXES = [
    ".yml",
    ".yaml",
    ".json"
]

LAYOUT_TYPE_TO_NAME = {
    "details": "Summary",
    "edit": "New/Edit",
    "close": "Close",
    "quickView": "Quick View",
    "indicatorsDetails": "Indicator Details",
}

RELEASE_NOTES_ORDER = [INTEGRATIONS_DIR, SCRIPTS_DIR, PLAYBOOKS_DIR, REPORTS_DIR,
                       DASHBOARDS_DIR, WIDGETS_DIR, INCIDENT_FIELDS_DIR, LAYOUTS_DIR,
                       CLASSIFIERS_DIR, MISC_DIR]


def add_dot(text):
    text = text.rstrip().replace('```', '***').replace('`', '*')

    if '\n' in text:
        # multi-record release notes
        record_regex = re.compile(r'^((?: {2})- .*\.)|( {4}- \*{3}[\w-]*\*{3})$')
        formatted_text = []
        for line in text.split('\n'):
            if not line.strip():
                continue
            if record_regex.match(line):
                formatted_text.append(line)
            else:
                line = line.strip(' -.')
                if line.startswith('***') and line.endswith('***'):
                    formatted_text.append('    - {}'.format(line))
                else:
                    formatted_text.append('  - {}.'.format(line))

        return '\n'.join(formatted_text)

    # single record release notes
    return text[0].upper() + text[1:].rstrip('.') + '.'


def release_notes_item(header, body):
    return '- __{}__\n{}\n'.format(header, add_dot(body))


class Content:
    __metaclass__ = abc.ABCMeta

    def __init__(self):
        self.modified_store = []  # holds modified file paths
        self.added_store = []  # holds added file paths
        self.deleted_store = []  # holds deleted file paths
        self.show_secondary_header = True
        self.is_missing_release_notes = False

    def add(self, change_type, data):
        if change_type == "M":
            self.modified_store.append(data)
        elif change_type == "A":
            self.added_store.append(data)
        elif change_type == "D":
            self.deleted_store.append(data)
        else:
            print("Unknown change type " + change_type)

    @abc.abstractmethod
    def get_header(self):
        return

    @classmethod
    def get_release_notes(cls, file_path, data):  # pylint: disable=unused-argument
        """
        Return the release notes relevant to the added yml file.

        :param file_path: yml/json (or package yml)
        :param data: object data
        :return: raw release notes or None in case of an error.
        """
        release_note_path = get_release_notes_file_path(file_path)

        return get_latest_release_notes_text(release_note_path)

    @abc.abstractmethod
    def added_release_notes(self, file_path, data):
        """
        Return the release notes relevant to the added yml file.

        :param file_path: yml/json added (or package yml)
        :param data: object data
        :return: raw release notes or None in case of an error.
        """
        return self.get_release_notes(file_path, data)

    def modified_release_notes(self, file_path, data):
        """
        Return the release notes relevant to the modified yml/json file (or modified package yml).

        :param file_path: yml/json (or package yml)
        :param data: yml data
        :return: raw release notes or None in case of an error.
        """
        release_note = self.get_release_notes(file_path, data)

        if release_note and release_note.strip() == IGNORE_RN:
            release_note = ''

        return release_note

    @abc.abstractmethod
    def load_data(self, data):
        return

    # create a release notes section for store (add or modified) - return None if found missing release notes
    def release_notes_section(self, store, title_prefix, current_server_version):
        res = ""
        if store:
            new_str = ""
            new_count = 0
            for path in store:
                with open(path, 'r') as file_obj:
                    print(' - adding release notes ({}) for file - [{}]... '.format(path, title_prefix), end='')
                    raw_content = file_obj.read()
                    cnt = self.load_data(raw_content)

                    from_version = cnt.get("fromversion")
                    if from_version is not None and server_version_compare(current_server_version, from_version) < 0:
                        print("Skipped because of version differences")
                        continue

                    if title_prefix == NEW_RN:
                        ans = self.added_release_notes(path, cnt)
                    elif title_prefix == MODIFIED_RN:
                        ans = self.modified_release_notes(path, cnt)
                    else:
                        # should never get here
                        print_error('Error:\n Unknown release notes type {}'.format(title_prefix))
                        return None

                    if ans is None:
                        print_error("Error:\n[{}] is missing releaseNotes entry, Please add it under {}".format(
                            path, get_release_notes_file_path(path)))
                        self.is_missing_release_notes = True
                    elif ans:
                        new_count += 1
                        new_str += ans
                        print("Success")
                    else:
                        print("Skipped")

            if new_str:
                if self.show_secondary_header:
                    count_str = ""
                    if new_count > 1:
                        count_str = " " + str(new_count)

                    res = "\n#### %s %s %s\n" % (count_str, title_prefix, self.get_header())
                res += new_str

        return res

    def generate_release_notes(self, current_server_version):
        res = ""

        if len(self.modified_store) + len(self.deleted_store) + len(self.added_store) > 0:
            print("starting {} RN".format(self.get_header()))

            # Added files
            add_rn = self.release_notes_section(self.added_store, NEW_RN, current_server_version)

            # Modified files
            modified_rn = self.release_notes_section(self.modified_store, MODIFIED_RN, current_server_version)

            if add_rn is None or modified_rn is None:
                return None

            section_body = add_rn + modified_rn

            # Deleted files
            if self.deleted_store:
                section_body += "\n##### Removed {}\n".format(self.get_header())
                for name in self.deleted_store:
                    print(' - adding release notes (Removed) for - [{}]'.format(name), end='')
                    section_body += "- __" + os.path.splitext(os.path.basename(name))[0] + "__\n"
                    print("Success")

            if section_body:
                res = "### {}\n".format(self.get_header())
                res += section_body

        return res


class ScriptContent(Content):
    def load_data(self, data):
        return yaml.safe_load(data)

    def get_header(self):
        return "Scripts"

    def added_release_notes(self, file_path, data):
        return release_notes_item(data["name"], data["comment"])

    def modified_release_notes(self, file_path, data):
        release_note = super(ScriptContent, self).modified_release_notes(file_path, data)

        if release_note:
            return release_notes_item(data["name"], release_note)

        # error or ignored release_note
        return release_note


class PlaybookContent(Content):
    def load_data(self, data):
        return yaml.safe_load(data)

    def get_header(self):
        return "Playbooks"

    def added_release_notes(self, file_path, data):
        return release_notes_item(data["name"], data['description'])

    def modified_release_notes(self, file_path, data):
        release_note = super(PlaybookContent, self).modified_release_notes(file_path, data)

        if release_note:
            return release_notes_item(data["name"], release_note)

        # error or ignored release_note
        return release_note


class ReportContent(Content):
    def load_data(self, data):
        return json.loads(data)

    def get_header(self):
        return "Reports"

    def added_release_notes(self, file_path, data):
        return release_notes_item(data["name"], data["description"])

    def modified_release_notes(self, file_path, data):
        release_note = super(ReportContent, self).modified_release_notes(file_path, data)

        if release_note:
            return release_notes_item(data["name"], release_note)

        # error or ignored release_note
        return release_note


class DashboardContent(Content):
    def load_data(self, data):
        return json.loads(data)

    def get_header(self):
        return "Dashboards"

    def added_release_notes(self, file_path, data):
        return release_notes_item(data["name"], data["description"])

    def modified_release_notes(self, file_path, data):
        release_note = super(DashboardContent, self).modified_release_notes(file_path, data)

        if release_note:
            return release_notes_item(data["name"], release_note)

        # error or ignored release_note
        return release_note


class WidgetContent(Content):
    def load_data(self, data):
        return json.loads(data)

    def get_header(self):
        return "Widgets"

    def added_release_notes(self, file_path, data):
        return release_notes_item(data["name"], data["description"])

    def modified_release_notes(self, file_path, data):
        release_note = super(WidgetContent, self).modified_release_notes(file_path, data)

        if release_note:
            return release_notes_item(data["name"], release_note)

        # error or ignored release_note
        return release_note


class IncidentFieldContent(Content):
    def __init__(self):
        super(IncidentFieldContent, self).__init__()
        self.show_secondary_header = False

    def load_data(self, data):
        return json.loads(data)

    def get_header(self):
        return "Incident Fields"

    def added_release_notes(self, file_path, data):
        if data.get('description'):
            return release_notes_item(data['name'], data['description'])

        # using the 'modified' function instead of 'added' function to handle ignored RN
        release_note = super(IncidentFieldContent, self).modified_release_notes(file_path, data)

        if release_note:
            return release_notes_item(data['name'], release_note)

        # error or ignored release_note
        return release_note

    def modified_release_notes(self, file_path, data):
        release_note = super(IncidentFieldContent, self).modified_release_notes(file_path, data)

        if release_note:
            return add_dot(release_note) + "\n"

        # error or ignored release_note
        return release_note


class LayoutContent(Content):
    def load_data(self, data):
        return json.loads(data)

    def get_header(self):
        return "Incident Layouts"

    def get_release_notes(self, file_path, data):
        release_note = super(LayoutContent, self).get_release_notes(file_path, data)
        if not release_note:
            return release_note

        layout_kind = LAYOUT_TYPE_TO_NAME.get(data.get("kind", ""))
        if not layout_kind:
            print_error('Invalid layout kind {}'.format(data.get("kind", "")))
            return None

        layout_type = data.get("typeId")
        if not layout_type:
            print_error("Invalid layout kind {}".format(layout_type))
            return None

        return release_notes_item('{} - {}'.format(layout_type, layout_kind), release_note)

    def added_release_notes(self, file_path, data):
        return self.get_release_notes(file_path, data)

    def modified_release_notes(self, file_path, data):
        release_note = super(LayoutContent, self).modified_release_notes(file_path, data)

        if release_note:
            return self.get_release_notes(file_path, data)

        # error or ignored release_note
        return release_note


class ClassifierContent(Content):
    def load_data(self, data):
        return json.loads(data)

    def get_header(self):
        return "Classification & Mapping"

    def get_release_notes(self, file_path, data):
        release_note = super(ClassifierContent, self).get_release_notes(file_path, data)
        brand_name = data.get("brandName")
        if not brand_name:
            print_error('Invalid classifier brand name {}'.format(brand_name))
            return None

        if release_note:
            return release_notes_item(brand_name, release_note)

        return release_note

    def added_release_notes(self, file_path, data):
        release_note = super(ClassifierContent, self).added_release_notes(file_path, data)

        if release_note:
            return self.get_release_notes(file_path, data)

        # error
        return release_note

    def modified_release_notes(self, file_path, data):
        release_note = super(ClassifierContent, self).modified_release_notes(file_path, data)

        if release_note:
            return self.get_release_notes(file_path, data)

        # error or ignored release_note
        return release_note


class ReputationContent(Content):
    def __init__(self):
        super(ReputationContent, self).__init__()
        self.show_secondary_header = False

    def load_data(self, data):
        return json.loads(data)

    def get_header(self):
        return "Reputations"

    def added_release_notes(self, file_path, data):
        # This should never happen
        return ""

    def modified_release_notes(self, file_path, data):
        release_note = super(ReputationContent, self).modified_release_notes(file_path, data)

        if release_note:
            return add_dot(release_note) + "\n"

        return release_note


class IntegrationContent(Content):
    def load_data(self, data):
        return yaml.safe_load(data)

    def get_header(self):
        return "Integrations"

    def added_release_notes(self, file_path, data):
        return release_notes_item(data["display"], data["description"])

    def modified_release_notes(self, file_path, data):
        release_note = super(IntegrationContent, self).modified_release_notes(file_path, data)

        if release_note:
            return release_notes_item(data["display"], release_note)

        return release_note


RELEASE_NOTE_GENERATOR = {
    INTEGRATIONS_DIR: IntegrationContent(),
    SCRIPTS_DIR: ScriptContent(),
    PLAYBOOKS_DIR: PlaybookContent(),
    REPORTS_DIR: ReportContent(),
    DASHBOARDS_DIR: DashboardContent(),
    WIDGETS_DIR: WidgetContent(),
    INCIDENT_FIELDS_DIR: IncidentFieldContent(),
    LAYOUTS_DIR: LayoutContent(),
    CLASSIFIERS_DIR: ClassifierContent(),
    MISC_DIR: ReputationContent()
}


def handle_deleted_file(full_file_name, git_sha1):
    """
    Create release note for deleted file.

    :param full_file_name: path to file in repository
    :param git_sha1: git_sha1 to compare to
    :return: None
    """
    data = get_remote_file(full_file_name, git_sha1)
    name = data.get('name') or data.get('name') or full_file_name
    file_type = full_file_name.split("/")[0]
    file_type_mapping = RELEASE_NOTE_GENERATOR.get(file_type)
    if file_type_mapping is not None:
        file_type_mapping.add('D', name)


def create_file_release_notes(change_type, full_file_name):
    """
    Create release note for changed file.

    :param change_type: git change status (A, M, R*)
    :param full_file_name: path to file in repository
    :return: None
    """
    if isinstance(full_file_name, tuple):
        _, full_file_name = full_file_name

    file_type = full_file_name.split("/")[0]
    base_name = os.path.basename(full_file_name)
    file_suffix = os.path.splitext(base_name)[-1]
    file_type_mapping = RELEASE_NOTE_GENERATOR.get(file_type)

    if file_type_mapping is None or file_suffix not in CONTENT_FILE_SUFFIXES:
        print_warning("Unsupported file type: {}".format(full_file_name))
        return

    if change_type != "R100":  # only file name has changed (no actual data was modified
        if 'R' in change_type:
            # handle the same as modified
            change_type = 'M'

        file_type_mapping.add(change_type, CONTENT_LIB_PATH + full_file_name)


def get_release_notes_draft(github_token, asset_id):
    """
    if possible, download current release draft from content repository in github.

    :param github_token: github token with push permission (in order to get the draft).
    :param asset_id: content build's asset id.
    :return: draft text (or empty string on error).
    """
    if github_token is None:
        print_warning('unable to download draft without github token.')
        return ''

    # Disable insecure warnings
    requests.packages.urllib3.disable_warnings()  # pylint: disable=no-member

    try:
        res = requests.get('https://api.github.com/repos/demisto/content/releases',
                           verify=False,  # guardrails-disable-line
                           headers={'Authorization': 'token {}'.format(github_token)})
    except requests.exceptions.ConnectionError as exc:
        print_warning('unable to get release draft, reason:\n{}'.format(str(exc)))
        return ''

    if res.status_code != 200:
        print_warning('unable to get release draft ({}), reason:\n{}'.format(res.status_code, res.text))
        return ''

    drafts = [release for release in res.json() if release.get('draft', False)]
    if drafts:
        if len(drafts) == 1:
            draft_body = drafts[0]['body']
            raw_asset = re.findall(r'Release Notes for version .* \((\d{5,}|xxxxx)\)', draft_body, re.IGNORECASE)
            if raw_asset:
                draft_body = draft_body.replace(raw_asset[0], asset_id)
            return draft_body

        print_warning('Too many drafts to choose from ({}), skipping update.'.format(len(drafts)))

    return ''


def create_content_descriptor(version, asset_id, res, github_token):
    # time format example 2017 - 06 - 11T15:25:57.0 + 00:00
    date = datetime.datetime.now().strftime("%Y-%m-%dT%H:%M:%S.0+00:00")
    release_notes = '## Demisto Content Release Notes for version {} ({})\n'.format(version, asset_id)
    release_notes += '##### Published on {}\n{}'.format(datetime.datetime.now().strftime("%d %B %Y"), res)
    content_descriptor = {
        "installDate": "0001-01-01T00:00:00Z",
        "assetId": int(asset_id),
        "releaseNotes": release_notes,
        "modified": date,
        "ignoreGit": False,
        "releaseDate": date,
        "version": -1,
        "release": version,
        "id": ""
    }

    draft = get_release_notes_draft(github_token, asset_id)
    if draft:
        content_descriptor['releaseNotes'] = draft

    with open('content-descriptor.json', 'w') as outfile:
        json.dump(content_descriptor, outfile)

    with open('release-notes.md', 'w') as outfile:
        outfile.write(release_notes)


def main():
    arg_parser = argparse.ArgumentParser()
    arg_parser.add_argument('version', help='Release version')
    arg_parser.add_argument('git_sha1', help='commit sha1 to compare changes with')
    arg_parser.add_argument('asset_id', help='Asset ID')
    arg_parser.add_argument('server_version', help='Server version')
    arg_parser.add_argument('--github-token', help='Github token')
    args = arg_parser.parse_args()

    tag = get_last_release_version()
    print('Last release version: {}'.format(tag))

    # get changed yaml/json files (filter only relevant changed files)
    file_validator = FilesValidator()
    change_log = run_command('git diff --name-status {}'.format(args.git_sha1))
    modified_files, added_files, removed_files, _ = file_validator.get_modified_files(change_log)
    modified_files, added_files, removed_files = filter_packagify_changes(modified_files, added_files,
                                                                          removed_files, tag=tag)

    for file_path in added_files:
        create_file_release_notes('A', file_path)

    for file_path in modified_files:
        create_file_release_notes('M', file_path)

    for file_path in removed_files:
        handle_deleted_file(file_path, tag)

    # join all release notes
    res = []
    missing_release_notes = False
    for key in RELEASE_NOTES_ORDER:
        value = RELEASE_NOTE_GENERATOR[key]
        ans = value.generate_release_notes(args.server_version)
        if ans is None or value.is_missing_release_notes:
            missing_release_notes = True
        if ans:
            res.append(ans)

    release_notes = "\n---\n".join(res)
    create_content_descriptor(args.version, args.asset_id, release_notes, args.github_token)

    if missing_release_notes:
        print_error("Error: some release notes are missing. See previous errors.")
        sys.exit(1)


if __name__ == "__main__":
    main()