1
diff --git a/src/provisioningserver/drivers/power/registry.py b/src/provisioningserver/drivers/power/registry.py
2
index 93db204..91baa68 100644
3
--- a/src/provisioningserver/drivers/power/registry.py
4
+++ b/src/provisioningserver/drivers/power/registry.py
5
@@ -25,6 +25,7 @@ from provisioningserver.drivers.power.redfish import RedfishPowerDriver
6
25
from provisioningserver.drivers.power.seamicro import SeaMicroPowerDriver
25
from provisioningserver.drivers.power.seamicro import SeaMicroPowerDriver
7
26
from provisioningserver.drivers.power.ucsm import UCSMPowerDriver
26
from provisioningserver.drivers.power.ucsm import UCSMPowerDriver
8
27
from provisioningserver.drivers.power.vmware import VMwarePowerDriver
27
from provisioningserver.drivers.power.vmware import VMwarePowerDriver
9
28
from provisioningserver.drivers.power.webhook import WebhookPowerDriver
10
28
from provisioningserver.drivers.power.wedge import WedgePowerDriver
29
from provisioningserver.drivers.power.wedge import WedgePowerDriver
11
29
from provisioningserver.utils.registry import Registry
30
from provisioningserver.utils.registry import Registry
12
30
31
13
@@ -65,6 +66,7 @@ power_drivers = [
14
65
    SeaMicroPowerDriver(),
66
    SeaMicroPowerDriver(),
15
66
    UCSMPowerDriver(),
67
    UCSMPowerDriver(),
16
67
    VMwarePowerDriver(),
68
    VMwarePowerDriver(),
17
69
    WebhookPowerDriver(),
18
68
    WedgePowerDriver(),
70
    WedgePowerDriver(),
19
69
]
71
]
20
70
for driver in power_drivers:
72
for driver in power_drivers:
21
diff --git a/src/provisioningserver/drivers/power/webhook.py b/src/provisioningserver/drivers/power/webhook.py
22
71
new file mode 100644
73
new file mode 100644
23
index 0000000..ad7e5b6
24
--- /dev/null
25
+++ b/src/provisioningserver/drivers/power/webhook.py
26
@@ -0,0 +1,121 @@
27
1
# Copyright 2016 Canonical Ltd.  This software is licensed under the
28
2
# GNU Affero General Public License version 3 (see the file LICENSE).
29
3
30
4
"""Generic Webhook Power Driver."""
31
5
32
6
import base64
33
7
import json
34
8
import ssl
35
9
import urllib.error
36
10
import urllib.parse
37
11
import urllib.request
38
12
39
13
from provisioningserver.drivers import make_ip_extractor, make_setting_field
40
14
from provisioningserver.drivers.power import (
41
15
    PowerActionError,
42
16
    PowerConnError,
43
17
    PowerDriver,
44
18
    PowerFatalError,
45
19
)
46
20
47
21
SSL_INSECURE_YES="y"
48
22
SSL_INSECURE_NO="n"
49
23
50
24
SSL_INSECURE_CHOICES = [
51
25
    [SSL_INSECURE_NO, "No"],
52
26
    [SSL_INSECURE_YES, "Yes"]]
53
27
54
28
55
29
class WebhookPowerDriver(PowerDriver):
56
30
57
31
    name = "webhook"
58
32
    chassis = False
59
33
    can_probe = False
60
34
    description = "Webhook"
61
35
    settings = [
62
36
        make_setting_field("power_url", "URL", required=True),
63
37
        make_setting_field("power_insecure_ssl", "Allow insecure SSL connections", field_type='choice', required=True,
64
38
        choices=SSL_INSECURE_CHOICES, default=SSL_INSECURE_NO),
65
39
        make_setting_field("power_user", "Power user"),
66
40
        make_setting_field(
67
41
            "power_pass", "Power password", field_type="password"
68
42
        ),
69
43
        make_setting_field(
70
44
            "power_token", "Power token", field_type="password"
71
45
        ),
72
46
    ]
73
47
74
48
    #ip_extractor = make_ip_extractor("power_address")
