If there is no new commit in the repo, we should not wait for a signature, as there is nothing to sign. Fixes: https://pagure.io/pungi/issue/1046 Signed-off-by: Lubomír Sedlář <lsedlar@redhat.com>
		
			
				
	
	
		
			88 lines
		
	
	
		
			2.5 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
			
		
		
	
	
			88 lines
		
	
	
		
			2.5 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
#!/usr/bin/env python
 | 
						|
# -*- coding: utf-8 -*-
 | 
						|
 | 
						|
"""
 | 
						|
Messaging hook to block compose progress until an ostree commit is signed.
 | 
						|
 | 
						|
The signing is implemented by robosignatory, which listens on the message bus
 | 
						|
and reacts to messages about new commits. It will create a signature and then
 | 
						|
update the ref in the repo to point to the new commit.
 | 
						|
 | 
						|
This script should not be used if Pungi is updating the reference on its own
 | 
						|
(since that does not leave time for the signature).
 | 
						|
"""
 | 
						|
 | 
						|
from __future__ import print_function
 | 
						|
 | 
						|
import argparse
 | 
						|
import datetime
 | 
						|
import fedmsg.config
 | 
						|
import json
 | 
						|
import os
 | 
						|
import sys
 | 
						|
import time
 | 
						|
 | 
						|
 | 
						|
def is_ref_updated(ref_file, commit):
 | 
						|
    """The ref is updated when the file points to the correct commit."""
 | 
						|
    try:
 | 
						|
        with open(ref_file) as f:
 | 
						|
            return f.read().strip() == commit
 | 
						|
    except IOError:
 | 
						|
        # Failed to open the file, probably it does not exist, so let's just
 | 
						|
        # wait more.
 | 
						|
        return False
 | 
						|
 | 
						|
 | 
						|
def ts_log(msg):
 | 
						|
    print("%s: %s" % (datetime.datetime.utcnow(), msg))
 | 
						|
 | 
						|
 | 
						|
if __name__ == '__main__':
 | 
						|
    parser = argparse.ArgumentParser()
 | 
						|
    parser.add_argument('cmd')
 | 
						|
    opts = parser.parse_args()
 | 
						|
 | 
						|
    if opts.cmd != 'ostree':
 | 
						|
        # Not an announcement of new ostree commit, nothing to do.
 | 
						|
        sys.exit()
 | 
						|
 | 
						|
    try:
 | 
						|
        data = json.load(sys.stdin)
 | 
						|
    except ValueError:
 | 
						|
        print('Failed to decode data', file=sys.stderr)
 | 
						|
        sys.exit(1)
 | 
						|
 | 
						|
    repo = data['local_repo_path']
 | 
						|
    commit = data['commitid']
 | 
						|
    if not commit:
 | 
						|
        print("No new commit was created, nothing will get signed.")
 | 
						|
        sys.exit(0)
 | 
						|
 | 
						|
    path = '%s/objects/%s/%s.commitmeta' % (repo, commit[:2], commit[2:])
 | 
						|
 | 
						|
    config = fedmsg.config.load_config()
 | 
						|
    config['active'] = True           # Connect out to a fedmsg-relay instance
 | 
						|
    config['cert_prefix'] = 'releng'  # Use this cert.
 | 
						|
    fedmsg.init(**config)
 | 
						|
    topic = 'compose.%s' % opts.cmd.replace('-', '.').lower()
 | 
						|
 | 
						|
    count = 0
 | 
						|
    while not os.path.exists(path):
 | 
						|
        ts_log("Commit not signed yet, waiting...")
 | 
						|
        count += 1
 | 
						|
        if count >= 60:  # Repeat every 5 minutes
 | 
						|
            print('Repeating notification')
 | 
						|
            fedmsg.publish(topic=topic, modname='pungi', msg=data)
 | 
						|
            count = 0
 | 
						|
        time.sleep(5)
 | 
						|
 | 
						|
    print("Found signature, waiting for ref to be updated.")
 | 
						|
 | 
						|
    ref_file = os.path.join(repo, "refs/heads", data["ref"])
 | 
						|
    while not is_ref_updated(ref_file, commit):
 | 
						|
        ts_log("Ref is not yet up-to-date, waiting...")
 | 
						|
        time.sleep(5)
 | 
						|
 | 
						|
    print("Ref is up-to-date. All done!")
 |