Add owner

src/Blue.sol

@@ -7,6 +7,8 @@ import {IOracle} from "src/interfaces/IOracle.sol";
 import {MathLib} from "src/libraries/MathLib.sol";
 import {SafeTransferLib} from "src/libraries/SafeTransferLib.sol";
 
+import {Ownable} from "src/Ownable.sol";
+
 uint constant WAD = 1e18;
 
 uint constant alpha = 0.5e18;
@@ -29,7 +31,7 @@ function irm(uint utilization) pure returns (uint) {
     return utilization / 365 days;
 }
 
-contract Blue {
+contract Blue is Ownable {
     using MathLib for uint;
     using SafeTransferLib for IERC20;
 
@@ -52,6 +54,8 @@ contract Blue {
     // Interests last update (used to check if a market has been created).
     mapping(Id => uint) public lastUpdate;
 
+    constructor(address owner) Ownable(owner) {}
+
     // Markets management.
 
     function createMarket(Market calldata market) external {

src/Ownable.sol

@@ -0,0 +1,39 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+pragma solidity >=0.8.0;
+
+/// @title Ownable
+/// @author Morpho Labs
+/// @custom:contact [email protected]
+/// @dev Greatly inspired by Solmate and OZ implementations.
+abstract contract Ownable {
+    /* EVENTS */
+
+    event OwnershipTransferred(address indexed oldOwner, address indexed newOwner);
+
+    /* STORAGE */
+
+    address public owner;
+
+    /* MODIFIERS */
+
+    modifier onlyOwner() virtual {
+        require(msg.sender == owner, "not owner");
+        _;
+    }
+
+    /* CONSTRUCTOR */
+
+    constructor(address newOwner) {
+        owner = newOwner;
+
+        emit OwnershipTransferred(address(0), newOwner);
+    }
+
+    /* PUBLIC */
+
+    function transferOwnership(address newOwner) public virtual onlyOwner {
+        owner = newOwner;
+
+        emit OwnershipTransferred(msg.sender, newOwner);
+    }
+}

test/forge/Blue.t.sol

@@ -28,7 +28,7 @@ contract BlueTest is Test {
 
     function setUp() public {
         // Create Blue.
-        blue = new Blue();
+        blue = new Blue(msg.sender);
 
         // List a market.
         borrowableAsset = new ERC20("borrowable", "B", 18);
@@ -91,6 +91,30 @@ contract BlueTest is Test {
 
     // Tests
 
+    function testOwner(address owner) public {
+        Blue blue2 = new Blue(owner);
+
+        assertEq(blue2.owner(), owner, "owner");
+    }
+
+    function testTransferOwnership(address oldOwner, address newOwner) public {
+        Blue blue2 = new Blue(oldOwner);
+
+        vm.prank(oldOwner);
+        blue2.transferOwnership(newOwner);
+        assertEq(blue2.owner(), newOwner, "owner");
+    }
+
+    function testTransferOwnershipWhenNotOwner(address attacker, address newOwner) public {
+        vm.assume(attacker != address(0xdead));
+
+        Blue blue2 = new Blue(address(0xdead));
+
+        vm.prank(attacker);
+        vm.expectRevert("not owner");
+        blue2.transferOwnership(newOwner);
+    }
+
     function testSupply(uint amount) public {
         amount = bound(amount, 1, 2 ** 64);