4. Write Klaystagram Smart Contract

Search…
1) Background
2) Contract setup
3) Set events and data structure
4) Write functions
uploadPhoto
transferOwnership
getPhoto
1) Background
We will make a simple contract called "Klaystagram".
PhotoData struct is defined to store various photo data.
User can upload photo and transfer the ownership photo via uploadPhoto and transferOwnership functions.
2) Contract setup
Specify solidity version. We recommend using 0.5.6 stable version.
We will make use of ERC721 standard to build non-fungible tokens.
Import ERC721.sol and ERC721Enumerable.sol
Check out detailed information about ERC721 at erc721.org​
1
pragma solidity 0.5.6;
2
​
3
import "./ERC721/ERC721.sol";
4
import "./ERC721/ERC721Enumerable.sol";
5
​
6
contract Klaystagram is ERC721, ERC721Enumerable {
Copied!
3) Set events and data structure
We need to set up an event to keep track of activities on blockchain.
As for data structure, mapping _photoList takes a uint256 tokenId to map a specific PhotoData struct. By defining PhotoUploaded event, transaction receipt will log this event whenever function containing this is called.
1
event PhotoUploaded (uint indexed tokenId, bytes photo, string title, string location, string description, uint256 timestamp);
2
​
3
mapping (uint256 => PhotoData) private _photoList;
4
​
5
struct PhotoData {
6
    uint256 tokenId;                       // Unique token id, starts from 1 and increases by 1
7
    address[] ownerHistory;                // History of all previous owners
8
    bytes photo;                           // Image source
9
    string title;                          // Title of photo
10
    string location;                       // Location where photo is taken
11
    string description;                    // Short description about the photo
12
    uint256 timestamp;                     // Uploaded time
13
}
Copied!
4) Write functions
Let's write some functions that interact with the contract. In this tutorial let us only consider two functions: uploadPhoto and transferOwnership. Check out Klaystagram.sol to see the whole set of functions.
uploadPhoto
uploadPhoto function takes 4 arguments including photo's image source. To keep things simple, tokenId will start from 1 and will increase by 1.
_mint function is from ERC721 contract. It creates a new token and assign it to a specific address, which in this case, msg.sender. In this application, logged in user will create transaction with their own private key. So msg.sender will be the user's public address.
Finally, initialize PhotoData struct, locate it inside _photoList mapping, and push the owner address into ownerHistory array. And don't forget to emit the event we just created. As mentioned above, this event will be included in transaction receipt.
1
function uploadPhoto(bytes memory photo, string memory title, string memory location, string memory description) public {
2
    uint256 tokenId = totalSupply() + 1;
3
​
4
    _mint(msg.sender, tokenId);
5
​
6
    address[] memory ownerHistory;
7
​
8
    PhotoData memory newPhotoData = PhotoData({
9
        tokenId : tokenId,
10
        ownerHistory : ownerHistory,
11
        photo : photo,
12
        title: title,
13
        location : location,
14
        description : description,
15
        timestamp : now
16
    });
17
​
18
    _photoList[tokenId] = newPhotoData;
19
    _photoList[tokenId].ownerHistory.push(msg.sender);
20
​
21
    emit PhotoUploaded(tokenId, photo, title, location, description, now);
22
}
Copied!
transferOwnership
Let's take a look at transferOwnership function. When transferring photo ownership, we need to do two things. First, we have to reassign the owner, and then we have to push new owner address into ownerHistory array.
To do this, transferOwnership first calls safeTransferFrom function from ERC721 standard, which eventually calls transferFrom function. As mentioned above, right after token transfer is successfully done, we have to push new owner information into ownerHistory array, and that is exactly why transferFrom is overridden as below.
1
/**
2
  * @notice safeTransferFrom function checks whether receiver is able to handle ERC721 tokens,
3
  *  thus less possibility of tokens being lost. After checking is done, it will call transferFrom function defined below
4
  */
5
function transferOwnership(uint256 tokenId, address to) public returns(uint, address, address, address) {
6
    safeTransferFrom(msg.sender, to, tokenId);
7
}
8
​
9
/**
10
  * @notice Recommend using transferOwnership, which uses safeTransferFrom function
11
  * @dev Override transferFrom function to make sure that every time ownership transfers
12
  *  new owner address gets pushed into ownerHistory array
13
  */
14
function transferFrom(address from, address to, uint256 tokenId) public {
15
    super.transferFrom(from, to, tokenId);
16
    _photoList[tokenId].ownerHistory.push(to);
17
}
Copied!
getPhoto
Finally, let's make a getter function that fetches data stored in the smart contract. By calling a single function, we want to fetch every information regarding a specific photo. So getPhoto function takes an index(token id) as an argument and returns every element in PhotoData struct.
1
function getPhoto(uint tokenId) public view
2
returns(uint256, address[] memory, bytes memory, string memory, string memory, string memory, uint256) {
3
    require(_photoList[tokenId].tokenId != 0, "Photo does not exist");
4
    return (
5
        _photoList[tokenId].tokenId,
6
        _photoList[tokenId].ownerHistory,
7
        _photoList[tokenId].photo,
8
        _photoList[tokenId].title,
9
        _photoList[tokenId].location,
10
        _photoList[tokenId].description,
11
        _photoList[tokenId].timestamp);
12
}
Copied!
This is it, now we can deploy this contract!