Update Smart contract

IT20077624/contract/contracts/Hr.sol

+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.0;
+
+contract StressManagementHRSystem {
+    address public admin;
+    bool public loginSuccess;
+    struct Employee {
+        string name;
+        uint256 employeeId;
+        string department;
+        uint256 stressLevel;
+        address[] assignedDocs;
+        bool isActive;
+    }
+
+    struct Doctor {
+        string name;
+        uint256 docId;
+        string specialty;
+        address[] assignedEmps;
+        bool isActive;
+    }
+
+    mapping(address => Employee) public employees;
+    mapping(uint256 => Employee) public employeesIdx;
+    mapping(address => Doctor) public doctors;
+    mapping(address => address[]) public employeeToDoctors;
+    mapping(address => address[]) public doctorToEmployees;
+    mapping(uint256 => address) public employeeIdsToAddress;
+    mapping(address => bool) public loggedIn;
+
+    uint256 public numEmployees;
+    uint256 public numDoctors;
+
+    address[] public docs;
+    address[] public emps;
+
+    event EmployeeRegistered(address indexed employee, string name);
+    event DoctorRegistered(address indexed doctor, string name);
+
+    modifier onlyAdmin() {
+        require(msg.sender == admin, "Only admin can perform this operation");
+        _;
+    }
+
+    modifier onlyEmployee() {
+        require(employees[msg.sender].isActive, "Employee is not active");
+        _;
+    }
+
+    modifier onlyDoctor() {
+        require(doctors[msg.sender].isActive, "Doctor is not active");
+        _;
+    }
+
+    constructor() {
+        admin = msg.sender;
+        numDoctors = 0;
+        numEmployees = 0;
+    }
+
+    function registerEmployee(string memory _name, uint256 _employeeId, string memory _department, uint256 _stressLevel) public {
+        require(employees[msg.sender].employeeId == 0, "Employee already registered");
+        address[] memory a = new address[](0);
+        employees[msg.sender] = Employee(_name, _employeeId, _department, _stressLevel, a, false);
+        emps.push(msg.sender);
+        numEmployees++;
+        emit EmployeeRegistered(msg.sender, _name);
+    }
+
+    function approveEmployee(address employee) public onlyAdmin {
+        require(employees[employee].employeeId != 0, "Employee not registered");
+        employees[employee].isActive = true;
+    }
+
+    function registerDoctor(string memory _name, uint256 _docID, string memory _specialty) public {
+        require(doctors[msg.sender].docId == 0, "Doctor already registered");
+        address[] memory a = new address[](0);
+
+        doctors[msg.sender] = Doctor(_name, _docID, _specialty, a, false);
+        docs.push(msg.sender);
+        numDoctors++;
+        emit DoctorRegistered(msg.sender, _name);
+    }
+
+    function approveDoctor(address doctor) public onlyAdmin {
+        require(doctors[doctor].docId != 0, "Doctor not registered");
+        doctors[doctor].isActive = true;
+    }
+
+    function assignEmployeeToDoctor(address employee, address doctor) public  {
+        require(employees[employee].employeeId != 0, "Employee not registered");
+        require(doctors[doctor].docId != 0, "Doctor not registered");
+        employeeToDoctors[employee].push(doctor);
+        doctorToEmployees[doctor].push(employee);
+    }
+
+    function getAssignedDoctors(address _addr) public view returns (address[] memory) {
+        return employeeToDoctors[_addr];
+    }
+
+    function getAssignedEmployees(address _addr) public view returns (address[] memory) {
+        return doctorToEmployees[_addr];
+    }
+
+    function updateEmployeeStress(uint256 _employeeId, uint256 _newStressLevel) public onlyAdmin {
+        address employeeAddress = employeeIdsToAddress[_employeeId];
+        require(employeeAddress != address(0), "Employee not found");
+        employees[employeeAddress].stressLevel = _newStressLevel;
+    }
+
+
+    function employeeLogin(address _userAddress) public view returns (bool) {
+
+        if (employees[_userAddress].isActive) {
+
+            return true;
+
+        } else {
+
+            return false;
+
+        }
+
+    }
+
+    function doctorLogin(address _userAddress) public view returns (bool) {
+        if (doctors[_userAddress].isActive) {
+
+            return true;
+
+        } else {
+
+            return false;
+
+        }
+    }
+
+    function adminLogin() public {
+        require(msg.sender == admin, "Only admin can log in");
+        loggedIn[msg.sender] = true;
+        loginSuccess = true;
+    }
+
+
+    function getLoginStatus() public view returns (bool) {
+        return loggedIn[msg.sender];
+    }
+
+
+    function logout() public {
+        loggedIn[msg.sender] = false;
+        loginSuccess = false;
+
+    }
+
+    function getAllEmployees() public view  returns (Employee[] memory) {
+        Employee[] memory employeeList = new Employee[](numEmployees);
+        uint256 currentIndex = 0;
+        for (uint256 i = 0; i < numEmployees; i++)
+        {
+            address empAdds = emps[i];
+            employeeList[currentIndex] = employees[empAdds];
+            currentIndex++;
+        }
+        return employeeList;
+    }
+
+    function getAllDoctors() public view  returns (Doctor[] memory) {
+        Doctor[] memory employeeList = new Doctor[](numDoctors);
+        uint256 currentIndex = 0;
+        for (uint256 i = 0; i < numDoctors; i++)
+        {
+            address empAdds = docs[i];
+            employeeList[currentIndex] = doctors[empAdds];
+            currentIndex++;
+        }
+        return employeeList;
+    }
+
+    function getEmployeeAddress(uint256 _userID) public view returns (address){
+        address adr = msg.sender;
+        for (uint256 i = 0; i < numEmployees; i++)
+        {
+            address empAdds = emps[i];
+            Employee memory emp = employees[empAdds];
+            if (emp.employeeId == _userID)
+            {
+                adr = empAdds;
+            }
+        }
+        return adr;
+    }
+
+    function getDoctorAddress(uint256 _userID) public view returns (address){
+        address adr = msg.sender;
+        for (uint256 i = 0; i < numDoctors; i++)
+        {
+            address empAdds = docs[i];
+            Doctor memory emp = doctors[empAdds];
+            if (emp.docId == _userID)
+            {
+                adr = empAdds;
+            }
+        }
+        return adr;
+    }
+
+    function assignDocToEmployee(address _empAddress, address docAddress) public {
+        Employee storage employee = employees[_empAddress];
+        for (uint256 i = 0; i < employee.assignedDocs.length; i++) {
+            require(employee.assignedDocs[i] != docAddress, "Address is already in assignedDocs");
+        }
+        employee.assignedDocs.push(docAddress);
+
+        Doctor storage doc = doctors[docAddress];
+        for (uint256 i = 0; i < doc.assignedEmps.length; i++) {
+            require(doc.assignedEmps[i] != _empAddress, "Address is already in assignedDocs");
+        }
+        doc.assignedEmps.push(_empAddress);
+    }
+
+
+    function getEmployee(address _empAddress) public view returns(Employee memory)
+    {
+        return employees[_empAddress];
+    }
+
+    function getDoctor(address _docAddress) public view returns(Doctor memory)
+    {
+        return doctors[_docAddress];
+    }
+
+
+}

