aws-scripts/ec2-attached-volumes-prune.php at master · ChartBlocks/aws-scripts · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
<?php

use Zend\Console\Getopt;
use Zend\Console\Console;
use Zend\Console\ColorInterface as Color;
use Aws\Ec2\Ec2Client;
use CbAws\Ec2\Ec2ClientHelper;

require(__DIR__ . '/vendor/autoload.php');

$console = Console::getInstance();
$opt = new Getopt(array(
    'key|O-s' => 'AWS access key',
    'secret|W-s' => 'AWS secret key',
    'instance|i-s' => 'Instance to backup',
    'region|r-s' => 'Region of host',
    'age|t-s' => 'Max age, in seconds (default = 1 week)',
    'dry-run|d' => 'Dry run, do not delete anything',
        ));
$opt->parse();

try {
    $key = empty($opt->key) ? getenv('AWS_ACCESS_KEY') : $opt->key;
    $secret = empty($opt->secret) ? getenv('AWS_SECRET_KEY') : $opt->secret;
    $region = empty($opt->region) ? getenv('AWS_REGION') : $opt->region;
    $instanceId = $opt->instance;
    $age = empty($opt->age) ? 86400 * 7 : (int) $opt->age;
    $dryRun = (bool) $opt->{'dry-run'};

    if ($age <= 0) {
        $console->writeLine('Invalid age parameter', Color::RED);
        echo $opt->getUsageMessage();
        exit(1);
    }

    $maxAge = new DateTime();
    $maxAge->sub(new DateInterval('PT' . $age . 'S'));

    $client = Ec2Client::factory(array(
                'key' => $key,
                'secret' => $secret,
                'region' => $region
    ));

    $clientHelper = new Ec2ClientHelper($client);

    if (empty($instanceId)) {
        $hostname = gethostname() . '.';
        $instance = $clientHelper->getInstanceByHostname($hostname);
        $instanceId = $instance['InstanceId'];
    } else {
        $instance = $clientHelper->getInstanceById($instanceId);
    }

    $instanceName = $clientHelper->resolveInstanceName($instance);
    $console->writeLine("Pruning snapshots on $instanceName older than " . $maxAge->format('c'));

    $volumes = $clientHelper->getVolumesByInstance($instanceId);
    if (count($volumes) === 0) {
        throw new Exception('No volumes found');
    }

    $volumeIds = array();
    foreach ($volumes as $volume) {
        array_push($volumeIds, $volume['VolumeId']);
    }

    $console->writeLine("Found " . count($volumeIds) . " volumes", Color::GRAY);
    $snapshots = $clientHelper->getCompleteSnapshotsForVolumes($volumeIds);
    $console->writeLine("Found " . count($snapshots) . " snapshots", Color::GRAY);

    $toDelete = array();
    foreach ($snapshots as $snapshot) {
        $created = new DateTime($snapshot['StartTime']);
        if ($created < $maxAge) {
            array_push($toDelete, $snapshot['SnapshotId']);
        }
    }

    $console->writeLine("Found " . count($toDelete) . " old snapshots", Color::YELLOW);


    foreach ($toDelete as $snapshotId) {
        $console->writeLine('Deleting' . ($dryRun ? ' (dry-run) ' : ' ') . $snapshotId, Color::GRAY);
        if (!$dryRun) {
            $client->deleteSnapshot(array('SnapshotId' => $snapshotId));
        }
    }

    $console->writeLine('Done' . ($dryRun ? ' (dry-run)' : ''), Color::GREEN);
} Catch (Exception $e) {
    $console->writeLine($e->getMessage(), Color::RED);
    exit(1);
}