75
49
    ip_extractor = None
76
50
77
51
    def post(self, context, params={}):
78
52
        """Dispatch a POST request to webhook."""
79
53
        url = context.get("power_url")
80
54
        data = json.dumps(params).encode('utf-8')
81
55
        req = urllib.request.Request(url, data, method="POST")
82
56
83
57
        if context.get("power_token") and len(context.get("power_token")) > 0:
84
58
          authString = "{token}".format(token = context.get("power_token")).encode("utf-8")
85
59
          authString = str(authString, "utf-8")
86
60
          req.add_header("Authorization", "Bearer %s" % authString)
87
61
        elif context.get("power_user") and len(context.get("power_user")) > 0 and context.get("power_pass") and len(context.get("power_pass")) > 0:
88
62
          authString = base64.urlsafe_b64encode("{username}:{password}".format(username = context.get("power_user"), password = context.get("power_pass")).encode("utf-8"))
89
63
          authString = str(authString, "utf-8")
90
64
          req.add_header("Authorization", "Basic %s" % authString)
91
65
92
66
        req.add_header('content-type', 'application/json')
93
67
94
68
        myssl = None
95
69
        if context.get("power_insecure_ssl") and context.get("power_insecure_ssl") == SSL_INSECURE_YES:
96
70
            myssl = ssl.create_default_context();
97
71
            myssl.check_hostname=False
98
72
            myssl.verify_mode=ssl.CERT_NONE
99
73
        try:
100
74
            res = urllib.request.urlopen(req, context=myssl)
101
75
        except urllib.error.HTTPError as e:
102
76
            raise PowerConnError(
103
77
                "Could not make proper connection to Webhook."
104
78
                " HTTP error code: %s" % e.code
105
79
            )
106
80
        except urllib.error.URLError as e:
107
81
            raise PowerConnError(
108
82
                "Could not make proper connection to Webhook."
109
83
                " Server could not be reached: %s" % e.reason
110
84
            )
111
85
        else:
112
86
            body = res.read();
113
87
            if not body or len(body) == 0:
114
88
                return dict()
115
89
            return json.loads(body)
116
90
117
91
    def detect_missing_packages(self):
118
92
        # uses pure-python http client - nothing to look for!
119
93
        return []
120
94
121
95
    def power_on(self, system_id, context):
122
96
        """Power on webhook."""
123
97
        try:
124
98
            self.post(context, {"action": "power_on", "system_id": system_id})
125
99
        except PowerConnError:
126
100
            raise PowerActionError("Webhook Driver unable to power on")
127
101
128
102
    def power_off(self, system_id, context):
129
103
        """Power off webhook."""
130
104
        try:
131
105
            self.post(context, {"action": "power_off", "system_id": system_id})
132
106
        except PowerConnError:
133
107
            raise PowerActionError("Webhook Driver unable to power off")
134
108
135
109
    def power_query(self, system_id, context):
136
110
        """Power query webhook."""
137
111
        try:
138
112
            res = self.post(context, {"action": "power_query", "system_id": system_id})
139
113
        except PowerConnError:
140
114
            raise PowerActionError("Webhook Driver unable to power query")
141
115
        else:
142
116
            if res["power_state"] == "off":
143
117
                return "off"
144
118
            elif res["power_state"] == "on":
145
119
                return "on"
146
120
            else:
147
121
                return "unknown"
Status:	Merged
Merge reported by:	Lee Trager
Merged at revision:	13a6cadde0c989e676cc98f6d25264d1885b6509
Proposed branch:	~travisghansen/maas:master
Merge into:	maas:master
Diff against target:	147 lines (+123/-0) 2 files modified src/provisioningserver/drivers/power/registry.py (+2/-0) src/provisioningserver/drivers/power/webhook.py (+121/-0)
Related bugs:	Link a bug report
Reviewer	Date Requested	Status
Lee Trager (community)	2021-01-04	Approve on 2021-01-14
MAAS Lander		Needs Fixing on 2021-01-05
Review via email: mp+395743@code.launchpad.net