implementation("uk.codecymru.drawbox:DrawBox-Enhanced:2.3.0")

val controller = remember { DrawController() }
DrawBox(drawController = controller, modifier = Modifier.fillMaxSize())

// The following optional parameters can be set to 
val controller = remember {
    DrawController(
        startingTool = CanvasTool.PENCIL,
        startingOpacity = 1f,
        startingStrokeWidth = 10f,
        startingColor = Color.Red,
        startingBackground = DrawBoxBackground.NoBackground,
        startingCanvasOpacity = 1f
    )
}

val enableUndo by controller.canUndo.collectAsState(false)
val enableRedo by controller.canRedo.collectAsState(false)

// Then create a button for each:
IconButton(onClick = controller::undo, enabled = enableUndo) {
    Icon(imageVector = Icons.Default.ArrowBack, contentDescription = "undo")
}
IconButton(onClick = controller::redo, enabled = enableRedo) {
    Icon(imageVector = Icons.Default.ArrowForward, contentDescription = "redo")
}

// To see the current tool
val tool by controller.canvasTool.collectAsState()

// To set the current tool, as an example:
TextButton(onClick = { controller.canvasTool.value = CanvasTool.PENCIL }) {
    Text("Pencil")
}
TextButton(onClick = { controller.canvasTool.value = CanvasTool.ERASER }) {
    Text("Eraser")
}
// etc... (PENCIL, BRUSH, ERASER, FILL, EYEDROPPER, SPRAY_CAN, TEXT, SHAPE_LINE, SHAPE_CURVE,
// SHAPE_RECT, SHAPE_ROUNDED_RECT, SHAPE_CIRCLE, SHAPE_POLYGON)

val bitmap: ImageBitmap = /* (...) */
drawController.open(bitmap)

// This provides the up-to-date drawn bitmap
controller.internalBitmap

// As an example, on the JVM target we could save this via:
TextButton(onClick = { saveImage(controller.internalBitmap) }) {
    Text("Save")
}

fun saveImage(bitmap: ImageBitmap) {
    val dialog = FileDialog(null as Frame?, "Save Image", FileDialog.SAVE)

    dialog.file = "image.png"
    dialog.isVisible = true

    val directory = dialog.directory
    val fileName = dialog.file

    if (directory != null && fileName != null) {
        val file = File(directory, fileName)

        ImageIO.write(
            bitmap.toAwtImage(),
            "png",
            file
        )
    }
}

// FinishDrawingUpdate means that this viewer won't show the user any in-progress strokes as they draw
DrawBoxViewer(
    controller = controller,
    subscription = DrawBoxSubscription.FinishDrawingUpdate,
    modifier = Modifier.size(200.dp).border(width = 1.dp, color = Color.Red)
)