GlobalsPersistence.java

/*
 * Copyright 2012 Red Hat, Inc. and/or its affiliates.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.drools.workbench.screens.globals.backend.server.util;

import org.drools.workbench.models.commons.backend.packages.PackageNameParser;
import org.drools.workbench.models.commons.backend.packages.PackageNameWriter;
import org.uberfire.commons.data.Pair;
import org.kie.workbench.common.services.datamodel.backend.server.builder.util.GlobalsParser;
import org.drools.workbench.screens.globals.model.Global;
import org.drools.workbench.screens.globals.model.GlobalsModel;

import java.util.ArrayList;
import java.util.List;

/**
 * This class persists the rule model to DRL and back
 */
public class GlobalsPersistence {

    private static final GlobalsPersistence INSTANCE = new GlobalsPersistence();

    protected GlobalsPersistence() {
    }

    public static GlobalsPersistence getInstance() {
        return INSTANCE;
    }

    public String marshal( final GlobalsModel model ) {
        final StringBuilder sb = new StringBuilder();

        PackageNameWriter.write( sb,
                                 model );

        for ( Global global : model.getGlobals() ) {
            sb.append( "global " ).append( global.getClassName() ).append( " " ).append( global.getAlias() ).append( ";\n" );
        }
        return sb.toString();
    }

    public GlobalsModel unmarshal( final String content ) {
        //De-serialize model
        final List<Pair<String, String>> parsedGlobalsContent = GlobalsParser.parseGlobals( content );
        final List<Global> globals = makeGlobals( parsedGlobalsContent );
        final GlobalsModel model = new GlobalsModel();
        model.setGlobals( globals );

        //De-serialize Package name
        final String packageName = PackageNameParser.parsePackageName( content );
        model.setPackageName( packageName );
        return model;
    }

    private List<Global> makeGlobals( final List<Pair<String, String>> parsedGlobalsContent ) {
        final List<Global> globals = new ArrayList<Global>();
        for ( Pair<String, String> p : parsedGlobalsContent ) {
            globals.add( new Global( p.getK1(),
                                     p.getK2() ) );
        }
        return globals;
    }

}