Method UpdateStatement

pkg/dsl/ast/schema.go:115–190

UpdateStatement updates a statement of a specific type (permission, relation, or attribute) for a given entity within the schema. It either updates an existing statement or appends a new one if not found.

(entityName string, newStmt Statement)

// UpdateStatement updates a statement of a specific type (permission, relation, or attribute)
// for a given entity within the schema. It either updates an existing statement or appends a new one if not found.
func (sch *Schema) UpdateStatement(entityName string, newStmt Statement) error {
	// Iterate through all statements in the schema to find the one matching the entity name.
	for _, statement := range sch.Statements {
		if statement.GetName() == entityName {
			// Convert the generic statement interface to a specific EntityStatement type.
			entityStmt, ok := statement.(*EntityStatement)
			if !ok {
				return errors.New(base.ErrorCode_ERROR_CODE_CANNOT_CONVERT_TO_ENTITY_STATEMENT.String())
			}

			// Construct a unique reference key for the new statement.
			referenceKey := fmt.Sprintf("%s#%s", entityName, newStmt.GetName())

			// Check if a reference for the statement already exists within the schema.
			if !sch.GetReferences().IsReferenceExist(referenceKey) {
				return errors.New(base.ErrorCode_ERROR_CODE_REFERENCE_NOT_FOUND.String())
			}

			// Based on the statement type, update the corresponding list in the EntityStatement.
			switch newStmt.StatementType() {
			case PERMISSION_STATEMENT, RELATION_STATEMENT, ATTRIBUTE_STATEMENT:
				var stmts *[]Statement // Pointer to the slice of statements to update.

				// Assign the correct slice based on the type of the new statement.
				switch newStmt.StatementType() {
				case PERMISSION_STATEMENT:
					stmts = &entityStmt.PermissionStatements
				case RELATION_STATEMENT:
					stmts = &entityStmt.RelationStatements
				case ATTRIBUTE_STATEMENT:
					stmts = &entityStmt.AttributeStatements
				}

				// Flag to check if the statement has been updated.
				updated := false

				// Iterate over the statements to find and update the one with the matching name.
				for i, stmt := range *stmts {
					if stmt.GetName() == newStmt.GetName() {
						(*stmts)[i] = newStmt
						updated = true
						break // Stop iterating once the statement is updated.
					}
				}

				// If the statement was not found and updated, append it to the slice.
				if !updated {
					*stmts = append(*stmts, newStmt)
				}

				// Update the reference in the schema based on the statement type.
				switch newStmt.StatementType() {
				case PERMISSION_STATEMENT:
					return sch.GetReferences().UpdatePermissionReference(referenceKey)
				case RELATION_STATEMENT:
					if rs, ok := newStmt.(*RelationStatement); ok {
						return sch.GetReferences().UpdateRelationReferences(referenceKey, rs.RelationTypes)
					}