|
@@ -1,22 +1,13 @@
|
|
|
pragma solidity ^0.4.4;
|
|
|
-contract Migrations {
|
|
|
- address public owner;
|
|
|
- uint public last_completed_migration;
|
|
|
+contract Migrations is Ownable {
|
|
|
+ uint public lastCompletedMigration;
|
|
|
|
|
|
- modifier restricted() {
|
|
|
- if (msg.sender == owner) _;
|
|
|
+ function setCompleted(uint completed) onlyOwner {
|
|
|
+ lastCompletedMigration = completed;
|
|
|
}
|
|
|
|
|
|
- function Migrations() {
|
|
|
- owner = msg.sender;
|
|
|
- }
|
|
|
-
|
|
|
- function setCompleted(uint completed) restricted {
|
|
|
- last_completed_migration = completed;
|
|
|
- }
|
|
|
-
|
|
|
- function upgrade(address new_address) restricted {
|
|
|
- Migrations upgraded = Migrations(new_address);
|
|
|
- upgraded.setCompleted(last_completed_migration);
|
|
|
+ function upgrade(address newAddress) onlyOwner {
|
|
|
+ Migrations upgraded = Migrations(newAddress);
|
|
|
+ upgraded.setCompleted(lastCompletedMigration);
|
|
|
}
|
|
|
}
|