blob: 404b61f51626a7678ec147fbd22f5b6a42c965c2 [file] [log] [blame]
Patrick Williamsc0f7c042017-02-23 20:41:17 -06001#!/usr/bin/env python3
Patrick Williamsc124f4f2015-09-15 14:41:29 -05002#
3# Copyright (C) 2014 Alex Damian
4#
Brad Bishopc342db32019-05-15 21:57:59 -04005# SPDX-License-Identifier: GPL-2.0-only
6#
Patrick Williamsc124f4f2015-09-15 14:41:29 -05007# This file re-uses code spread throughout other Bitbake source files.
8# As such, all other copyrights belong to their own right holders.
9#
Patrick Williamsc124f4f2015-09-15 14:41:29 -050010
Patrick Williamsc0f7c042017-02-23 20:41:17 -060011"""
12This command takes a filename as a single parameter. The filename is read
13as a build eventlog, and the ToasterUI is used to process events in the file
14and log data in the database
15"""
Patrick Williamsc124f4f2015-09-15 14:41:29 -050016
Patrick Williamsc124f4f2015-09-15 14:41:29 -050017import os
Patrick Williamsc0f7c042017-02-23 20:41:17 -060018import sys
19import json
20import pickle
21import codecs
Andrew Geissler5199d832021-09-24 16:47:35 -050022import warnings
23warnings.simplefilter("default")
Patrick Williamsc0f7c042017-02-23 20:41:17 -060024
25from collections import namedtuple
Patrick Williamsc124f4f2015-09-15 14:41:29 -050026
27# mangle syspath to allow easy import of modules
Patrick Williamsc0f7c042017-02-23 20:41:17 -060028from os.path import join, dirname, abspath
29sys.path.insert(0, join(dirname(dirname(abspath(__file__))), 'lib'))
Patrick Williamsc124f4f2015-09-15 14:41:29 -050030
31import bb.cooker
32from bb.ui import toasterui
Patrick Williamsc124f4f2015-09-15 14:41:29 -050033
Patrick Williamsc0f7c042017-02-23 20:41:17 -060034class EventPlayer:
35 """Emulate a connection to a bitbake server."""
Patrick Williamsc124f4f2015-09-15 14:41:29 -050036
Patrick Williamsc0f7c042017-02-23 20:41:17 -060037 def __init__(self, eventfile, variables):
38 self.eventfile = eventfile
39 self.variables = variables
40 self.eventmask = []
Patrick Williamsc124f4f2015-09-15 14:41:29 -050041
Patrick Williamsc0f7c042017-02-23 20:41:17 -060042 def waitEvent(self, _timeout):
43 """Read event from the file."""
44 line = self.eventfile.readline().strip()
45 if not line:
46 return
47 try:
48 event_str = json.loads(line)['vars'].encode('utf-8')
49 event = pickle.loads(codecs.decode(event_str, 'base64'))
50 event_name = "%s.%s" % (event.__module__, event.__class__.__name__)
51 if event_name not in self.eventmask:
52 return
53 return event
54 except ValueError as err:
55 print("Failed loading ", line)
56 raise err
Patrick Williamsc124f4f2015-09-15 14:41:29 -050057
Patrick Williamsc0f7c042017-02-23 20:41:17 -060058 def runCommand(self, command_line):
59 """Emulate running a command on the server."""
60 name = command_line[0]
61
62 if name == "getVariable":
63 var_name = command_line[1]
64 variable = self.variables.get(var_name)
65 if variable:
66 return variable['v'], None
67 return None, "Missing variable %s" % var_name
68
69 elif name == "getAllKeysWithFlags":
70 dump = {}
71 flaglist = command_line[1]
72 for key, val in self.variables.items():
73 try:
74 if not key.startswith("__"):
75 dump[key] = {
76 'v': val['v'],
77 'history' : val['history'],
78 }
79 for flag in flaglist:
80 dump[key][flag] = val[flag]
81 except Exception as err:
82 print(err)
83 return (dump, None)
84
85 elif name == 'setEventMask':
86 self.eventmask = command_line[-1]
87 return True, None
88
89 else:
90 raise Exception("Command %s not implemented" % command_line[0])
91
92 def getEventHandle(self):
Patrick Williamsc124f4f2015-09-15 14:41:29 -050093 """
Patrick Williamsc0f7c042017-02-23 20:41:17 -060094 This method is called by toasterui.
95 The return value is passed to self.runCommand but not used there.
96 """
97 pass
Patrick Williamsc124f4f2015-09-15 14:41:29 -050098
Patrick Williamsc0f7c042017-02-23 20:41:17 -060099def main(argv):
100 with open(argv[-1]) as eventfile:
101 # load variables from the first line
102 variables = json.loads(eventfile.readline().strip())['allvariables']
Patrick Williamsc124f4f2015-09-15 14:41:29 -0500103
Patrick Williamsc0f7c042017-02-23 20:41:17 -0600104 params = namedtuple('ConfigParams', ['observe_only'])(True)
105 player = EventPlayer(eventfile, variables)
Patrick Williamsc124f4f2015-09-15 14:41:29 -0500106
Patrick Williamsc0f7c042017-02-23 20:41:17 -0600107 return toasterui.main(player, player, params)
Patrick Williamsc124f4f2015-09-15 14:41:29 -0500108
109# run toaster ui on our mock bitbake class
110if __name__ == "__main__":
Patrick Williamsc0f7c042017-02-23 20:41:17 -0600111 if len(sys.argv) != 2:
112 print("Usage: %s <event file>" % os.path.basename(sys.argv[0]))
Patrick Williamsc124f4f2015-09-15 14:41:29 -0500113 sys.exit(1)
114
Patrick Williamsc0f7c042017-02-23 20:41:17 -0600115 sys.exit(main(sys.argv))