android/src/com/android/tools/idea/editors/strings/table/StringResourceTableModel.java

/*
 * Copyright (C) 2014 The Android Open Source Project
 *
 * 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 com.android.tools.idea.editors.strings.table;

import com.android.tools.idea.configurations.LocaleMenuAction;
import com.android.tools.idea.editors.strings.StringResourceData;
import com.android.tools.idea.editors.strings.StringResourceViewPanel;
import com.android.tools.idea.rendering.Locale;
import org.jetbrains.android.facet.AndroidFacet;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import javax.swing.table.AbstractTableModel;

public class StringResourceTableModel extends AbstractTableModel {
  @Nullable private StringResourceData myData;

  public StringResourceTableModel(@Nullable StringResourceData data) {
    myData = data;
  }

  public void setData(@NotNull StringResourceData data) {
    myData = data;
  }

  @NotNull
  public String keyOfRow(int row) {
    return myData == null ? "" : myData.getKeys().get(row);
  }

  @Nullable
  public Locale localeOfColumn(int column) {
    return (column < ConstantColumn.COUNT || myData == null) ? null : myData.getLocales().get(column - ConstantColumn.COUNT);
  }

  @Override
  public int getRowCount() {
    return myData == null ? 0 : myData.getKeys().size();
  }

  @Override
  public int getColumnCount() {
    return myData == null ? 0 : myData.getLocales().size() + ConstantColumn.COUNT;
  }

  @Override
  public void setValueAt(Object value, int row, int column) {
    assert myData != null;

    if (ConstantColumn.KEY.ordinal() == column) {
      myData.changeKeyName(row, (String)value);
      fireTableRowsUpdated(0, myData.getKeys().size());
    }
    else if (ConstantColumn.UNTRANSLATABLE.ordinal() == column) {
      Boolean doNotTranslate = (Boolean)value;
      if (myData.setDoNotTranslate(keyOfRow(row), doNotTranslate)) {
        fireTableCellUpdated(row, column);
      }
    }
    else {
      if (myData.setTranslation(keyOfRow(row), localeOfColumn(column), (String)value)) {
        fireTableCellUpdated(row, column);
      }
    }
  }

  @NotNull
  @Override
  public Object getValueAt(int row, int column) {
    if (myData == null) {
      return "";
    }

    if (column >= ConstantColumn.COUNT) {
      Locale locale = localeOfColumn(column);
      return myData.getTranslations().contains(keyOfRow(row), locale) ? StringResourceData
        .resourceToString(myData.getTranslations().get(keyOfRow(row), locale)) : "";
    }
    switch (ConstantColumn.values()[column]) {
      case KEY:
        return keyOfRow(row);
      case DEFAULT_VALUE:
        return myData.getDefaultValues().containsKey(keyOfRow(row))
               ? StringResourceData.resourceToString(myData.getDefaultValues().get(keyOfRow(row))) : "";
      case UNTRANSLATABLE:
        return myData.getUntranslatableKeys().contains(keyOfRow(row));
      default:
        return "";
    }
  }

  @Override
  public String getColumnName(int column) {
    if (column >= ConstantColumn.COUNT) {
      // The names of the translation columns are set by TranslationHeaderCellRenderer.
      // The value returned here will be shown only if the renderer somehow cannot set the names.
      return LocaleMenuAction.getLocaleLabel(localeOfColumn(column), false);
    }
    return ConstantColumn.values()[column].name;
  }

  @Override
  public Class getColumnClass(int column) {
    if (column >= ConstantColumn.COUNT) {
      return ConstantColumn.DEFAULT_VALUE.sampleData.getClass();
    } else {
      return ConstantColumn.values()[column].sampleData.getClass();
    }
  }

  @Override
  public boolean isCellEditable(int row, int column) {
    return true;
  }

  @Nullable
  public String getCellProblem(int row, int column) {
    if (myData == null) {
      return null;
    }

    String key = keyOfRow(row);
    if (ConstantColumn.KEY.ordinal() == column) {
      return myData.validateKey(key);
    }
    else {
      Locale l = localeOfColumn(column);
      return myData.validateTranslation(key, l);
    }
  }
}