Migrate company configuration

from enum import Enum
from pydantic import BaseModel, Field
from harnest.authoring import ChangePlan, ProjectContext, ProjectPack

class Environment(str, Enum):
    DEVELOPMENT = "development"
    PRODUCTION = "production"

class Options(BaseModel):
    team: str = Field(min_length=1)
    environment: Environment = Environment.DEVELOPMENT

class Settings(BaseModel):
    team: str = Field(min_length=1)
    environment: Environment

pack = ProjectPack(
    name="acme",
    schema_version=2,
    options=Options,
    config_file="acme-agent.yaml",
    config_model=Settings,
)

@pack.initialize
def initialize(context: ProjectContext) -> ChangePlan:
    return ChangePlan(
        context.yaml.set(
            "acme-agent.yaml", key=("team",), value=context.options.team,
        ),
        context.yaml.set(
            "acme-agent.yaml",
            key=("environment",),
            value=context.options.environment.value,
        ),
    )

@pack.migration(from_version=1, to_version=2)
def migrate_ownership(context: ProjectContext) -> ChangePlan:
    return ChangePlan(
        context.yaml.rename_key(
            "acme-agent.yaml", source=("owner",), destination=("team",),
        ),
    )