Code.java

import java.util.*;

public class Code
{
    //instance variables
    private Peg[] code;
    private int numPositions;
    private int numColors;
    
    //constructor to generate a random code from the computer using an array
    public Code(int np, int nc)
    {
        numPositions = np;
        numColors = nc;
        code = new Peg[numPositions];
        //adds numPositions random pegs into array of code
        for(int i = 0; i < numPositions; i++){
            Peg p1 = new Peg(numColors);
            code[i] = p1;
        }
    }
    
    //constructor to generates code based on string passed in
    public Code(int np, int nc, String input)
    {
        numPositions = np;
        numColors = nc;
        
        if(input.length() != numPositions)
        {
            throw new IllegalArgumentException("Your input is not the correct length!");
        }
        
        this.code = new Peg[numPositions];
        
        //add pegs with color and add them to code array
        for(int i = 0; i < numPositions; i++)
        {
            Peg peg = new Peg(numColors, input.charAt(i));
            code[i] = peg;
        }
    }
    
    public Peg getPeg(int pos)
    {
        if(pos < 0 || pos >= numPositions)
        {
            throw new IllegalArgumentException("Violation of precondition: pos is out of bounds.");
        }
        
        return this.code[pos];
    }
    
    public boolean equals(Code other)
    {
        for(int i = 0; i < numPositions; i++)
        {
            if(!this.getPeg(i).equals(other.getPeg(i)))
            {
                return false;
            }
        }
        return true;
    }
    
    public String toString() 
    {
        String str = "";
        
        //loop through Code, get Peg and get its color
        for(int i = 0; i < numPositions; i++)
        {
            Peg p1 = this.getPeg(i);
            char c = p1.getColor();
            str += c;
        }
        
        return str;
    }
    
    public int size()
    {
        return numPositions;
    }
}