feat: Add support for Token Based validators (WIP #2)

Author: Steve Ramage

src/main/kotlin/net/sjrx/intellij/plugins/systemdunitfiles/intentions/ReplaceInvalidLiteralChoiceQuickFix.kt

@@ -0,0 +1,28 @@
+package net.sjrx.intellij.plugins.systemdunitfiles.intentions
+
+import com.intellij.codeInspection.LocalQuickFix
+import com.intellij.codeInspection.ProblemDescriptor
+import com.intellij.openapi.project.Project
+import com.intellij.psi.util.PsiTreeUtil
+import net.sjrx.intellij.plugins.systemdunitfiles.psi.impl.UnitFilePropertyImpl
+
+class ReplaceInvalidLiteralChoiceQuickFix(val offset: Int, val invalidToken : String, val replacementToken : String) : LocalQuickFix {
+
+  override fun getName(): String {
+    return "Replace '${invalidToken}' with '${replacementToken}'"
+  }
+
+  override fun getFamilyName(): String {
+    return "Replace invalid value"
+  }
+
+  override fun applyFix(project: Project, descriptor: ProblemDescriptor) {
+    val fullPropertyValue = descriptor.psiElement.text
+
+    val newText = fullPropertyValue.substring(0, offset) + replacementToken + fullPropertyValue.substring(offset+invalidToken.length)
+    val newElement = UnitElementFactory.createProperty(project, (descriptor.psiElement.parent as UnitFilePropertyImpl).key, newText)
+    val property = PsiTreeUtil.getParentOfType(descriptor.psiElement, UnitFilePropertyImpl::class.java)?: return
+
+    property.replace(newElement)
+  }
+}

src/main/kotlin/net/sjrx/intellij/plugins/systemdunitfiles/semanticdata/SemanticDataRepository.kt

@@ -12,9 +12,6 @@ import com.intellij.openapi.util.NotNullLazyValue
 import com.intellij.openapi.util.text.StringUtil
 import com.intellij.util.ObjectUtils
 import net.sjrx.intellij.plugins.systemdunitfiles.semanticdata.optionvalues.*
-import net.sjrx.intellij.plugins.systemdunitfiles.semanticdata.optionvalues.grammar.AllGrammarOptions
-import net.sjrx.intellij.plugins.systemdunitfiles.semanticdata.optionvalues.grammar.GrammarOptionValue
-import net.sjrx.intellij.plugins.systemdunitfiles.semanticdata.optionvalues.grammar.IOLimitOptionValue
 import org.apache.commons.io.IOUtils
 import java.io.BufferedReader
 import java.io.IOException
@@ -90,8 +87,11 @@ class SemanticDataRepository private constructor() {
           validatorMap.putAll(AllowedCpuSetOptionValue.validators)
           validatorMap.putAll(TtySizeOptionValue.validators)
           validatorMap.putAll(ExecDirectoriesOptionValue.validators)
+          validatorMap.putAll(CPUWeightOptionValue.validators)
           validatorMap.putAll(IOLimitOptionValue.validators)
-          validatorMap.putAll(AllGrammarOptions.validators)
+          validatorMap.putAll(ImagePolicyOptionValue.validators)
+          validatorMap.putAll(CPUWeightOptionValue.validators)
+          validatorMap.putAll(CPUSharesOptionValue.validators)
 
          // Scopes are not supported since they aren't standard unit files.
 

src/main/kotlin/net/sjrx/intellij/plugins/systemdunitfiles/semanticdata/optionvalues/CPUSharesOptionValue.kt

@@ -0,0 +1,22 @@
+package net.sjrx.intellij.plugins.systemdunitfiles.semanticdata.optionvalues
+
+import net.sjrx.intellij.plugins.systemdunitfiles.semanticdata.Validator
+import net.sjrx.intellij.plugins.systemdunitfiles.semanticdata.optionvalues.grammar.*
+
+
+class CPUSharesOptionValue : GrammarOptionValue(validatorName, GRAMMAR) {
+
+  companion object {
+    val validatorName = "config_parse_cpu_shares"
+
+    val GRAMMAR =
+      SequenceCombinator(
+          IntegerTerminal(2, 262145),
+          EOF()
+    )
+
+    val validators = mapOf(
+      Validator(validatorName, "0") to CPUSharesOptionValue()
+    )
+  }
+}

src/main/kotlin/net/sjrx/intellij/plugins/systemdunitfiles/semanticdata/optionvalues/CPUWeightOptionValue.kt

@@ -0,0 +1,25 @@
+package net.sjrx.intellij.plugins.systemdunitfiles.semanticdata.optionvalues
+
+import net.sjrx.intellij.plugins.systemdunitfiles.semanticdata.Validator
+import net.sjrx.intellij.plugins.systemdunitfiles.semanticdata.optionvalues.grammar.*
+
+
+class CPUWeightOptionValue : GrammarOptionValue(validatorName, GRAMMAR) {
+
+  companion object {
+    val validatorName = "config_parse_cg_cpu_weight"
+
+    val GRAMMAR =
+      SequenceCombinator(
+        AlternativeCombinator(
+          FlexibleLiteralChoiceTerminal("idle"),
+          IntegerTerminal(1, 10001)
+        ),
+        EOF()
+    )
+
+    val validators = mapOf(
+      Validator(validatorName, "0") to CPUWeightOptionValue()
+    )
+  }
+}

src/main/kotlin/net/sjrx/intellij/plugins/systemdunitfiles/semanticdata/optionvalues/IOLimitOptionValue.kt

@@ -0,0 +1,18 @@
+package net.sjrx.intellij.plugins.systemdunitfiles.semanticdata.optionvalues
+
+import net.sjrx.intellij.plugins.systemdunitfiles.semanticdata.Validator
+import net.sjrx.intellij.plugins.systemdunitfiles.semanticdata.optionvalues.grammar.*
+
+
+
+class IOLimitOptionValue() : GrammarOptionValue("config_parse_io_limit", GRAMMAR) {
+
+  companion object {
+    val GRAMMAR = SequenceCombinator(OneOrMore(SequenceCombinator(DEVICE, BYTES)), EOF())
+
+    val validators = mapOf(
+      Validator("config_parse_io_limit", "0") to IOLimitOptionValue()
+    )
+  }
+}
+

src/main/kotlin/net/sjrx/intellij/plugins/systemdunitfiles/semanticdata/optionvalues/ImagePolicyOptionValue.kt

@@ -0,0 +1,33 @@
+package net.sjrx.intellij.plugins.systemdunitfiles.semanticdata.optionvalues
+
+import net.sjrx.intellij.plugins.systemdunitfiles.semanticdata.Validator
+import net.sjrx.intellij.plugins.systemdunitfiles.semanticdata.optionvalues.grammar.*
+
+class ImagePolicyOptionValue : GrammarOptionValue(validatorName, IMAGE_POLICY_COMBINATOR) {
+
+  companion object {
+    val validatorName = "config_parse_image_policy"
+
+    // Image Polcies
+    // https://www.freedesktop.org/software/systemd/man/latest/systemd.image-policy.html
+    val IMAGE_POLICY_SEPARATOR = LiteralChoiceTerminal(":")
+    val PARTITION_IDENTIFIER = FlexibleLiteralChoiceTerminal( "root", "usr", "home", "srv", "esp", "xbootldr", "swap", "root-verity", "root-verity-sig", "usr-verity", "usr-verity-sig", "tmp", "var")
+    val PARTITION_POLICY_FLAG = FlexibleLiteralChoiceTerminal("unprotected", "verity", "signed", "encrypted", "unused", "absent", "read-only-off", "read-only-on", "growfs-off", "growfs-on")
+    val PARTITION_POLICY_FLAG_SEPARATOR = LiteralChoiceTerminal("+")
+    val PARTITION_IDENTIFIER_FLAG_SEPARATOR = LiteralChoiceTerminal("=")
+
+
+    val PARTITION_POLICY_FLAGS = SequenceCombinator(PARTITION_POLICY_FLAG, ZeroOrMore(SequenceCombinator(PARTITION_POLICY_FLAG_SEPARATOR, PARTITION_POLICY_FLAG)))
+
+
+    val SINGLE_EXPLICIT_IMAGE_POLICY = SequenceCombinator(PARTITION_IDENTIFIER, PARTITION_IDENTIFIER_FLAG_SEPARATOR, PARTITION_POLICY_FLAGS)
+    val DEFAULT_IMAGE_POLICY = SequenceCombinator(PARTITION_POLICY_FLAG_SEPARATOR, PARTITION_POLICY_FLAGS)
+    var IMAGE_POLICY= AlternativeCombinator(SINGLE_EXPLICIT_IMAGE_POLICY, DEFAULT_IMAGE_POLICY)
+    val IMAGE_POLICY_COMBINATOR = SequenceCombinator(IMAGE_POLICY, ZeroOrMore(SequenceCombinator(IMAGE_POLICY_SEPARATOR, IMAGE_POLICY)), EOF())
+
+    val validators = mapOf(
+      Validator(validatorName, "0") to ImagePolicyOptionValue()
+    )
+  }
+
+}