IT20077624/contract/migrations/2_migrate_hr.js

+const Hr = artifacts.require("StressManagementHRSystem");
+
+module.exports = function (deployer) {
+    deployer.deploy(Hr);
+};
\ No newline at end of file

IT20077624/contract/truffle-config.js

+/**
+ * Use this file to configure your truffle project. It's seeded with some
+ * common settings for different networks and features like migrations,
+ * compilation, and testing. Uncomment the ones you need or modify
+ * them to suit your project as necessary.
+ *
+ * More information about configuration can be found at:
+ *
+ * https://trufflesuite.com/docs/truffle/reference/configuration
+ *
+ * Hands-off deployment with Infura
+ * --------------------------------
+ *
+ * Do you have a complex application that requires lots of transactions to deploy?
+ * Use this approach to make deployment a breeze 🏖️:
+ *
+ * Infura deployment needs a wallet provider (like @truffle/hdwallet-provider)
+ * to sign transactions before they're sent to a remote public node.
+ * Infura accounts are available for free at 🔍: https://infura.io/register
+ *
+ * You'll need a mnemonic - the twelve word phrase the wallet uses to generate
+ * public/private key pairs. You can store your secrets 🤐 in a .env file.
+ * In your project root, run `$ npm install dotenv`.
+ * Create .env (which should be .gitignored) and declare your MNEMONIC
+ * and Infura PROJECT_ID variables inside.
+ * For example, your .env file will have the following structure:
+ *
+ * MNEMONIC = <Your 12 phrase mnemonic>
+ * PROJECT_ID = <Your Infura project id>
+ *
+ * Deployment with Truffle Dashboard (Recommended for best security practice)
+ * --------------------------------------------------------------------------
+ *
+ * Are you concerned about security and minimizing rekt status 🤔?
+ * Use this method for best security:
+ *
+ * Truffle Dashboard lets you review transactions in detail, and leverages
+ * MetaMask for signing, so there's no need to copy-paste your mnemonic.
+ * More details can be found at 🔎:
+ *
+ * https://trufflesuite.com/docs/truffle/getting-started/using-the-truffle-dashboard/
+ */
+
+// require('dotenv').config();
+// const { MNEMONIC, PROJECT_ID } = process.env;
+
+// const HDWalletProvider = require('@truffle/hdwallet-provider');
+
+module.exports = {
+  /**
+   * Networks define how you connect to your ethereum client and let you set the
+   * defaults web3 uses to send transactions. If you don't specify one truffle
+   * will spin up a managed Ganache instance for you on port 9545 when you
+   * run `develop` or `test`. You can ask a truffle command to use a specific
+   * network from the command line, e.g
+   *
+   * $ truffle test --network <network-name>
+   */
+
+  networks: {
+    // Useful for testing. The `development` name is special - truffle uses it by default
+    // if it's defined here and no other network is specified at the command line.
+    // You should run a client (like ganache, geth, or parity) in a separate terminal
+    // tab if you use this network and you must also set the `host`, `port` and `network_id`
+    // options below to some value.
+    //
+    development: {
+     host: "127.0.0.1",     // Localhost (default: none)
+     port: 7545,            // Standard Ethereum port (default: none)
+     network_id: "5777",       // Any network (default: none)
+    },
+    //
+    // An additional network, but with some advanced options…
+    // advanced: {
+    //   port: 8777,             // Custom port
+    //   network_id: 1342,       // Custom network
+    //   gas: 8500000,           // Gas sent with each transaction (default: ~6700000)
+    //   gasPrice: 20000000000,  // 20 gwei (in wei) (default: 100 gwei)
+    //   from: <address>,        // Account to send transactions from (default: accounts[0])
+    //   websocket: true         // Enable EventEmitter interface for web3 (default: false)
+    // },
+    //
+    // Useful for deploying to a public network.
+    // Note: It's important to wrap the provider as a function to ensure truffle uses a new provider every time.
+    // goerli: {
+    //   provider: () => new HDWalletProvider(MNEMONIC, `https://goerli.infura.io/v3/${PROJECT_ID}`),
+    //   network_id: 5,       // Goerli's id
+    //   confirmations: 2,    // # of confirmations to wait between deployments. (default: 0)
+    //   timeoutBlocks: 200,  // # of blocks before a deployment times out  (minimum/default: 50)
+    //   skipDryRun: true     // Skip dry run before migrations? (default: false for public nets )
+    // },
+    //
+    // Useful for private networks
+    // private: {
+    //   provider: () => new HDWalletProvider(MNEMONIC, `https://network.io`),
+    //   network_id: 2111,   // This network is yours, in the cloud.
+    //   production: true    // Treats this network as if it was a public net. (default: false)
+    // }
+  },
+
+  // Set default mocha options here, use special reporters, etc.
+  mocha: {
+    // timeout: 100000
+  },
+
+  // Configure your compilers
+  compilers: {
+    solc: {
+      version: "0.8.0",      // Fetch exact version from solc-bin (default: truffle's version)
+      // docker: true,        // Use "0.5.1" you've installed locally with docker (default: false)
+      // settings: {          // See the solidity docs for advice about optimization and evmVersion
+      //  optimizer: {
+      //    enabled: false,
+      //    runs: 200
+      //  },
+      //  evmVersion: "byzantium"
+      // }
+    }
+  },
+
+  // Truffle DB is currently disabled by default; to enable it, change enabled:
+  // false to enabled: true. The default storage location can also be
+  // overridden by specifying the adapter settings, as shown in the commented code below.
+  //
+  // NOTE: It is not possible to migrate your contracts to truffle DB and you should
+  // make a backup of your artifacts to a safe location before enabling this feature.
+  //
+  // After you backed up your artifacts you can utilize db by running migrate as follows:
+  // $ truffle migrate --reset --compile-all
+  //
+  // db: {
+  //   enabled: false,
+  //   host: "127.0.0.1",
+  //   adapter: {
+  //     name: "indexeddb",
+  //     settings: {
+  //       directory: ".db"
+  //     }
+  //   }
+  // }
+};