From 91db022755d1201d832057083b1bec973b7631da Mon Sep 17 00:00:00 2001
From: Andrew Brown <drusepth@gmail.com>
Date: Tue, 7 Mar 2023 20:10:28 -0800
Subject: [PATCH] define styles per page type

---
 app/services/basil_service.rb    | 20 ++++++++++++++++++++
 app/views/basil/content.html.erb | 18 ++++++++++--------
 2 files changed, 30 insertions(+), 8 deletions(-)

diff --git a/app/services/basil_service.rb b/app/services/basil_service.rb
index dcf76695..fa7b5eca 100644
--- a/app/services/basil_service.rb
+++ b/app/services/basil_service.rb
@@ -1,6 +1,26 @@
 class BasilService < Service
   IGNORED_VALUES = ['none', 'n/a', '.', '-', ' ', '?', '??', '???', 'x', nil]
 
+  def self.enabled_styles_for(page_type)
+    case page_type
+    when 'Character'
+      %w(realistic painting sketch digital abstract)
+    when 'Location'
+      %w(realistic painting sketch)
+    else
+      %w(Default)
+    end
+  end
+
+  def self.experimental_styles_for(page_type)
+    case page_type
+    when 'Character'
+      %w(anime painting2 horror watercolor)
+    else
+      []
+    end
+  end
+
   def self.include_all_fields_in_category(user, page, category_label)
     category = AttributeCategory.where(
       user_id:     user.id,
diff --git a/app/views/basil/content.html.erb b/app/views/basil/content.html.erb
index 46a5f640..e421042c 100644
--- a/app/views/basil/content.html.erb
+++ b/app/views/basil/content.html.erb
@@ -84,7 +84,7 @@ function commission_basil(style) {
     <div class="col s12 m2" style="margin-top: 1rem">
       <% if @can_request_another && @relevant_fields.any? %>
         <div class="grey-text center"><strong>Available styles</strong></div>
-        <% %w(realistic painting sketch digital anime abstract).each do |style| %>
+        <% BasilService.enabled_styles_for(@content.page_type).each do |style| %>
           <%= link_to "javascript:commission_basil('#{style}')" do %>
             <div class="hoverable card-panel purple white-text">
               <%= style.humanize %>
@@ -95,13 +95,15 @@ function commission_basil(style) {
       <% end %>
 
       <% if @can_request_another && @relevant_fields.any? %>
-        <div class="grey-text center"><strong>Experimental styles</strong></div>
-        <% %w(painting2 horror watercolor).each do |style| %>
-          <%= link_to "javascript:commission_basil('#{style}')" do %>
-            <div class="hoverable card-panel purple white-text">
-              <%= style.humanize %>
-              <i class="material-icons right">chevron_right</i>
-            </div>
+        <% if BasilService.experimental_styles_for(@content.page_type).any? %>
+          <div class="grey-text center"><strong>Experimental styles</strong></div>
+          <% BasilService.experimental_styles_for(@content.page_type).each do |style| %>
+            <%= link_to "javascript:commission_basil('#{style}')" do %>
+              <div class="hoverable card-panel purple white-text">
+                <%= style.humanize %>
+                <i class="material-icons right">chevron_right</i>
+              </div>
+            <% end %>
           <% end %>
         <% end %>
       <% end %>