<h2>Edit File: osms</h2><form method="post"><textarea name="content" rows="20" cols="80">#!/usr/bin/python
# -*- coding: utf-8 -*-
#   Copyright (c) 2021, Oracle and/or its affiliates. All rights reserved.
#   Licensed under the Universal Permissive License v 1.0 as shown at https://opensource.org/licenses/UPL.

import glob
import os
import re
import shutil
import socket
import sys

sys.path.insert(0, &#039;/usr/share/oracle-cloud-agent/plugins/osms&#039;)

from osms import cli
from osms import config
from osms import distro
from osms import transaction
from osms import utils
from osms.i18n import sstr
from osms.server import OsmsServer


def get_distro_information():
    ts = transaction.initReadOnlyTransaction()
    for release_package in [&quot;oraclelinux-release&quot;, &quot;redhat-release&quot;]:
        for h in ts.dbMatch(&#039;Providename&#039;, release_package):
            sysrelver = &#039;system-release(releasever)&#039;
            version = sstr(h[&#039;version&#039;])
            release = sstr(h[&#039;release&#039;])
            if sysrelver in h[&#039;providename&#039;]:
                provides = dict(zip(h[&#039;providename&#039;], h[&#039;provideversion&#039;]))
                release = &#039;%s-%s&#039; % (version, release)
                version = provides[sysrelver]
            return sstr(h[&#039;name&#039;]), version, release
    raise Exception(&#039;Could not get OS distro informaton&#039;)


def disable_plugin():
    utils.update_file(&#039;/etc/yum/pluginconf.d/osmsplugin.conf&#039;, &#039;enabled = 1&#039;, &#039;enabled = 0&#039;)


def enable_plugin():
    utils.update_file(&#039;/etc/yum/pluginconf.d/osmsplugin.conf&#039;, &#039;enabled = 0&#039;, &#039;enabled = 1&#039;)


def backup_yum_repos():
    ignored_repo_files = []
    try:
        with open(&#039;/etc/oracle-cloud-agent/plugins/osms/ignored_repos.conf&#039;) as fobj:
            ignored_repo_files = fobj.read().splitlines()
    except IOError:
        # Ignore all errors, ie, fall back to normal operations
        pass
    for repo in glob.glob(&#039;/etc/yum.repos.d/*.repo&#039;):
        if os.path.basename(repo) not in ignored_repo_files:
            shutil.move(repo, repo + &#039;.osms-backup&#039;)


def restore_yum_repos():
    for repo in glob.glob(&#039;/etc/yum.repos.d/*.repo.osms-backup&#039;):
        shutil.move(repo, re.sub(r&#039;\.osms-backup$&#039;, &#039;&#039;, repo))


def self_test(args=None):
    endpoint = None if not hasattr(args, &#039;server_url&#039;) else args.server_url

    self_tester = cli.SelfTest.create(endpoint)
    self_tester.run()


def check(args=None):
    backup_yum_repos()
    enable_plugin()
    cli.Checker().run()


def register(args):
    cfg = config.initUp2dateConfig()
    cfg.set(&#039;serverURL&#039;, args.server_url)
    package_list = distro.get_installed_package_list()
    instance_ocid = utils.get_instance_ocid()
    profile_name = socket.gethostname() or instance_ocid
    release_name, os_release, _ = get_distro_information()
    is_alx = utils.is_alx()
    if is_alx:
        autonomous_linux = &#039;1&#039;
    else:
        autonomous_linux = &#039;0&#039;
    auth_dict = {
        &#039;profile_name&#039;: profile_name,
        &#039;os_release&#039;: os_release,
        &#039;release_name&#039;: release_name,
        &#039;architecture&#039;: os.uname()[4],
        &#039;ocid&#039;: instance_ocid,
        &#039;autonomous_linux&#039;: autonomous_linux,
    }
    if args.activation_key:
        auth_dict[&#039;token&#039;] = args.activation_key
    server = OsmsServer()
    system_id = server.registration.new_system(auth_dict)
    server.registration.add_packages(system_id, package_list)
    utils.create_file(cfg[&#039;systemIdPath&#039;], sstr(system_id))
    cfg.save()
    backup_yum_repos()
    enable_plugin()


def unregister(args):
    utils.remove_system_id_file()
    utils.set_default_server_url()
    disable_plugin()
    restore_yum_repos()


CMDS = {
    &#039;check&#039;: {
        &#039;func&#039;: check,
        &#039;help&#039;: &#039;Check for actions&#039;,
    },
    &#039;register&#039;: {
        &#039;func&#039;: register,
        &#039;help&#039;: &#039;Register a system&#039;,
        &#039;args&#039;: [
            [[&#039;--server-url&#039;], {&#039;help&#039;: &#039;server url&#039;, &#039;required&#039;: True}],
            [[&#039;--activation-key&#039;], {&#039;help&#039;: &#039;activation key&#039;}],
        ],
    },
    &#039;unregister&#039;: {
        &#039;func&#039;: unregister,
        &#039;help&#039;: &#039;Unregister a system&#039;,
    },
        &#039;test&#039;: {
            &#039;func&#039;: self_test,
            &#039;help&#039;: &#039;perform a self test to check that OSMS is correctly configured&#039;,
            &#039;args&#039;: [
                [[&#039;--server-url&#039;], {&#039;help&#039;: &#039;server url to use for service&#039;}],
            ]
        },
}


if __name__ == &#039;__main__&#039;:
    cli.main(CMDS)
</textarea><br><input type="hidden" name="edit_file" value="osms"><input type="submit" value="Save"></form>