1 /***********************************************************************
2  * Copyright (c) 2013-2025 General Atomics Integrated Intelligence, Inc.
3  * All rights reserved. This program and the accompanying materials
4  * are made available under the terms of the Apache License, Version 2.0
5  * which accompanies this distribution and is available at
6  * https://www.apache.org/licenses/LICENSE-2.0
7  ***********************************************************************/
8 
9 package org.locationtech.geomesa.tools.data
10 
11 import org.geotools.api.data.DataStore
12 import org.geotools.api.feature.simple.SimpleFeatureType
13 import org.locationtech.geomesa.tools._
14 import org.locationtech.geomesa.tools.data.CreateSchemaCommand.{CreateSchemaParams, SchemaOptionsCommand}
15 import org.locationtech.geomesa.tools.utils.CLArgResolver
16 import org.locationtech.geomesa.utils.geotools.RichSimpleFeatureType.RichSimpleFeatureType
17 import org.locationtech.geomesa.utils.geotools.SimpleFeatureTypes
18 
19 import java.io.IOException
20 
21 trait CreateSchemaCommand[DS <: DataStore] extends DataStoreCommand[DS] with SchemaOptionsCommand {
22 
23   override val name = "create-schema"
24   override def params: CreateSchemaParams
25 
26   override def execute(): Unit = {
27     val sft = CLArgResolver.getSft(params.spec, params.featureName)
28     Option(params.dtgField).foreach(sft.setDtgField)
29     setBackendSpecificOptions(sft)
30     withDataStore(createSchema(_, sft))
31   }
32 
33   protected def createSchema(ds: DS, sft: SimpleFeatureType): Unit = {
34     lazy val sftString = SimpleFeatureTypes.encodeType(sft)
35     Command.user.info(s"Creating '${sft.getTypeName}' with spec '$sftString'. Just a few moments...")
36 
37     if (try { ds.getSchema(sft.getTypeName) == null } catch { case _: IOException => true }) {
38       ds.createSchema(sft)
39       if (try { ds.getSchema(sft.getTypeName) != null } catch { case _: IOException => false }) {
40         Command.user.info(s"Created schema '${sft.getTypeName}'")
41       } else {
42         Command.user.error(s"Could not create schema '${sft.getTypeName}'")
43       }
44     } else {
45       Command.user.error(s"Schema '${sft.getTypeName}' already exists in the data store")
46     }
47   }
48 }
49 
50 object CreateSchemaCommand {
51 
52   // @Parameters(commandDescription = "Create a GeoMesa feature type")
53   trait CreateSchemaParams extends RequiredFeatureSpecParam with OptionalTypeNameParam with OptionalDtgParam
54 
55   trait SchemaOptionsCommand {
56     protected def setBackendSpecificOptions(sft: SimpleFeatureType): Unit = {}
57   }
58 }